I'm trying to implement the YOLOv8 and YOLOv9 on the code as other versions, but there is always a problem with the reshaping of the output tensor from the YOLO model, the outputs i get is torch.Size([4, 512, 100, 100]), while the output that should be is torch.Size([4, 603, 100, 100]), though I'm implementing the YOLO correctly.
I'm trying to implement the YOLOv8 and YOLOv9 on the code as other versions, but there is always a problem with the reshaping of the output tensor from the YOLO model, the outputs i get is torch.Size([4, 512, 100, 100]), while the output that should be is torch.Size([4, 603, 100, 100]), though I'm implementing the YOLO correctly.