--img-size stride-multiple verification

This commit is contained in:
Glenn Jocher committed 2020-06-12 22:10:46 -07:00
1 parent 4ba47b1b1b
commit 099e6f5ebd
4 files changed
+11 -4

No files matched your search

+1
View File
@@ -245,6 +245,7 @@ if __name__ == '__main__':
parser.add_argument('--augment', action='store_true', help='augmented inference')
parser.add_argument('--verbose', action='store_true', help='report mAP by class')
opt = parser.parse_args()
opt.img_size = check_img_size(opt.img_size)
opt.save_json = opt.save_json or opt.data.endswith('coco.yaml')
opt.data = glob.glob('./**/' + opt.data, recursive=True)[0] # find file
print(opt)