欢迎光临散文网 会员登陆 & 注册

Tensorflow 1.15 训练 ssd mobilenet

2020-03-16 01:09 作者:刘开基  | 我要投稿

最近准备参加高通的端AI的比赛,做一个目标检测的小应用。涉及到目标检测模型的重新训练,从tensorflow lite的官方example里面找了一个基于pretrained model重新训练的教程:

https://medium.com/tensorflow/training-and-serving-a-realtime-mobile-object-detector-in-30-minutes-with-cloud-tpus-b78971cf1193

说是30分钟的教程,我是整整跑了一天。原因主要是tensorflow版本问题:

- tensorflow的example都是TF1的语法写的

- tensorflow的gpu安装教程都是对TF2写的

所以要用TF1的gpu训练的话,必然需要重新彻底的把nvidia driver和依赖重新装一遍。

最终用了tensorflow-gpu==1.15, cuda=10.0的版本总算配置好了环境。



Tensorflow 1.15 训练 ssd mobilenet的评论 (共 条)

分享到微博请遵守国家法律