
基于timm的PyTorch-Image-Models多标签分类
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目利用PyTorch-Image-Models库中的预训练模型,通过微调实现对图像的多标签分类任务,在计算机视觉领域具有广泛应用前景。
PyTorch图像模型多标签分类基于timm的实现已更新至2021年3月22日,此次更新主要针对./timm/models/multi_label_model.py、./train.py以及./validate.py文件进行了修改,以计算每个标签的具体精度。
本项目旨在进行多标签分类,并且代码基于罗斯的工作。我于2021年2月27日下载了他的原始代码作为基础。尽管我认为我的多标签分类实现应该与他的最新版本兼容,但未做进一步验证确认这一点。
该存储库是实施多标签分类的重要参考资料之一。同时也要感谢Dmitry Retinskiy和Satya Mallick对项目的支持。
为了更好地理解项目的背景以及所使用的数据集,请花时间阅读相关文档或描述(尽管这一步并非强制要求)。所有图像应放置在./fashion-product-images/images/目录下,以供程序使用。
为实现多标签分类功能,我从Ross的pytorch-image-models项目中修改了部分文件:包括但不限于上述提及的内容。
全部评论 (0)
还没有任何评论哟~


