tgisaturday / S3FD_ATSS_SAPD

Applying Adaptive Training Sample Selection and Soft Anchor Point Detection to S3FD
39 stars 11 forks source link

does there any resnets backbone version or mobilenets? #2

Open lucasjinreal opened 4 years ago

lucasjinreal commented 4 years ago

does there any resnets backbone version or mobilenets?

tgisaturday commented 4 years ago

Sorry but current model only supports vgg-16 backbone since it is based on SSD. But it seems like a good idea to apply different backbones to the model.

AlphaNumeric99 commented 4 years ago

Great work @tgisaturday ! Perhaps you can take a look at NVIDIA's SSD implementation using Pytorch and Resnet-50 at https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Detection/SSD and provide option to use different backbone accordingly. Thanks