加入CODE CHINA

· 不限速    · 不限空间    · 不限人数    · 私仓免费

免费加入
README.md

EfficientNet-Pytorch

A demo for train your own dataset on EfficientNet

Thanks for the >A PyTorch implementation of EfficientNet, I just simply demonstrate how to train your own dataset based on the EfficientNet-Pytorch.

Step 1:Prepare your own classification dataset


Then the data directory should looks like:

-dataset\
    -model\
    -train\
        -1\
        -2\
        ...
    -test\
        -1\
        -2\
        ...

Step 2: train and test

(1)You can choose to download the pre-trained model automatically or not by modify the line 169.

The pre-trained model is available on >release.

You can download them under the folder eff_weights.

(2)Change some settings to match your dataset. i.e. line13-22

   run efficientnet_sample.py to start train and test

(3)You can get the final results and the best model on dataset/model/.

项目简介

🚀 Github 镜像仓库 🚀

源项目地址

https://github.com/levigty/efficientnet-pytorch

发行版本

当前项目没有发行版本

贡献者 1

开发语言

  • Python 100.0 %