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

updates

This commit is contained in:
Glenn Jocher 2019-04-14 23:22:35 +02:00
parent 09949cdafa
commit 3c6b168a0a
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test.py
View File

@ -15,7 +15,7 @@ def test(
batch_size=16,
img_size=416,
iou_thres=0.5,
conf_thres=0.1,
conf_thres=0.001,
nms_thres=0.5,
save_json=False,
model=None

2
train.py
View File

@ -180,7 +180,7 @@ def train(
# Calculate mAP
with torch.no_grad():
results = test.test(cfg, data_cfg, batch_size=batch_size, img_size=img_size, model=model)
results = test.test(cfg, data_cfg, batch_size=batch_size, img_size=img_size, model=model, conf_thres=0.1)
# Write epoch results
with open('results.txt', 'a') as file: