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()