yuyifan1991/efficientnet

Classify Object

EfficientNet

Install requirement

(1)pytorch version == v1.1
(2)cuda version >= 10.0
(3)python version >=3.6

How to train your owe data

(1)Pepare your data, make the directory list as:

• /your foldername
  • --train
    • --picture1
    • --picture2
    • -- ...
  • --test
    • --picture1
    • --picture2
    • -- ...
  • --model

(2) Run the train.py file

$ python train.py

The weight model will be saved in /your foldername/train/mode/xxx

How to test picture in batch*

Run the test.py file

$ python test.py

How to test picture in single*

Run the test-single.py file

$ python test-single

If you want make a API, you can reference the file jjk.py