摘要:
猫狗分类,练手级代码,与手写数字识别相比,主要修改的地方是输出全连接层,将输出通道由10(十个数字)改成2(猫狗二分类)。还有一个是对数据集处理,因pytorch没有内置数据集函数,因此图片要自己处理。 数据要用opencv处理,归一化。 数据集:data __train__Cat | |__Dog 阅读全文
摘要:
手写数字识别是深度学习界的“HELLO WPRLD”。网上代码很多,找一份自己读懂,对整个学习网络理解会有帮助。不必多说,直接贴代码吧(代码是网上找的,时间稍久,来处不可考,侵删) 阅读全文