想遍历文件夹里的图片进行分类,我应该是在with里还是外加循环?
tensorflow吧
全部回复
仅看楼主
level 1
LempScar 楼主
求助!!
我已经保存了我训练好的model,想在测试集进行测试。
我是应该1. for循环遍历图片 2. with 创建对话加载模型 3. for循环滑动窗口 4. sess.run进行分类。 还是应该1和2交换位置?
因为对TF不太懂,发现就是写上tf.reset_default_graph() 仍然会在循环过程中速度不断变慢 本来一张图0.8左右,50张图之后都需要20s了。。
2018年12月28日 04点12分 1
level 3
看你测试集大小了
2018年12月28日 07点12分 2
测试集只有150张整图,但是滑动窗口有4个尺寸。我是想着要重置默认图的话 应该是在每个图片开始时重置,所以with应该在放在遍历图片的循环里,但好像并没有改变速度越来越慢的事实。
2018年12月29日 01点12分
测试集只有150张整图,但是滑动窗口有4个尺寸。我是想着要重置默认图的话 应该是在每个图片开始时重置,所以with应该在放在遍历图片的循环里,但好像并没有改变速度越来越慢的事实。
2018年12月29日 01点12分
level 8
首先 预测时间随时间增长 是因为 你每做一次循环 都会加载一次session,加载一次session耗时很大,解决办法就是在循环外面加载session,with tf.Session() as sess
然后 加载模型也要放在循环外面,这样子 先加载模型,然后运行会话,基本上每次预测只需要0.8S
2018年12月29日 03点12分 3
level 8
上面两个都加载之后 再开始做预测
2018年12月29日 03点12分 4
好的~谢谢~ 我先试试看~
2018年12月31日 00点12分
@LempScar 嗯,做预测的时候,最耗时间的就是创建会话和加载图
2018年12月31日 03点12分
1