search

chensong1995 / HybridPose

HybridPose: 6D Object Pose Estimation under Hybrid Representation (CVPR 2020)
MIT License
412 stars 64 forks source link

About prediction_file in evaluate.py #16

Closed firstdeep closed 4 years ago

firstdeep commented 4 years ago

Thank you for sharing your excellent research.

I'm going to run train 1 and then test 1. However, the ADD value cannot be output because there is no prediction_file to proceed with the test.

So I want you to tell me how to save and get prediction_file file. https://github.com/chensong1995/HybridPose/blob/master/src/evaluate.py#L12

https://github.com/chensong1995/HybridPose/blob/master/src/evaluate.py#L65

Thank you very much.

Regards, David.

chensong1995 commented 4 years ago

Hello David,

Thank you for your interest in our research!

Please refer to this line for the prediction file. If you run train_core.py with our released pre-trained weights , the file will be automatically generated.

I hope this helps.