search

shanice-l / gdrnpp_bop2022

PyTorch Implementation of GDRNPP, winner (most of the awards) of the BOP Challenge 2022 at ECCV'22
Apache License 2.0
215 stars 49 forks source link

combine inferences into a full pipeline #12

Open tzsombor95 opened 1 year ago

tzsombor95 commented 1 year ago

Hi, I have trained my own custom model with synthetic data and would like to test it on my own real data. As far as I guess there is currently no code snippet that combines data loading, detection, gdrn and refinement algorithms. I see that there is a TODO in the core/gdrn_modeling/demo/demo_gdrn.py file. When do you expect this script to be ready? Maybe there is already a code snippet that I could start from to build the pipeline that would load the rgb and depth image and get the transformation matrices as output?

Thanks for your answers.

shanice-l commented 1 year ago

Sorry, we haven't made such a pipeline. Welcome to finish the code snippet and make a pull request :)

Martin9797 commented 1 year ago

Hi, if you are working on that kind of pipeline I would be very interested in your progress/results, since I am trying to integrate this into a bin-picking process for a bachelors thesis.

tzsombor95 commented 1 year ago

Hi, Yes, I managed to connect the individual elements into a complete pipeline. When I have some time, I will refactor some things on my code and make a pull request.

Martin9797 commented 1 year ago

If you already have a python file to build a pipeline, maybe you can just share that on its own on your github or something? I'm sure it would be a great help and something that I could build from.