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

updates

This commit is contained in:
Glenn Jocher 2019-03-25 15:32:00 +01:00
parent 8f0caff640
commit ff06b698ed
2 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -190,8 +190,6 @@ def train(
os.system('cp ' + latest + ' ' + weights + 'backup%g.pt' % epoch) os.system('cp ' + latest + ' ' + weights + 'backup%g.pt' % epoch)
# Calculate mAP # Calculate mAP
if type(model) is nn.parallel.DistributedDataParallel:
model = model.module
with torch.no_grad(): with torch.no_grad():
P, R, mAP = test.test(cfg, data_cfg, weights=latest, batch_size=batch_size, img_size=img_size) P, R, mAP = test.test(cfg, data_cfg, weights=latest, batch_size=batch_size, img_size=img_size)

1
weights/download_yolov3_weights.sh
View File

@ -18,4 +18,3 @@ wget -c https://pjreddie.com/media/files/darknet53.conv.74
# ./darknet partial cfg/yolov3-tiny.cfg yolov3-tiny.weights yolov3-tiny.conv.15 15 # ./darknet partial cfg/yolov3-tiny.cfg yolov3-tiny.weights yolov3-tiny.conv.15 15
# mv yolov3-tiny.conv.15 ../ # mv yolov3-tiny.conv.15 ../
cd ..