
基于Pytorch的猫狗分类的代码演练
这个代码是以tensorflow为框架训练的,由于之前一直没有系统的了解tensorflow框架和Pytorch框架的区别,得着今天闲来无事,了解了一下,顺便看看能不能自己用Pytorch框架来训练猫狗分类的代码。Pytorch主要凭借其动态计算图和用户的易使用性在广大研究中脱颖而出,它以Python为基础,并提供了丰富的工具和接口,使得构建和训练神经网络变得简单快捷。这里一定要注意对图像进行归一化处理,保证每一张图的尺寸都是一样大小的,所以在这里我重新定义了transform1,将图片进行统一的预处理。






