Classification model (Pytorch) ?

lucifer443 / SpineNet-Pytorch

This project is a kind of implementation of SpineNet(CVPR 2020) using mmdetection.

Apache License 2.0

96 stars 11 forks source link

Classification model (Pytorch) ? #6

Open shruum opened 4 years ago

shruum commented 4 years ago

Hi @lucifer443 , Thanks for the pytorch implementation. Is there a Pytorch model of Spinenet pretrained on Imagenet (classification)? If I have to pre-train this classifier network using imagenet, what is the extra step? Thanks