목록으로

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

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

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