首页 文章

如何从头开始训练模型,例如,使用初始v3或v4使用COCO数据集进行对象检测?

提问于
浏览
0

实际上我们正在使用faster_rcnn_inception_resnet_v2_atrous_coco预训练模型来训练我们自己的数据集图像,但我们想要改进我们的对象检测 .

如何从头开始训练模型,例如,使用初始v3或v4使用COCO数据集进行对象检测?我想生成一个名为faster_rcnn_inception_v3_coco的预训练模型,如https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md所描述的其他模型

1 回答

  • 0

    您可以手动替换基础网络,例如在ImageNet预训练的这些张量流量超薄模型:https://github.com/tensorflow/models/tree/master/research/slim(自述文件:预训练模型)

    我采用了InceptionV4的权重和模型,并将其用作SSD(单次多盒检测器)的基础网络 . 但是,它需要一些张量流的知识 .

    如果您没有时间或缺乏知识,您可能希望坚持使用高级API . 你可以试试NASnet模型 .

相关问题