tensorflow_models_learning
老铁要是觉得不错,给个“star”
1.生成record训练数据
dataset已经包含了训练和测试的图片,请直接运行create_tf_record.py
对于InceptionNet V1:设置resize_height和resize_width = 224
对于InceptionNet V3:设置resize_height和resize_width = 299
其他模型,请根据输入需要设置resize_height和resize_width的大小
if __name__ == '__main__':
# 参数设置
resize_height = 224 # 指定存储图片高度
resize_width = 224 # 指定存储图片宽度
shuffle=True
log=5
#
1