|
@@ -36,7 +36,7 @@ class Yolov2BaseConfig(object):
|
|
|
self.val_topk = 1000
|
|
self.val_topk = 1000
|
|
|
self.val_conf_thresh = 0.001
|
|
self.val_conf_thresh = 0.001
|
|
|
self.val_nms_thresh = 0.7
|
|
self.val_nms_thresh = 0.7
|
|
|
- self.test_topk = 100
|
|
|
|
|
|
|
+ self.test_topk = 300
|
|
|
self.test_conf_thresh = 0.3
|
|
self.test_conf_thresh = 0.3
|
|
|
self.test_nms_thresh = 0.5
|
|
self.test_nms_thresh = 0.5
|
|
|
|
|
|