Study

[obejct detection] Mask로부터 bounding box(bbox) 추출하기

Insight Miner 2022. 7. 18. 09:54
반응형

데이터 셋에 image와 mask만 있는 경우, obejct detection을 수행할 수 없다. 

Segmentation은 mask만 있어도 가능하다, bbox를 필요로 하는 알고리즘도 있다(ex: Mask R-CNN).

이때, mask를 통해 bbox를 추출하는 기능을 소개한다. 

 

https://pytorch.org/vision/main/auto_examples/plot_repurposing_annotations.html

 

Repurposing masks into bounding boxes — Torchvision main documentation

Shortcuts

pytorch.org

 

반응형