i think there is a mistake in yoloLoss.py ， line 75
jinneigeyu opened this issue · 0 comments
jinneigeyu commented
coord_mask = target_tensor[:,:,5] > 0
noobj_mask = target_tensor[:,:,5] == 0
shuold be
coord_mask = target_tensor[:,:,4] > 0
noobj_mask = target_tensor[:,:,4] == 0