Open hannadiamond opened 2 years ago
I have the same problem, have you solved it yet?
@Pigdrum Unfortunately no, I ended up just using https://github.com/AlexeyAB/darknet for yolov4 tiny
maybe should change yolov4-tiny.cfg or change cfg.py example : filters=(your class num+5)*3
For the error in the second case you have mentioned -
RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces).
Solution: Replace all the view() command with reshape() command. It works in my case.
I have changed the cfg for the correct number of classes, filters, max batches, and steps
in the yolo loss function I have changed:
if self.n_anchors is set to 2, I get the error:
if the self.n_anchors is set to 3, then it can get all the way to the evaluate function but the hits this error:
Any help would be greatly appreciated