Adding .png to the list of allowed extensions in the im2rec.py (apach…

…e#9043)
rahul003 · Dec 12, 2017 · ee92983 · ee92983
1 parent cc34358
commit ee92983
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/im2rec.py b/tools/im2rec.py
@@ -216,7 +216,7 @@ def parse_args():
                         help='If this is set im2rec will create image list(s) by traversing root folder\
         and output to <prefix>.lst.\
         Otherwise im2rec will read <prefix>.lst and create a database at <prefix>.rec')
-    cgroup.add_argument('--exts', nargs='+', default=['.jpeg', '.jpg'],
+    cgroup.add_argument('--exts', nargs='+', default=['.jpeg', '.jpg', '.png'],
                         help='list of acceptable image extensions.')
     cgroup.add_argument('--chunks', type=int, default=1, help='number of chunks.')
     cgroup.add_argument('--train-ratio', type=float, default=1.0,