PyTorch图像模型多标签分类
基于timm的多标签分类。
更新2021/03/22
更新了./timm/models/multi_label_model.py、./train.py和./validate.py,以计算每个标签的精度。
介绍
该存储库用于多标签分类。 该代码基于 。 感谢罗斯的出色工作。
我于2021年2月27日下载了他的代码。 我认为我的多标签分类代码将与他的最新版本兼容,但我没有检查。
该是多标签分类的主要参考资料。 感谢Dmitry Retinskiy和Satya Mallick。 为了理解我们的上下文和数据集,尽管您无需阅读此处的特定代码,但请花5分钟阅读上面的链接。 将所有图像放入./fashion-product-images/images/。
为了实现多标签分类,我从Ross的pytorch-image-models中修改(添加)以下文件:
./
1