Skip to content

Terminal-K/HelloWorld

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

31 Commits
 
 
 
 

Repository files navigation

HelloWorld

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

About

for the 2th AI challenge of SCNU

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages