训练mindspore官方提供的mask_rcnn模型max_instance_count

问题描述:

官方提供的mask_rcnn代码

【操作步骤&问题现象】

1、mask_rcnn内处理数据集的代码中 mask标签会被pad的成 batch_size * max_instance_count * img_h * img_w的维度, 这样的话如果图片尺寸大或者一张图内目标数量特别多( >1000),那这个数据量会特别的大,想问下怎么修改能够不需要这个pad操作?

解答:

这个是由于不同图片的label数量不同,不做padding无法直接concat,组成一个batch时会报错。

在pytorch中,可以通过 collate_fn 函数解决该问题。

在mindspore中, Dataset基类 的 batch接口有一个 per_batch_map 参数,可以将batch对齐的函数传给 per_batch_map 参数来解决该问题。 详见:https://www.mindspore.cn/docs/zh-CN/master/api_python/dataset/mindspore.dataset.GeneratorDataset.html?highlight=per_batch_map#mindspore.dataset.GeneratorDataset.batch

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值