yjh0410 1 an în urmă
părinte
comite
c91a179cf1
1 a modificat fișierele cu 2 adăugiri și 0 ștergeri
  1. 2 0
      train.py

+ 2 - 0
train.py

@@ -81,6 +81,8 @@ def parse_args():
                         help='load pretrained weight')
                         help='load pretrained weight')
     parser.add_argument('-r', '--resume', default=None, type=str,
     parser.add_argument('-r', '--resume', default=None, type=str,
                         help='keep training')
                         help='keep training')
+    parser.add_argument('--no_multi_labels', action='store_true', default=False,
+                        help='Perform NMS operations regardless of category.')
     parser.add_argument('--nms_class_agnostic', action='store_true', default=False,
     parser.add_argument('--nms_class_agnostic', action='store_true', default=False,
                         help='Perform NMS operations regardless of category.')
                         help='Perform NMS operations regardless of category.')