Tshzzz
Tshzzz
convert model to onnx.
I have the same problems. Is the YOLO_small.ckpt trained from scratch or covert from darknet models?
why real local_x should be 2?
> thank you maybe i was misunderstand > i think the center is 2.7,3.5 so this object goal is in the right bottom of box(2,3) and offset is (0.7,0.5) >...
请问你用的pytorch是什么版本？是否修改过代码。这个问题本身应该是，pred_cls或者pred_conf的维度不是2维。
不好意思，我检查一下。
> 谢谢了啊,麻烦了 我训练到290个epoch。也没什么问题。你可以把训练的 log发给我一下吗？[email protected]。
我使用的是pytorch 1.0.1，程序在我这里没有问题。可能因为你们使用的是pytorch 0.4导致的问题？我修改的代码，尝试去解决这个问题，你们可以去试试。
> 我在cls_prob = pred_cls[:, cls].float() * pred_conf[:, 0]这行代码前加了 > if pred_cls.shape == torch.Size([0]): > continue > 可以执行 嗯嗯！非常感谢