if model.find('iter_120000') == -1: continue caffemodel = path + model print('Start evaluating: ' + caffemodel) net = caffe.Net(prototxt, caffemodel, caffe.TEST) net.name = os.path.splitext(os.path.basename(model))[0] cfg.net_name = net.name try: iter = int(net.name.split('_')[-1]) except: iter = 000000 if single_scale is True: single_scale_test_net(net, imdb, targe_size=input_size) else: if input_size == 320: multi_scale_test_net_320(net, imdb) else: multi_scale_test_net_512(net, imdb) mAP[iter] = cfg.mAP keys = mAP.keys() keys.sort() templine = [] print( "#########################################################################################################" ) print( "#########################################################################################################" ) if 'voc' in test_set: for key in keys:
net.name = os.path.splitext(os.path.basename(model))[0] cfg.net_name = net.name try: iter = int(net.name.split('_')[-1]) except: iter = 000000 if single_scale is True: single_scale_test_net(net, imdb, targe_size=input_size, vis=visualizeDets, redoInference=redoInference) else: if input_size == 320: multi_scale_test_net_320(net, imdb, vis=visualizeDets, redoInference=redoInference) else: multi_scale_test_net_512(net, imdb, vis=visualizeDets, redoInference=redoInference) mAP[iter] = cfg.mAP mPrec[iter] = cfg.prec mRec[iter] = cfg.rec # Also available for further evaluation (From MATLAB, not used yet): miss[iter] = cfg.miss #roc[iter] = cfg.roc #gt[iter] = cfg.gt #dt[iter] = cfg.dt