分类：CNN图像分类

2020
10-08

Pytorch 使用CNN图像分类的实现

需求在4*4的图片中，比较外围黑色像素点和内圈黑色像素点个数的大小将图片分类如上图图片外围黑色像素点5个大于内圈黑色像素点1个分为0类反之1类想法通过numpy、PIL构造4*4的图像数据集构造自己的数据集类读取数据集对数据集选取减少偏斜cnn设计因为特征少，直接1*1卷积层或者在4*4外围添加padding成6*6,设计2*2的卷积核得出3*3再接上全连接层代码importtorchimporttorchvisionimporttorchvision.transformsastrans... 继续阅读 >