MMsegmentation如何输出新的模型评价指标

在使用MMsegmentation框架进行模型训练过程中,模型会默认的输出指定的评价指标,如果需要对评价指标进行更改,打印符合自己的评价指标,可以在“/mmsegmentation-main/mmseg/evaluation/metrics/iou_metric.py”中进行修改,这个文件中提供了以下三种类型

 

 每种类型会有自己的评价指标,可以选择自己需要的,如果都不符合,可以对其代码进行修改,让其打印符合自己要求的评价指标

 

 

  • 17
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
要修改mmsegmentation模型,您可以按照以下步骤进行操作: 1. 打开mmseg/core/optimizer/__init__.py文件,并在其中添加您的模块。例如,如果您的模块名为MyOptimizer,可以添加以下代码:`from .my_optimizer import MyOptimizer`。这样,注册器(registry)将会发现这个的模块并添加它。\[1\] 2. 在配置文件中找到optimizer域,并根据您的需求进行修改。例如,如果您想使用ADAM优化器,可以将optimizer字段修改为`optimizer = dict(type='Adam', lr=0.0003, weight_decay=0.0001)`。请注意,使用ADAM优化器可能会导致数值表现下降。\[2\] 3. 如果您想修改特定模型(如DANet),可以找到对应的配置文件。例如,如果您想修改danet_r50-d8_512x512_20k_voc12aug.py配置文件,可以打开该文件并进行修改。 - 在文件中找到_base_字段,并根据您的需求修改其中的路径。例如,您可以修改骨架模型的路径、数据集的路径等。\[3\] - 在model字段中,根据您的需求修改decode_head和auxiliary_head的num_classes参数,以适应您的分类类别数。\[3\] 请根据您的具体需求进行相应的修改。希望对您有所帮助! #### 引用[.reference_title] - *1* *2* [MMsegmentation教程 4: 自定义模型](https://blog.csdn.net/yanqianglifei/article/details/122598010)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [mmsegmentation教程1:自定义数据集、config文件修改、训练教程](https://blog.csdn.net/weixin_42748439/article/details/122845505)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值