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

컴퓨터 비전 | 객체 검출 | mAP

  • 상금 : 1,000만원
  • 414팀
  • 마감
참여

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

  • day2020.10.06 00:09
  • views649 views
  • language Python
  • writer by 지비
댓글 4
이 대회에는 baseline으로 참고할 수 있는 자료가 2가지 있습니다. 1. 대회안내 > 규칙 > 참고자료 2. 비슷한 대회인 "위성 이미지 객체 검출 경진대회"의 코드공유 자료 (https://dacon.io/competitions/official/235492/codeshare) 하지만 baseline 코드를 보면서 실행해도 잘 되지 않는 저같은 초보자분들도 많으실거라 생각합니다. 제가 보기엔 simplified_rbox_cnn 모델이 요구하는 labels.geojson 데이터가 이 대회에서는 다른 형태로 제공되는 것 같습니다. (특히 이 부분에서 정말 많은 시간이 소요되었습니다.) 수많은 시행착오 끝에 tfrecord 파일을 생성하는 것까지 성공했는데, 제 코드가 많은 분들에게 도움이 되면 좋겠습니다. 모든 코드는 Google Colab 상에서 작성 및 실행되었습니다. (혹시 제가 실수하거나 잘못 이해한 부분이 있으면 댓글로 알려주시면 감사드리겠습니다.)
코드
댓글 4개
  • YONGMANHONGYONGMANHONG 2020.10.07 12:00

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

    • 지비지비 2020.10.07 21:48

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

    • YONGMANHONGYONGMANHONG 2020.10.08 09:19

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

    • 지비지비 2020.10.08 09:26

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

로그인이 필요합니다로그인이 필요합니다
목록으로