CMSEA (PyTorch)
CMSEA Architecture
Requirements
pip install torch>=1.4.0
pip install torchvision>=0.5.0
pip install pyyaml
Start Up
You can run the train.py to train or evaluate as follow:
python train.py /CaltechBirds-200/ --num-classes 200 --lr 3e-4 --epochs 200 --model tf_efficientnetv2_s
Contact Information
If you have any suggestion or question, you can leave a message here or contact us directly: [email protected]. Thanks for your attention!
Acknowledgment
Thanks to pytorch-image-models and EfficientNet.
