finetune_alexnet_with_tensorflow:用于在TensorFlow> = 1.2rc0中微调AlexNet的代码 源码

使用Tensorflow微调AlexNet 更新15.06.2016 我修改了整个代码库,以使用TensorFlow> =版本1.2rc0附带的新输入管道。 您可以在新的找到有关新输入管道的说明。您可以像以前一样使用此代码在自己的数据集上微调AlexNet,仅不再需要依赖OpenCV。 可以在找到旧代码。 该存储库包含在任意数据集上所需的所有代码。 除了代码本身的注释外,我还写了一篇文章,您可以在找到它的进一步解释。 您需要的只是预训练的权重,您可以在找到,也可以使用从caffe库进行。 如果您自己转换它们,请查看.npy weights文件的结构(字典的dict或列表的dict)。