search

aaronWool / rcvpose

(ECCV 22 Oral)Vote from the Center: 6 DoF Pose Estimation in RGB-D Images by Radial Keypoint Voting
GNU General Public License v3.0
49 stars 9 forks source link

UnboundLocalError: local variable 'keypoint_count' referenced before assignment #14

Open lin-fangzhou opened 1 year ago

lin-fangzhou commented 1 year ago

(venv310) blab@blab-server:~/pporzz/rcvpose$ python main.py --root_dataset="/home/bimlab/pporzz/rcvpose/Datasets/6dPoseData/" --mode=test --demo_mode True ape Traceback (most recent call last): File "/home/bimlab/pporzz/rcvpose/main.py", line 70, in trainer = Trainer(data, opts, vis) File "/home/bimlab/pporzz/rcvpose/train.py", line 25, in init self.Test() File "/home/bimlab/pporzz/rcvpose/train.py", line 194, in Test estimate_6d_pose_lm(self.opts) File "/home/bimlab/pporzz/rcvpose/AccumulatorSpace.py", line 518, in estimate_6d_pose_lm model_path = opts.model_dir + class_name+"_pt"+str(keypoint_count)+".pth.tar" UnboundLocalError: local variable 'keypoint_count' referenced before assignment

I don't have any ideas on this issue. Can someone help me? Or give a hint?

aaronWool commented 11 months ago

Hi, this bug should be fixed now. Thanks for pointing this out!