本文整理汇总了Python中ssd.build_ssd方法的典型用法代码示例。如果您正苦于以下问题:Python ssd.build_ssd方法的具体用法?Python ssd.build_ssd怎么用?Python ssd.build_ssd使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类ssd
的用法示例。
在下文中一共展示了ssd.build_ssd方法的3个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_voc
# 需要导入模块: import ssd [as 别名]
# 或者: from ssd import build_ssd [as 别名]
def test_voc():
# load net
num_classes = len(VOC_CLASSES) + 1 # +1 background
net = build_ssd('test', 300, num_classes) # initialize SSD
net.load_state_dict(torch.load(args.trained_model))
net.eval()
print('Finished loading model!')
# load data
testset = VOCDetection(args.voc_root, [('2007', 'test')], None, VOCAnnotationTransform())
if args.cuda:
net = net.cuda()
cudnn.benchmark = True
# evaluation
test_net(args.save_folder, net, args.cuda, testset,
BaseTransform(net.size, (104, 117, 123)),
thresh=args.visual_threshold)
示例2: test_voc
# 需要导入模块: import ssd [as 别名]
# 或者: from ssd import build_ssd [as 别名]
def test_voc():
# load net
num_classes = len(VOC_CLASSES) + 1 # +1 background
net = build_ssd('test',args.model, 300, num_classes) # initialize SSD
net.load_state_dict(torch.load(args.trained_model))
net.eval()
print('Finished loading model!')
# load data
testset = VOCDetection(args.voc_root, [('2007', 'test')], None, VOCAnnotationTransform())
if args.cuda:
net = net.cuda()
cudnn.benchmark = True
# evaluation
test_net(args.save_folder, net, args.cuda, testset,
BaseTransform(net.size, (104, 117, 123)),
thresh=args.visual_threshold)
示例3: test_gtdb
# 需要导入模块: import ssd [as 别名]
# 或者: from ssd import build_ssd [as 别名]
def test_gtdb(args):
gpu_id = 0
if args.cuda:
gpu_id = helpers.get_freer_gpu()
torch.cuda.set_device(gpu_id)
# load net
num_classes = 2 # +1 background
# initialize SSD
net = build_ssd(args, 'test', exp_cfg[args.cfg], gpu_id, args.model_type, num_classes)
logging.debug(net)
net.to(gpu_id)
net = nn.DataParallel(net)
net.load_state_dict(torch.load(args.trained_model, map_location={'cuda:1':'cuda:0'}))
net.eval()
logging.debug('Finished loading model!')
dataset = GTDBDetection(args, args.test_data, split='test',
transform=BaseTransform(args.model_type, (246,246,246)),
target_transform=GTDBAnnotationTransform())
if args.cuda:
net = net.to(gpu_id)
cudnn.benchmark = True
# evaluation
test_net_batch(args, net, gpu_id, dataset,
BaseTransform(args.model_type, (246,246,246)),
thresh=args.visual_threshold)