张量流-cnn-finetune
这个仓库是关于使用TensorFlow对一些著名的卷积神经网络进行微调的。
ConvNets:
要求:
Python 2.7或3.x
Tensorflow 1.x(已通过1.15.1测试)
OpenCV2(用于数据扩充)
数据集文件
您需要设置两个数据集文件以进行训练和验证。 格式必须如下所示:
/absolute/path/to/image1.jpg class_index
/absolute/path/to/image2.jpg class_index
...
class_index必须从0开始。
可以在和找到样本数据集文件。
不要忘了通过--num_classes运行时,标志finetune.py脚本。
亚历克斯网
进入alexnet文件夹
cd alexnet
微调
如果您以前没有下载过砝码,请下载。
./download_we
1