tensorflow_Resnet_train_test:用于在自己的数据集上训练不同体系结构(DenseNet,ResNet,AlexNet,GoogLeNet,VGG,NiN)的代码+多GPU支持+批处理和单个图像测试支持

时间:2024-06-07 13:50:54
【文件属性】:

文件名称:tensorflow_Resnet_train_test:用于在自己的数据集上训练不同体系结构(DenseNet,ResNet,AlexNet,GoogLeNet,VGG,NiN)的代码+多GPU支持+批处理和单个图像测试支持

文件大小:12.72MB

文件格式:ZIP

更新时间:2024-06-07 13:50:54

Python

tensorflow_Resnet_train_test 用于在自己的数据集上训练不同体系结构(DenseNet,ResNet,AlexNet,GoogLeNet,VGG,NiN)的代码+多GPU支持+批处理和单个图像测试支持 该存储库提供了一种易于使用的方法,用于在您自己的数据集上训练和测试不同的知名深度学习架构。 该代码直接从磁盘加载图像。 此外,还支持多GPU和传输学习,您也可以选择批量或单次测试图像。 基于存储库: #用法示例: 训练: 准备训练数据列表:python train_val_datalist_creater.py 培训或转移学习:python train.py 测试: python eval.py 或分批测试: 准备测试数据清单: python train_val_datalist_creater.py --create_data val 批量


【文件预览】:
tensorflow_Resnet_train_test-master
----README.md(1022B)
----eval.py(9KB)
----val.txt(3KB)
----log()
--------events.out.tfevents.1524149660.Sean(23.73MB)
----train.py(15KB)
----data_loader.py(5KB)
----train_val_datalist_creater.py(806B)
----architectures()
--------nin.py(2KB)
--------common.py(7KB)
--------resnet.py(2KB)
--------alexnet.py(2KB)
--------densenet.py(3KB)
--------__init__.py(1B)
--------vgg.py(2KB)
--------googlenet.py(5KB)
----train.txt(3KB)
----data()
--------train_data()
----arch.py(1KB)

网友评论