光点计划

介绍

有关光点计划计算机视觉方向代码

目录

1. 训练模型Blex
2. 训练模型vgg-16
3. 训练模型googlenet
4. 测试代码部分

使用说明

1. 训练模型Blex:

对Alex模型进行了修改，因为我认为图片过小，加入三层池化会使得图片之间差异性过小，可以适当减少池化层数量，并采用合适的步长，填充，卷积核大小使得卷积前后图片大小保持不变。

2. 训练模型vgg-16:

训练模型采用vgg-16，并用transform进行数据增强，训练样例和参数因为过大无法上传

3. 训练模型googlenet:

训练使用Googlenet模型，但因为该模型架构偏大，导致电脑GPU只能跑很小的batch_size；且测试时一次无法跑1000个测试样本。过拟合严重，模型效果不是很好。

4. 测试代码：

对测试代码进行了调整和修改，加入了transform对测试集进行归一化，并改为GPU训练，以适应训练的模型需要。同时测试代码保留了一个采用了transform进行数据增强的vgg-16模型，但效果一般。