mmDetection3d

Cho's log·2024년 2월 6일

DeepLearning

목록 보기
1/1


1. 데이터셋 구조

  • 학습 전 데이터셋의 구조

  • 해당 블로그에선 sweeps 와 lidarseg는 제외

2. create_data.py 하기전 해야 할 코드 수정

  • 'CAM_FRONT' 를 제외하고 주석처리
  • configs/base/datasets/nus-mono.py


  • tools/dataset_converters/nuscenes_converter.py

  • tools/dataset_converters/update_infos_to_v2.py


  • update_infos_to_v2.py 수정 하지 않을 시 발생 에러 (train 중 에러발생)
  • 해당 이슈 Github

3. create_data.py 실행

  • python tools/create_data.py nuscenes --root-path ./data/nuscenes --out-dir ./data/nuscenes --extra-tag nuscenes --max-sweeps 0 실행

  • --max-sweeps 0 는 해당 블로그의 데이터셋에 sweeps 파일이 없기에 옵션에 추가

  • 성공 시 구조

4. train.py 전 코드 수정 및 실행

  • configs/_base__/default_runtime.py

  • interval=1 로 바꿀 시 epoch당 weights 파일 저장

  • mmdet3d/evaluation/metrics/nuscenes_metric.py

  • python tools/train.py ${CONFIG_FILE} [optional arguments]
  • --work-dir ${ouput_dir} 을 추가하면 output이 해당 경로에 저장된다.

0개의 댓글