병변 검출 AI 경진대회

알고리즘 | CV | 분류 | 검출 | mAP

상금 : 총 1,000만원
814명 마감

연습

대회안내 데이터 코드 공유 토크 리더보드

제출

train dataset COCO 변환

이여름

2021.11.16 17:41 3,782 조회 language

trainset annotation을 일괄적으로 COCO format으로 바꿔줍니다.

코드

댓글 6개

로그인이 필요합니다

comment

0 / 1000

stb

2021.11.16 19:01

감사합니다.

주냉이

2021.11.22 15:17

convert_to_coco('.', './train_annotations.json')
draw_bbox('./train/', './train_annotations.json', './examples/before_train/')
위 코드에서 각 경로는 어떻게 설정해 주면 되나요?

이여름

2021.11.23 13:12

convert_to_coco에서 첫번째 파라미터는 이 대회 데이터의 최상위 디렉토리 경로를 입력하시면 되고, 두번째는 저장할 디텍토리 경로/저장할 파일이름.json 을 지정해주시면 됩니다. draw_bbox에서 첫번째 파라미터는 convert_to_coco처럼 최상위 디렉토리 경로가 아닌 최상위 디렉토리/train/으로 입력하시면 되고, 두번째는 coco annotation 경로를, 세번째는 사진이 저장될 경로를 입력해주시면 됩니다.

주냉이

2021.11.23 16:32

오 감사합니다~!

머리올리자

2021.11.22 20:31

코드 공유 감사합니다!

올려 주신 코드에서 아래 화살표 부분이 추가되어야 정상 작동되던데 맞을까요~??

        for shape in tmp['shapes']:
            box = np.array(shape["points"]) # <===================
            x1, y1, x2, y2 = \
                    min(box[:, 0]), min(box[:, 1]), max(box[:, 0]), max(box[:, 1])

이여름

2021.11.23 13:16

아 맞네요 좋은 지적 감사합니다. 코드 수정했으니 참고해주시면 감사하겠습니다.

📣 댓글 작성 창의 위치가 댓글 리스트 상단으로 이동하였습니다!