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

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

2020.10.06 00:09 6,659 조회 language

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

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

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

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

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

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

코드
로그인이 필요합니다
0 / 1000
YONGMANHONG
2020.10.07 12:00

혹시 tfrecord 생성 하시고, 제공된 프로젝트에서 train.py로 실행되사나요?

지비
2020.10.07 21:48

그대로 실행하면 안되고, 추가로 설정해야할 것들이 좀 있습니다. 특히 config 파일이요.

YONGMANHONG
2020.10.08 09:19

아. config도 맞게 수정했는데, CUDA 관련 에러가 나서요 혹시 돌려보셨으면 Nvidia Driver / CUDA / CUDNN 버젼 어떤 거 쓰셨는지요?

지비
2020.10.08 09:26

저는 colab 쓰기 때문에 버전은 잘 모르겠네요.