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

updates

This commit is contained in:
Glenn Jocher 2020-01-17 17:58:37 -08:00
parent dec2c7d9a6
commit a4b8815ed9
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test.py
View File

@ -16,9 +16,9 @@ def test(cfg,
conf_thres=0.001, conf_thres=0.001,
iou_thres=0.5, # for nms iou_thres=0.5, # for nms
save_json=False, save_json=False,
single_cls=False,
model=None, model=None,
dataloader=None, dataloader=None):
single_cls=False):
# Initialize/load model and set device # Initialize/load model and set device
if model is None: if model is None:
device = torch_utils.select_device(opt.device, batch_size=batch_size) device = torch_utils.select_device(opt.device, batch_size=batch_size)

4
train.py
View File

@ -319,8 +319,8 @@ def train():
conf_thres=0.001 if final_epoch else 0.1, # 0.1 for speed conf_thres=0.001 if final_epoch else 0.1, # 0.1 for speed
iou_thres=0.6 if final_epoch and is_coco else 0.5, iou_thres=0.6 if final_epoch and is_coco else 0.5,
save_json=final_epoch and is_coco, save_json=final_epoch and is_coco,
dataloader=testloader, single_cls=opt.single_cls,
single_cls=opt.single_cls) dataloader=testloader)
# Update scheduler # Update scheduler
scheduler.step() scheduler.step()