victory的博客

长安一片月,万户捣衣声

0%

Computer Vision | DogsVSCats

Dogs VS Cats

猫狗大战任务

猫狗大战是来源于Kaggle的一个比赛项目,任务为给定一个有猫狗照片数据集,设计一种算法对测试集中的猫狗图片进行分类。
比赛项目网址:
https://www.kaggle.com/
网络

layer1: Conv2d(3, 16, 3, padding=1)
layer2: relu():
layer3: max_pool2d(2)
layer4: Conv2d(16, 16, 3, padding=1)
layer5: relu()
layer6: max_pool2d(2)
layer7: Linear(50*50*16, 128)
layer8: relu()
layer9: Linear(128, 64)
layer10: relu()
layer11: Linear(64, 2)