[pytorch] 图像分类tricks (持续更新...)

data augmentation

auto-augment:https://github.com/DeepVoltaire/AutoAugment/blob/master/autoaugment.py
fast-autoaugment: https://github.com/kakaobrain/fast-autoaugment
augmix: https://github.com/google-research/augmix
mixup/cutout: https://github.com/PistonY/torch-toolbox

model

pretrained-models.pytorch: https://github.com/Cadene/pretrained-models.pytorch

metric learning

pytorch-metric-learning: https://github.com/KevinMusgrave/pytorch-metric-learning
topkoptimization: https://github.com/BG2CRW/top_k_optimization

loss

Class-balanced-loss-pytorch: https://github.com/vandit15/Class-balanced-loss-pytorch

framework

pytorchimageclassification: https://github.com/hysts/pytorch_image_classification

……

发布了342 篇原创文章 · 获赞 794 · 访问量 178万+

猜你喜欢

转载自blog.csdn.net/u014365862/article/details/104216732