search

princeton-vl / Coupled-Iterative-Refinement

MIT License
104 stars 20 forks source link

Detector training pipeling #7

Closed Basilel7 closed 1 year ago

Basilel7 commented 1 year ago

Hello, Thanks for sharing your great work.

I wanted to know if you had a training pipeline for the MaskRCNN-detector in order to train the model on a custom dataset ?

Thanks

lahavlipson commented 1 year ago

We use the pre-trained MaskRCNN weights from CosyPose. The instructions for running their training pipeline are at the very bottom of their README: https://github.com/ylabbe/cosypose/blob/c90a04f434b1e89f02341cc03899eb63ea8facba/README.md#detection