加入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 %