Mask_rcnn數據訓練

Mask rcnn往里面送的是數據原圖，可能沒有把mask送里面，而且mask的數據 56*56，仿照coco——inspect_data重點考慮

Bounding box都是長方形

?原圖掩模，消耗內存

?

Mask尺寸變?yōu)?6*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

總結

以上是生活随笔為你收集整理的关于BDD100k数据输入处理mask变为56*56的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。