b [:, 4] + = math.log (8 / (640 / s) ** 2) # obj (8 objects per 640 image)
b [:, 5 :] + = math.log (0.6 / (m.nc-0.99)) cf가 None이면 else torch.log (cf / cf.sum ()) # cls
↓
b.data[:, 4] += math.log(8 / (640 / s) ** 2) # obj (8 objects per 640 image)
b.data[:, 5:] += math.log(0.6 / (m.nc - 0.99)) if cf is None else torch.log(cf / cf.sum()) # cls
yolo.py 수정
'====Pytorch > Scaled Yolov4' 카테고리의 다른 글
ImportError: libGL.so.1: cannot open shared object file: No such file or directory (0) | 2021.07.06 |
---|
댓글