maybe there is something wrong with pre-trained model?
when I python prepare_detection.py, I got this:
Traceback (most recent call last):
File "/home/cver/GithubExec/GaitGraph/src/preparation/prepare_detection.py", line 78, in
detection(**vars(args))
File "/home/cver/GithubExec/GaitGraph/src/preparation/prepare_detection.py", line 26, in detection
detector = DetectorYOLOv3(
File "/home/cver/GithubExec/GaitGraph/src/detector/detector_yolov3.py", line 33, in init
self.model.load_darknet_weights(self.weights_path)
File "/home/cver/GithubExec/GaitGraph/src/detector/models.py", line 307, in load_darknet_weights
conv_w = torch.from_numpy(weights[ptr : ptr + num_w]).view_as(conv_layer.weight)
RuntimeError: shape '[1024, 512, 3, 3]' is invalid for input of size 4617550
maybe there is something wrong with pre-trained model? when I python prepare_detection.py, I got this: Traceback (most recent call last): File "/home/cver/GithubExec/GaitGraph/src/preparation/prepare_detection.py", line 78, in
detection(**vars(args))
File "/home/cver/GithubExec/GaitGraph/src/preparation/prepare_detection.py", line 26, in detection
detector = DetectorYOLOv3(
File "/home/cver/GithubExec/GaitGraph/src/detector/detector_yolov3.py", line 33, in init
self.model.load_darknet_weights(self.weights_path)
File "/home/cver/GithubExec/GaitGraph/src/detector/models.py", line 307, in load_darknet_weights
conv_w = torch.from_numpy(weights[ptr : ptr + num_w]).view_as(conv_layer.weight)
RuntimeError: shape '[1024, 512, 3, 3]' is invalid for input of size 4617550