아리랑 위성영상 AI 객체 검출 경진대회

알고리즘 | 이미지 | 객체 탐지 | 컴퓨터 비전 | mAP

Prize : 1,000만원
680명 마감

Practice

Overview Data Code (share) Talk Leaderboard

Submission

[baseline 따라잡기]Google Colab에서 tfrecord 파일 생성하기

지비

2020.10.06 00:09 9,499 Views language

이 대회에는 baseline으로 참고할 수 있는 자료가 2가지 있습니다.

1. 대회안내 > 규칙 > 참고자료
2. 비슷한 대회인 "위성 이미지 객체 검출 경진대회"의 코드공유 자료 (https://dacon.io/competitions/official/235492/codeshare)

하지만 baseline 코드를 보면서 실행해도 잘 되지 않는 저같은 초보자분들도 많으실거라 생각합니다.

제가 보기엔 simplified_rbox_cnn 모델이 요구하는 labels.geojson 데이터가 이 대회에서는 다른 형태로 제공되는 것 같습니다. (특히 이 부분에서 정말 많은 시간이 소요되었습니다.)

수많은 시행착오 끝에 tfrecord 파일을 생성하는 것까지 성공했는데, 제 코드가 많은 분들에게 도움이 되면 좋겠습니다.

모든 코드는 Google Colab 상에서 작성 및 실행되었습니다.
(혹시 제가 실수하거나 잘못 이해한 부분이 있으면 댓글로 알려주시면 감사드리겠습니다.)

Code