microsoft / singleshotpose

This research project implements a real-time object detection and pose estimation method as described in the paper, Tekin et al. "Real-Time Seamless Single Shot 6D Object Pose Prediction", CVPR 2018. (https://arxiv.org/abs/1711.08848).
MIT License
720 stars 215 forks source link

When I run train.py in PYCHARM, errors occured as follows: #188

Open longlizhi opened 1 year ago

longlizhi commented 1 year ago

/home/hndx/anaconda3/envs/yolo6d-1/bin/python /home/hndx/11z/singleshotpose-master-1/train. py Traceback (most recent calt Last): File"/hame/hngx/LLz/singleshotpose-master-/trainpy", Line 338, in modeL. Load_weights_until_last(initweightfile) FiLe"/home/hngx/u2/sinqleshotpose-master-/darknetpy", line 321, in Load_weights_until_Last start=load_convbn(buf, start, model[ el, model[1]) File "/home/hngx/1z/sinqieshotpose-master-1/ofq. py", tine 175, in 1oad. conv_hn conymodeL. weight. data. copy(torch. from_numpy (buf[ start: start+num-wl). view_as(conv_modeL. weight. data)); start =start+num-w RuntimeError: shape'[128,64,3,3]' is invalid for input of size 69839