-
1 votesanswersviews
如何在前进时从tensorflow slim模型VGG中提取特征?
我使用CASIA(人脸识别数据集)作为训练数据集,使用TensorFlow slim模型vgg训练了一个分类模型 . 我想通过使用LFW数据集来测试模型,它是一个面部匹配任务 . 所以我需要提取像fc7 / fc8这样的网络特征,而不是softmax图层,并比较特征之间的距离,以确定它们是否是同一个人 . 如何提取超薄型号的功能? 这是培训代码的一部分 . import tensorflow as... -
0 votesanswersviews
Tensorflow-没有名为'object_detection.protos'的模块
这是Geetha,我是一名做Msc的学生,我正在尝试使用google colab执行tensorflow对象检测 . 我正在尝试训练系统,我在执行train.py时遇到错误正在研究中,我的object_detection文件夹是在研究文件夹中 /usr/local/lib/python3.6/dist-packages/h5py/ init .py:36:FutureWarning:不推荐将i... -
0 votesanswersviews
尽管有批处理,Tensorflow-gpu上的内存不足(OOM)
我正在尝试使用tensorflow 1.8.0训练CNN . 这是一个相当大的网络,但我正在使用“批处理”在GPU上执行培训 . 但即使我在训练期间给出有效的批量大小,张量流也会因OOM而失败 . 当保持批量大小相同的值减少训练数据集时,tensorflow成功启动训练过程 . 我不明白的是,当训练样本数增加且批量值保持不变时,为什么tensorflow会给OOM . 批处理用于限制每次迭代的训练... -
1 votesanswersviews
将张量输入到张量流中的图形时,我得到一个无效的参数错误?
我在Windows 10上使用tensorflow 1.5版 . 我正在使用Tensorflow超薄模型的Inception V4网络,它已经从网站上获取,使用他们的预训练权重并在最后添加我自己的图层来对120个不同的对象进行分类 . 除了包含导入模块和数据集路径的行之外,这是完整的代码 . image_size = 299 tf.logging.set_verbosity(tf.logging.... -
3 votesanswersviews
无法在Google Cloud中训练我的Tensorflow探测器模型
我正在尝试基于Tensorflow样本和this post训练我自己的Detector模型 . 我确实在Macbook Pro上进行了本地培训 . 问题是我没有GPU并且在CPU上执行它太慢(每次迭代大约25秒) . 这样,我试图在tutorial之后运行Google Cloud ML Engine,但我无法使其正常运行 . 我的文件夹结构如下所述: + data - train.record ...