--img-size stride-multiple verification
This commit is contained in:
1 parent
4ba47b1b1b
commit
099e6f5ebd
4 files changed
+11
-4
No files matched your search
@@ -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():
|
||||
|
||||
Reference in new issue
Block a user