是否可以更改Tensorflow对象检测api中的损失?

问题描述 投票:0回答:2

我想为对象检测(例如SSD)更改对象检测的丢失,

Q1:我想知道在哪里修改SSD的损耗函数,

Q2:是否可以通过定义损失微调数据集上的ssd_mobilenet?是好的还是必须使用我的损失函数从头开始训练ssd_mobile?

tensorflow deep-learning object-detection-api
2个回答
0
投票

Q1:如果使用object detection api,则使用配置来定义网络和损耗,例如:


0
投票

您可以在配置文件中更改丢失功能,例如链接中的第198行-https://github.com/tensorflow/models/blob/master/research/object_detection/samples/configs/ssd_mobilenet_v1_coco.config,这样做时,如果重新培训网络性能,性能可能会大大降低。

© www.soinside.com 2019 - 2024. All rights reserved.