TFOD API-用于物体检测再训练的基本卫生参数

时间:2018-10-20 02:47:21

标签: tensorflow object-detection

我正在使用TFOD API进行对象检测。我大约有230类对象,并且尝试使用Faster RCNN + resnet101体系结构。我使用了faster_rcnn_resnet101_coco预训练模型。在第一个迭代计划中,我将迭代次数设置为40000,并导出了训练好的ckpt模型并开始进行预测。 (当前,我正在学习了解事物的工作原理,因此我对获得更高的准确性不感兴趣)。

我已经注释了另外130个班级,并希望重新训练模型。但是,这次我想使用导出的模型(ckpt和co文件)代替faster_rcnn_resnet101_coco模型。这次我将迭代次数设置为80000。但是,当我开始训练时,TFOD API从30001迭代而不是1开始。

我的问题是,如何使用我以前训练的模型,并从迭代1开始而不是从迭代30001开始在模型上添加更多的类。

0 个答案:

没有答案