danzhewuju/CNN_Image_class

CNN 图像多分类问题

Environments:

pip install tensorflow

pip install numpy

pip install skimage

Python3+

Introduction:

这是一个CNN来实现二分类的问题，（代码设计可以实现多分类）你需要训练自己的模型 run CNN.py ，测试自己的模型run CNN_test.py

训练后的结果会自动保存，数据集在dataset里面的文件夹，你需要自己对自己数据集的进行标注