skhadem / 3D-BoundingBox

PyTorch implementation for 3D Bounding Box Estimation Using Deep Learning and Geometry
MIT License
435 stars 96 forks source link

build config file for pytorch training model #15

Open ai815 opened 4 years ago

ai815 commented 4 years ago

Hi there, I would like to tansfer the pytorch model to caffemodel myself. The cfg file, which saves the neural structure is necessary to be built. However the model we get after training is devided into 3 parts, demension, oriented, and confidence. so I am a bit confused how to write the cfg file on my own. Will be so appreciate if you can provide a solution or suggestion for me. Best regards.