Tensorflow 1.15 训练 ssd mobilenet
最近准备参加高通的端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的版本总算配置好了环境。