zju3dv / disprcnn

Code release for Stereo 3D Object Detection via Shape Prior Guided Instance Disparity Estimation (CVPR 2020, TPAMI 2021)
Apache License 2.0
213 stars 36 forks source link

Prediction file reading error #23

Closed zailongchen closed 4 years ago

zailongchen commented 4 years ago

Hi nice guy, I have run your trained model and obtained multiple files with the suffix .pkl. When I process them using the "pickle" package, I get the follow error:
data = pickle.load(f) ModuleNotFoundError: No module named 'disprcnn' Could you tell me how to obtain the inference results and analyse them? I am looking forward to your reply! Thank you!

My processing code is as follows: import pickle path = './000001.pkl' f = open(path, 'rb') data = pickle.load(f) print(data)

ootts commented 4 years ago

Perform the installation.

zailongchen commented 4 years ago

I have performed the installation and ran the trained model. I got the output file with the suffix .pkl. But I cannot open them. How can I read them and anlyse the results?

ootts commented 4 years ago

The average precision (AP) is supposed to be reported automatically once you've performed the evaluation. For further analysis, we have provided visualization code here.

zailongchen commented 4 years ago

Okay, got it! Thank you again!

ootts commented 4 years ago

Issue closed. Feel free to re-open it if you have more questions.