Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add multi_scale train option to argparser

This commit is contained in:
Glenn Jocher 2018-11-05 09:08:48 +01:00
parent dc7b58bb3c
commit 2ccf68cf96
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -12,7 +12,7 @@ parser.add_argument('-data_config_path', type=str, default='cfg/coco.data', help
parser.add_argument('-cfg', type=str, default='cfg/yolov3.cfg', help='cfg file path') parser.add_argument('-cfg', type=str, default='cfg/yolov3.cfg', help='cfg file path')
parser.add_argument('-img_size', type=int, default=32 * 13, help='size of each image dimension') parser.add_argument('-img_size', type=int, default=32 * 13, help='size of each image dimension')
parser.add_argument('-resume', default=False, help='resume training flag') parser.add_argument('-resume', default=False, help='resume training flag')
parser.add_argument('-multi_scale', default=True, help='train at random img_size 320-608') parser.add_argument('-multi_scale', default=True, help='train at random img_size 320-608') # ensure memory for 608 size
opt = parser.parse_args() opt = parser.parse_args()
print(opt) print(opt)