/identify

猫狗图片识别

Primary LanguagePython

学pytorch做的一个简单的小例子

data目录下的2个目录是猫和狗的图片

one.pth是训练1000次后保存的模型

训练模型

python train.py

开启web服务

python server.py