PyTorch_YOLO-Family
PyTorch_YOLO-Family copied to clipboard
target tensor with 0's
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