https://github.com/Oh-JongJin/OBBE
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가 실행됨.
반응형
'Development > SAM' 카테고리의 다른 글
[SAM to OBB] Generate Point coord Tool (0) | 2024.11.11 |
---|---|
[SAM to OBB] Generate Point coord Tool (0) | 2024.11.10 |
SegmentAnyRGBD (0) | 2024.11.08 |
SAM2 (Segment Anything Model 2)에서 포인트를 이용한 세그멘테이션 (0) | 2024.11.04 |
SAM (OBB) 객체 추출 (0) | 2024.10.19 |