Selaa lähdekoodia

Allow loading labels from file

Martijn Buijs 7 vuotta sitten
vanhempi
commit
f6c43b515b
1 muutettua tiedostoa jossa 7 lisäystä ja 2 poistoa
  1. 7 2
      labelme/app.py

+ 7 - 2
labelme/app.py

@@ -1130,13 +1130,18 @@ def main():
     parser.add_argument('--output', '-O', '-o', help='output label name')
     parser.add_argument('--nodata', dest='store_data', action='store_false',
                         help='stop storing image data to JSON file')
-    parser.add_argument('--labels', help='comma separated list of labels')
+    parser.add_argument('--labels', help='comma separated list of labels OR file containing one label per line')
     parser.add_argument('--nosortlabels', dest='sort_labels',
                         action='store_false', help='stop sorting labels')
     args = parser.parse_args()
 
     if args.labels is not None:
-        args.labels = [l for l in args.labels.split(',') if len(l) > 0]
+        if os.path.isfile(args.labels):
+            args.labels = [l.strip()
+                           for l in open(args.labels, 'r').readlines()
+                           if len(l.strip()) > 0]
+        else:
+            args.labels = [l for l in args.labels.split(',') if len(l) > 0]
 
     app = QApplication(sys.argv)
     app.setApplicationName(__appname__)