高效网式火炬
在EfficientNet上训练自己的数据集的演示
感谢> ,我只是简单地演示如何基于EfficientNet-Pytorch训练自己的数据集。
步骤1:准备自己的分类数据集
然后,数据目录应如下所示:
-dataset\
-model\
-train\
-1\
-2\
...
-test\
-1\
-2\
...
步骤2:训练和测试
(1)您可以通过修改line 169来选择是否自动下载预训练模型。
可以在> 上获得预训练的模型。
您可以在eff_weights文件夹下下载它们。
(2)更改一些设置以匹配您的数据集。 即line13-22
run efficientnet_sample.py to start train and test
1