for the 2th AI challenge of SCNU
首先纪念第一次敲AI代码(不是全部啦😁),之前听了b站一个up主讲pytorch讲得贼好,同时又碰上第二届AI挑战赛,因此来尝试把流程完整地写一次,于是就有了这个HelloWorld。每件事都有helloworld,这不仅是github的,也是作为我AI学习的helloworld,所以就让它作为一块牌子立在起点这里吧。
以后可能会忘记怎么再去写这套流程,所以或许可以为以后的一些AI项目提供一个拉跨基线,当然以后如果有时间的话会改一改提高一下水平哈哈哈~~
ps: 比赛是菜品的识别任务,共172道菜,历时8天。简单采用了预训练模型resnet50,然后又是一波调参...,应该多去找找该类任务的新模型来尝试跑跑的... 这次比赛收获到的东西还是有的,也了解到了自己和18级师兄们的差距[抱拳]。
torch = 1.8.0
torchvision = 0.9.0
tensorboard
tqdm
numpy
pandas
opencv
PIL
caculate.py 计算训练集中各类的数量
cal_mean_std.py 计算训练集的均值与方差
data_ready.py 将数据及标签写入txt,以及划分数据集
dataloader.py 生成加载每批的数据
model.py 定义模型,以及优化器、损失函数和学习率衰减策略的配置
run.py 开始炼丹
1.计算数据集均值与方差 https://blog.csdn.net/dcrmg/article/details/102467434
2.数据集加载与处理 https://blog.csdn.net/shunshune/article/details/89316572
3.resnet50结构 https://blog.csdn.net/shanglianlm/article/details/86376627