文件名称:TensorRT-Program:使用TensorRT运行TensorFlow Official ResNet-50和VGG-19
文件大小:1.03MB
文件格式:ZIP
更新时间:2024-05-28 10:36:59
Python
TensorFlow-Slim图像分类模型库 是TensorFlow的新轻量级高级API( tensorflow.contrib.slim ),用于定义,训练和评估复杂模型。 该目录包含用于训练和评估使用TF-slim的几种广泛使用的卷积神经网络(CNN)图像分类模型的代码。 它包含脚本,使您可以从头开始训练模型或从预先训练的网络权重中微调模型。 它还包含用于下载标准图像数据集,将其转换为TensorFlow的本机TFRecord格式并使用TF-Slim的数据读取和排队实用程序进行读取的代码。 您可以轻松地在这些数据集中的任何一个上训练任何模型,如下所示。 我们还包括一个,其中提供了如何使用TF-Slim进行图像分类的工作示例。 有关开发或修改自己的模型的信息,另请参见。 联络人 TF-slim的维护者: 内森·西尔伯曼(Nathan Silberman),github:内森·西尔伯曼(