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,