配置训练模型xxxxx.config(tensorflow models zoo)

在如下文件夹中找到对应模型的config 

 这里以下图为例:

修改1:改成自己需要分类的数目 

修改2:改成自己一次需要训练的图片数目  

你的gpu够牛的话,随你玩。

修改3:改成自己下载模型的目录   

 修改4:改成自己下训练数据和标签的的目录

  修改5:改成自己下测试数据和标签的的目录

暂时只改了这几处。 

猜你喜欢

转载自blog.csdn.net/qq_30460949/article/details/89040604
zoo