product_detection icon indicating copy to clipboard operation
product_detection copied to clipboard

Published 20 hours ago • verified publisher icon tyomj

Model weights

Open mrgiba opened this issue 2 years ago • 5 comments

Hello,

Are the model weights available somewhere ?

Thank you

mrgiba avatar Oct 11 '22 23:10 mrgiba

Hi,

Yes, they are, please take a look at the comment here.

tyomj avatar Oct 12 '22 05:10 tyomj

Awesome

And is there any caveat for doing transfer learning using that pre trained model as a base ?

Thanks

mrgiba avatar Oct 12 '22 08:10 mrgiba

There are built-in features in the mmdetection framework to use a pre-trained checkpoint, such as load_from or resume_from. You may find more info in the original documentation.

tyomj avatar Oct 12 '22 09:10 tyomj

Is it needed to follow the mmdet v1 installation steps at https://github.com/open-mmlab/mmdetection/blob/v1.0.0/docs/INSTALL.md or should mmdet be installed differently ?

The conda install step there would install the latest pytorch version, but by the number of errors and deprecation messages I see here when running pip install -v -e . , I wonder if the installation steps should restrain pytorch to a specific version instead

mrgiba avatar Oct 12 '22 19:10 mrgiba

Yes, by restraining to PyTorch 1.4 backwards and also restricting the version of a couple of dependencies, I was able to setup, do transfer learning and run inference.

I can contribute with the exact installation steps I did

mrgiba avatar Oct 19 '22 11:10 mrgiba