Tensorflow自定义对象检测训练速度

时间:2018-07-23 08:53:42

标签: tensorflow tensorflow-model-analysis

我试图训练一个frcnn来检测钱,它有3个班级。

我使用了此链接 https://github.com/EdjeElectronics/TensorFlow-Object-Detection-API-Tutorial-Train-Multiple-Objects-Windows-10 产生一些东西。 而且它使用了fast_rcnn_inception_v2_coco_2018_01_28配置

我花了太多时间才能解决这个问题或这是正常现象吗?

INFO:tensorflow:global_step/sec: 0.049895
INFO:tensorflow:Recording summary at step 1830.
INFO:tensorflow:Recording summary at step 1830.
INFO:tensorflow:global step 1831: loss = 0.2820 (22.984 sec/step)
INFO:tensorflow:global step 1831: loss = 0.2820 (22.984 sec/step)
INFO:tensorflow:global step 1832: loss = 0.2418 (``20.894 sec/step)

并保持那样的时代

我的Tensorflow版本b'v1.9.0-0-g25c197e023'1.9.0 Cuda编译工具,版本9.1,V9.1.85

我的计算机具有8GB内存GTX 950M和4GB内存 2.6Ghz i7-6700HQ

1 个答案:

答案 0 :(得分:0)

确定。

CUDA 9.1与Tensorflow GPU不兼容,所以我没有使用Tf-GPU