ImageClassification

图片分类模型

用于自定义图片embedding以及训练自定义的图片分类模型

准备

1.新建data/目录，然后按类别放置图片，比如：

data/类别1/图片1.jpg
data/类别2/图片2.jpg

2.新建checkpoint目录用于放置权重

3.训练：

python main.py

4.测试：

python test.py