data.py 整理数据并保存成csv
main.py
dataloader:处理数据
argumentation:数据增强
train：torch.hub里多种预训练模型
test：acc
依赖库：
pytorch(0.4以上)
apex: git clone https://github.com/NVIDIA/apex && cd apex && pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
albumentations
其余库可以直接pip install