该仓库主要完成了对数据集ChineseFoodNet的数据集类的PyTorch版本的编写,有助于各种深度学习框架快速训练。
并且使用EffiNet-L2(SAM)完成了对食物的208类分类网络的搭建。
这个是一个关于中国传统食物图片的数据集。其中包含了208类食物,数据集图片总量大小约20G,数量约18W,关于数据集的详细描述可以参见ChineseFoodNet。
该数据集作者声明可做公开学术交流,由于数据集过大,仓库不上传该数据集,仅提供下载连接,供有需要的下载学习。
- 图片数据:
- 标签数据:
该版本的代码由PyTorch实现,版本为1.8.1。
配置好环境后,需要将数据集下载到ChineseFoodNet文件夹,按照下图的文件目录组织即可:
- 项目仅作为学习交流使用
- 有任何问题可以留言
