how to use json2yolo

by

JSON2YOLO 포스팅

소개

JSON2YOLO는 YOLO(Object Detection 알고리즘 중 하나)에서 사용하는 데이터 포맷인 YOLO 포맷으로 변환하는 유틸리티입니다. 기존에 JSON 형식으로 되어 있는 객체 검출 데이터를 YOLO 포맷으로 변환하여 YOLO를 통해 객체 검출을 수행할 수 있습니다.

설치 및 사용법

설치

pip install json2yolo

사용법

  1. JSON 파일 준비

YOLO 포맷으로 변환할 원본 JSON 파일을 준비합니다. JSON 파일은 다음과 같은 구조를 가져야 합니다:

json
{
"image": "path/to/image.jpg",
"annotations": [
{
"class": "person",
"xmin": 100,
"ymin": 200,
"xmax": 300,
"ymax": 400
},
{
"class": "car",
"xmin": 500,
"ymin": 600,
"xmax": 700,
"ymax": 800
}
]
}

  1. JSON2YOLO 실행

bash
json2yolo --input path/to/input.json --output path/to/output.txt --classes path/to/classes.txt

  • --input: 입력 JSON 파일의 경로
  • --output: 출력 YOLO 포맷 파일의 경로

  • --classes: 클래스 이름이 기록된 텍스트 파일의 경로

  • 결과 확인

JSON2YOLO를 실행하면 변환된 YOLO 포맷의 파일이 --output으로 지정한 경로에 생성됩니다. 생성된 파일은 다음과 같은 형식을 갖습니다:

path/to/image.jpg 0.2 0.5 0.3 0.4 0
path/to/image.jpg 0.6 0.7 0.8 0.9 1

각 줄은 각 객체의 정보를 나타내며, 클래스의 인덱스, 바운딩 박스의 정규화된 좌표(xmin, ymin, xmax, ymax), 그리고 객체의 클래스를 포함합니다.

결론

이처럼 JSON2YOLO 유틸리티를 사용하면 기존에 JSON 형식으로 되어 있는 데이터를 YOLO 포맷으로 변환할 수 있습니다. YOLO 알고리즘을 사용해 객체 검출을 수행할 때 유용하게 활용할 수 있습니다.