文件名称:finetune_alexnet_with_tensorflow:用于在TensorFlow> = 1.2rc0中微调AlexNet的代码
文件大小:1.46MB
文件格式:ZIP
更新时间:2024-02-25 18:35:21
deep-learning tensorflow alexnet fine-tune DeeplearningJupyterNotebook
使用Tensorflow微调AlexNet 更新15.06.2016 我修改了整个代码库,以使用TensorFlow> =版本1.2rc0附带的新输入管道。 您可以在新的找到有关新输入管道的说明。您可以像以前一样使用此代码在自己的数据集上微调AlexNet,仅不再需要依赖OpenCV。 可以在找到旧代码。 该存储库包含在任意数据集上所需的所有代码。 除了代码本身的注释外,我还写了一篇文章,您可以在找到它的进一步解释。 您需要的只是预训练的权重,您可以在找到,也可以使用从caffe库进行。 如果您自己转换它们,请查看.npy weights文件的结构(字典的dict或列表的dict)。
【文件预览】:
finetune_alexnet_with_tensorflow-master
----datagenerator.py(5KB)
----images()
--------llama.jpeg(37KB)
--------sealion.jpeg(41KB)
--------zebra.jpeg(137KB)
----finetune.py(7KB)
----LICENSE(1KB)
----.gitignore(15B)
----alexnet.py(8KB)
----caffe_classes.py(21KB)
----.ipynb_checkpoints()
--------validate_alexnet_on_imagenet-checkpoint.ipynb(833KB)
----README.md(3KB)
----validate_alexnet_on_imagenet.ipynb(832KB)