
slim微调
ShaoDu
了解图像处理,目标检测,目前专注Java相关领域
展开
-
Slim微调-fine-tuning
第一步,下载自己的数据集,数据集的格式要符合,否则对后面转化为TFRecords格式的文件有影响,导致不成功。 第二步,转换数据集为TFRecords格式的文件 进入下载以后的文件夹中/models/research/slim/,使用pycharm打开slim文件夹,打开转换格式的文件download_and_convert_data.py,只有cifar10、flowers、mnist这几种下载...原创 2019-03-02 19:34:03 · 670 阅读 · 1 评论 -
微调时训练时resnet替换最后一层表示与renset最后一层的输出节点
训练时resnet替换最后一层表示: checkpoint_exclude_scopes=resnet_v1_101/logits \ --trainable_scopes=resnet_v1_101/logits \ 输出节点: --output_node_name=resnet_v1_50/predictions/Softmax ...原创 2019-03-02 19:39:48 · 3975 阅读 · 3 评论 -
slim微调后的模型导出,对单张图片进行测试
这是slim微调之后的模型导出为pb文件后,值可以对单张图片进行测试的相关代码(单张,而非多张),如果多张只是使用for循化输入多张图片的话,会慢慢引起内存爆炸问题 多张图片测试:https://blog.youkuaiyun.com/ShaoDu/article/details/88781449 import tensorflow as tf import numpy as np from classify...原创 2019-03-24 20:49:54 · 485 阅读 · 0 评论