PyTorch_YOLO-Family icon indicating copy to clipboard operation
PyTorch_YOLO-Family copied to clipboard

target tensor with 0's

Open kulkarnikeerti opened this issue 2 years ago • 0 comments

I am trying to use mapillary with yolo-nano model. And I get the target tensor after applying this gt_creator with 0's in it. In if condition which is present in the gt_creator function (where gt_tensor is assigned with values) is not assigning the values to gt_tensor, hence it just returns the tensor wit 0's which is initialized in the beginning.

https://github.com/yjh0410/PyTorch_YOLO-Family/blob/24b92e3989a7506ba7e047c284fef69d7250a3b9/train.py#L302

Okay, so I tried to run the original script instead with mapillary data, even with coco dataset its the same. I get a tensor with 0's. This isn't the expected behavior right? We should have values in the gt_tensor?

Thanks in advance, Keerti

kulkarnikeerti avatar Apr 26 '22 19:04 kulkarnikeerti