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

xy and wh losses respectively merged

This commit is contained in:
Glenn Jocher 2019-02-20 12:50:40 +01:00
parent 44333ddd9f
commit 315fe6ec14
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
models.py
View File

@ -131,6 +131,7 @@ class YOLOLayer(nn.Module):
if p.is_cuda: if p.is_cuda:
self.grid_xy = self.grid_xy.cuda() self.grid_xy = self.grid_xy.cuda()
self.anchor_vec = self.anchor_vec.cuda() self.anchor_vec = self.anchor_vec.cuda()
self.anchor_wh = self.anchor_wh.cuda()
# p.view(bs, 255, 13, 13) -- > (bs, 3, 13, 13, 80) # (bs, anchors, grid, grid, classes + xywh) # p.view(bs, 255, 13, 13) -- > (bs, 3, 13, 13, 80) # (bs, anchors, grid, grid, classes + xywh)
p = p.view(bs, self.nA, self.nC + 5, nG, nG).permute(0, 1, 3, 4, 2).contiguous() # prediction p = p.view(bs, self.nA, self.nC + 5, nG, nG).permute(0, 1, 3, 4, 2).contiguous() # prediction

2
utils/gcp.sh
View File

@ -1,7 +1,7 @@
#!/usr/bin/env bash #!/usr/bin/env bash
# Start # Start
sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3 && python3 train.py --freeze sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3 && python3 train.py
# Resume # Resume
python3 train.py --resume python3 train.py --resume