We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1a8f590 commit 0ebcd2eCopy full SHA for 0ebcd2e
inference.py
@@ -29,7 +29,7 @@
29
help='number of data loading workers (default: 2)')
30
parser.add_argument('-b', '--batch-size', default=256, type=int,
31
metavar='N', help='mini-batch size (default: 256)')
32
-parser.add_argument('--img-size', default=224, type=int,
+parser.add_argument('--img-size', default=None, type=int,
33
metavar='N', help='Input image dimension')
34
parser.add_argument('--mean', type=float, nargs='+', default=None, metavar='MEAN',
35
help='Override mean pixel value of dataset')
0 commit comments