关于BDD100k数据输入处理mask变为56*56

本文探讨了Mask_RCNN模型的训练方法,特别是如何处理不同大小的输入图像和掩模数据。文章介绍了使用mini-mask减少内存负载的技术,并讨论了在高分辨率图像上的应用策略。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

Mask_rcnn数据训练

Mask rcnn往里面送的是数据原图,可能没有把mask送里面,而且mask的数据 56*56,仿照coco——inspect_data重点考虑

Bounding box都是长方形

 原图掩模,消耗内存

 

Mask尺寸变为56*56,影响不大

关于训练:

把mask变成这么小以后,接着在原图框出,送进去训练

因为shapes的图片128*128所以,可以原图直接进去,而BDD的图片太大,怎么处理?

Ballon案例中,数量比较少,直接原图送入

Examples of generated masks. These then get scaled and placed on the image in the right location.(GitHub-matterport)

所以mask产生以后,放在图片上,然后将图片送进去训练

在config.py中默认   

# If enabled, resizes instance masks to a smaller size to reduce

    # memory load. Recommended when using high-resolution images.

    USE_MINI_MASK = True

    MINI_MASK_SHAPE = (56, 56)  # (height, width) of the mini-mask

在model MaskRcnn中model.train()

Data_generator load_image_gt

load_image_gt(dataset, config, image_id, augment=augment,

                              augmentation=None,

                              use_mini_mask=config.USE_MINI_MASK)

转载于:https://www.cnblogs.com/lucky466/p/10241219.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值