search

hnmspirit / malaria_centernet

CenterNet for Malaria parasites detection
MIT License
3 stars 1 forks source link

Performance backbone! #1

Open phucnhs opened 4 years ago

phucnhs commented 4 years ago

Hi , I see you using a lot of backbone as effi,mb2,mb3,..do you compare the backbone to each other ? - do you have the performance of each one ? - i find effi quite good, why don't you use! Thanks !

hnmspirit commented 4 years ago

I did trained all the backbones for 20 epoch for comparison and found that EfficientNet, even B0, and ResNet34 have higher performance. However, they are much slower than ResNet18, especially EfficientNets as they use Depthwise Separable Convolution, which was not highly optimized in CUDA. So, in the limitation of time and hardware (gg colab), I only further trained ResNet18 up to 70 epoch to get the final model. The backbone can be further upgrade with stacks of BiFPN and please let me know if they achieved remarkable performance without much drop of efficiency. Tks for your response.