--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
@@ -156,6 +156,7 @@ if __name__ == '__main__':
parser.add_argument('--agnostic-nms', action='store_true', help='class-agnostic NMS')
parser.add_argument('--augment', action='store_true', help='augmented inference')
opt = parser.parse_args()
opt.img_size = check_img_size(opt.img_size)
print(opt)
with torch.no_grad():