load_name = os.path.join(
input_dir, 'fpn_{}_{}_{}.pth'.format(args.checksession,
args.checkepoch, args.checkpoint))
# classes = np.asarray(['__background__',
# 'aeroplane', 'bicycle', 'bird', 'boat',
# 'bottle', 'bus', 'car', 'cat', 'chair',
# 'cow', 'diningtable', 'dog', 'horse',
# 'motorbike', 'person', 'pottedplant',
# 'sheep', 'sofa', 'train', 'tvmonitor'])
classes = np.asarray(['__background__', 'diabete'])
if args.cascade:
if args.net == 'detnet59':
fpn = detnet_cascade(classes,
59,
pretrained=False,
class_agnostic=args.class_agnostic)
else:
print("network is not defined")
pdb.set_trace()
else:
if args.net == 'detnet59':
fpn = detnet_noncascade(classes,
59,
pretrained=False,
class_agnostic=args.class_agnostic)
else:
print("network is not defined")
pdb.set_trace()
fpn.create_architecture()
gt_boxes = gt_boxes.cuda()
# make variable
im_data = Variable(im_data)
im_info = Variable(im_info)
num_boxes = Variable(num_boxes)
gt_boxes = Variable(gt_boxes)
if args.cuda:
cfg.CUDA = True
# initilize the network here.
if args.cascade:
if args.net == 'detnet59':
FPN = detnet_cascade(imdb.classes,
59,
pretrained=True,
class_agnostic=args.class_agnostic)
else:
print("network is not defined")
pdb.set_trace()
else:
if args.net == 'detnet59':
FPN = detnet_noncascade(imdb.classes,
59,
pretrained=True,
class_agnostic=args.class_agnostic)
else:
print("network is not defined")
pdb.set_trace()
FPN.create_architecture()
