Retrain Tensorflow Object detection API

前端 未结 1 1468
一向
一向 2020-12-09 07:16

I have read tutorials on how to train a new class using TensorFlow Object Detection API. But what I want to do is to add a new class to the already trained classes of a pre-

相关标签:
1条回答
  • 2020-12-09 07:45

    Tensorflow object detection API supports starting from a pre-trained MS COCO checkpoint. Just set

    fine_tune_checkpoint: "/usr/home/username/tmp/model.ckpt-#####"
    from_detection_checkpoint: true
    

    In your detection pipeline. (Official reference: https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/configuring_jobs.md)

    You should add the images with your extra class to the coco dataset and finetune for all 91 classes, or your network might forget what it learned previously and only detect the new object.

    0 讨论(0)
提交回复
热议问题