YOLO3D icon indicating copy to clipboard operation
YOLO3D copied to clipboard

Possible bug in region loss calculation

Open sidml opened this issue 5 years ago • 0 comments

Hi

I was going through the region_loss code. I found a possible bug in build_targets function,

for t in range(nTrueBox):
    if target[b][t][1] == 0:
        break

I think it should have been if target[b][t][0] == 0 because the class is supposed to be n index 0. Index 1 contains the x coordinate. In the current code, the loss will be updated even if its a background class. Please correct me if my interpretation is incorrect. Thanks.

sidml avatar Sep 23 '19 08:09 sidml