DeviLily / food_cls

Food_cls Baseline

A simple baseline for SIGS_Big_Data_ML_Exam_2021.

https://www.kaggle.com/t/b7ed697207f0401b94a1f5c49c559d68

Environment

• python 3.6
• torch 1.5.1
• torchvision 0.6.0
• tqdm

必须用 GPU 跑

Download

下载数据到指定路径 ./data/food/,将三个文件夹分别移动到:

• ./data/food/train
• ./data/food/val
• ./data/food/test

Prepare

生成索引文件，创建数据集：

python prepare.py --src ./data/food/train --out ./data/food/train.txt

python prepare.py --src ./data/food/val --out ./data/food/val.txt

Hyper-parameter

root修改为你的项目本地路径

Train

CUDA_VISIBLE_DEVICES=0 python train.py

Inferance

功能为用训练好的模型测试 ./data/food/test 路径下的所有图片，并生成 ./data/submission.txt 文件

CUDA_VISIBLE_DEVICES=0 python inferance.py