sota

Author: HiKapok

dataset/convert_tfrecords.py

@@ -29,7 +29,7 @@
 
 import dataset_common
 
-'''How to order your dataset:
+'''How to organize your dataset folder:
   VOCROOT/
        |->VOC2007/
        |    |->Annotations/

eval_ssd.py

@@ -291,7 +291,7 @@ def ssd_model_fn(features, labels, mode, params):
     with tf.device('/cpu:0'):
         bboxes_pred = decode_fn(location_pred)
         bboxes_pred = tf.concat(bboxes_pred, axis=0)
-        selected_bboxes, selected_scores = parse_by_class(cls_pred, location_pred,
+        selected_bboxes, selected_scores = parse_by_class(cls_pred, bboxes_pred,
                                                         params['num_classes'], params['select_threshold'], params['min_size'],
                                                         params['keep_topk'], params['nms_topk'], params['nms_threshold'])
 

test_tf.py

@@ -186,6 +186,7 @@ def voc_eval(detpath,
     # extract gt objects for this class
     class_recs = {}
     npos = 0
+
     for imagename in imagenames:
         R = [obj for obj in recs[imagename] if obj['name'] == classname]
         bbox = np.array([x['bbox'] for x in R])
@@ -195,11 +196,10 @@ def voc_eval(detpath,
         class_recs[imagename] = {'bbox': bbox,
                                  'difficult': difficult,
                                  'det': det}
-
     # read dets
-    detfile = detpath.format(classname)
-    with open(detfile, 'r') as f:
+    with open(detpath, 'r') as f:
         lines = f.readlines()
+
     if any(lines) == 1:
 
         splitlines = [x.strip().split(' ') for x in lines]