MapleStory Finger Point

Development/SAM

Oriented Bounding Boxes Label Editor

吳鍾振 2024. 11. 8. 11:37

https://github.com/Oh-JongJin/OBBE

 

GitHub - Oh-JongJin/OBBE: Oriented Bounding Boxes Editor

Oriented Bounding Boxes Editor. Contribute to Oh-JongJin/OBBE development by creating an account on GitHub.

github.com

OBB 데이터셋 레이블 수정 가능한 툴 작성 완료
OBB 데이터셋 포맷은 class_index x1 y1 x2 y2 x3 y3 x4 y4 의 형태로 이루어져 있음.
위 Github 링크의 OBBE/main.py에서 

if __name__ == '__main__':
    try:
        label_dir = "labels/DOTA-v1.5_val_RESULT"
        image_dir = "images"
        
        app = QApplication(sys.argv)
        viewer = PolygonViewer(label_dir, image_dir)
        viewer.show()
        sys.exit(app.exec_())
        
    except FileNotFoundError:
        print("Error: Could not find the polygon data file.")

입력으로 레이블이 포함된 상위 폴더, 레이블과 같은 파일명의 이미지가 포함된 상위 폴더 경로가 필요함!!
객체 클릭 후 키보드 E키 입력 시 수정 모드로 들어가게 되며 4개의 Point 좌표를 변경할 수 있음.
객체 더블 클릭 시 class 번호를 수정할 수 있는 Input Dialog가 실행됨.

반응형