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
반응형