From ee92983cd7ae0de4bb8c412a7b4af3ffd78a6e05 Mon Sep 17 00:00:00 2001 From: Eric Junyuan Xie Date: Tue, 12 Dec 2017 15:32:17 -0800 Subject: [PATCH] Adding .png to the list of allowed extensions in the im2rec.py (#9043) --- tools/im2rec.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/im2rec.py b/tools/im2rec.py index ec6de1969414..911a65c915a7 100644 --- 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 .lst.\ Otherwise im2rec will read .lst and create a database at .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,