CuteBoiz / EfficientNet_Classifier

EficientNet_Classifier

I. Train EfficientNet Classifier.

python3 train.py

• Arguments:

Arguments	Type	Default	Help

• Data path example:

DataPath
|
|__train.txt
|__test.txt
|__valid.txt
|__label.txt

• train/valid/test.txt files:

"path/to/image" class_id
"path/to/image1.png" 0
"path/to/image2.jpg" 2
"path/to/image3.png" 1

• label.txt:

dog
cat

II. Run test set and visualize result onto excel file.

python3 test.py

• Arguments:

Arguments	Type	Default	Help

III. Inference on image/video/images folder.

python3 infer.py

• Arguments:

Arguments	Type	Default	Help

IV. Export to ONNX engine.

python3 export.py

• Arguments:

Arguments	Type	Default	Help