search

kachiO / class-agnostic-counting-pytorch

PyTorch implementation of Class Agnostic Counting model
5 stars 1 forks source link

Pretrained model #1

Closed Viresh-R closed 3 years ago

Viresh-R commented 3 years ago

Hey, any chance you could share your pre-trained imagenet vid model ? I need to apply it to a different task, and would be great to get the pretrained model.

Thanks, Viresh

kachiO commented 3 years ago

Hey Viresh,

I forgot to save the weights to my local machine and the repo. I no longer have access to machine on which I used to train. A possible workaround to not training the model yourself is to load the pretrained weights from the original author’s keras model. There are a few examples online on how to do this, one that comes to mind is from HuggingFace. Good luck,

Kachi

On Sun, Nov 1, 2020 at 3:35 PM Ranjan notifications@github.com wrote:

Hey, any chance you could share your pre-trained imagenet vid model ?

Thanks, Viresh

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub https://github.com/kachiO/class-agnostic-counting-pytorch/issues/1, or unsubscribe https://github.com/notifications/unsubscribe-auth/ADHUWISEAOS7XQUMAKFM6FDSNXA73ANCNFSM4TGXQTXA .

Viresh-R commented 3 years ago

Hi Kachi, Thanks for the suggestion ! I ended up using the original Keras training code from authors. Anyways, issue resolved 👍

Viresh