[ssd] 사용법

[peternara]

어디?

• src : https://github.com/weiliu89/caffe/tree/ssd

설치

• 기본 Caffe 설치
• make에서 boost error 문제

  • Makefile을 기존 faster가 아니고, caffe-sl(visiona)꺼 사용
  • LIBRARIES += glog gflags protobuf boost_regex boost_system boost_iostreams boost_filesystem m hdf5_hl hdf5 부분에서

    • boost_system 추가
    • 파일 첨부 : Makefile

데이터셋 만들기

• dataset 만드는게 이해안감 ㅠ 다른것보다 어려움.
• 기본 데이터셋에 기반하여 그대로 똑같은 형식을 만듬

  • create_list.sh(caffe-ssd/data/VOC0712) 실행결과를 보면, 3가지 파일이 만들어짐

    • labelmap_voc.prototxt test.txt test_name_size.txt trainval.txt

  • 즉, create_list.sh과 동일한 역할을 하는 프로그램을 만듬.

    • tgpuc005.lab:~/users/peternara/ssd/make_training_set : 패션 데이터셋을 기반으로 만듬.

      • -> main 함수 참고

        • maker_ssd_VOCSSTYLE_XM_set() - ./main.out x_labels.list x x_20160514 x_sdd

          • test.txt test_name_size.txt text_fasterrcnn_testname.txt text_fasterrcnn_trainname.txt trainval.txt 파일 생성

        • maker_labelmap() - ./main.out x.list -> labelmap_voc.prototxt 생성

    • 실행하면 3가지 파일을 만들어짐
    • create_data.sh 를 실행하여 lmdb 파일생성 : x_ssd_trainval_lmdb & x_ssd_test_lmdb

      • VOCdevkit 의 형식과 일치 (image, annotation file path등등)

학습

• 기준이 되는 dir은 ./caffe/data, ./caffe/data/VOCdevkit
• examples/ssd/ssd_pascal.py 이 기본 학습 python
• 이를 이용하여, examples/ssd/ssd_x.py 하나 copy한다.

  • 이 파일 안에서, 다음과 같이 수정

    • 위에서 생성한 학습셋 패스 설정

      • train_data = "../VOCdevkit/x_ssd/lmdb/x_ssd_trainval_lmdb"
      • test_data = "../VOCdevkit/x_ssd/lmdb/x_ssd_test_lmdb"

    • resize_width, resize_height = 300 or 500
    • model_name = "" 부터 원하는 dir 정확히 바꿈 ..예들들어, snapshot_dir, job_dir, output_result_dir,
    • name_size_file = "data/x_ssd/test_name_size.txt"
    • pretrain_model = "models/VGGNet/VGG_ILSVRC_16_layers_fc_reduced.caffemodel"
    • label_map_file = "data/fxssd/labelmap_voc.prototxt"
    • num_classes = 13
    • gpus = "0,1,2,3,4,5,6,7" // tgpu 서버는, 하나만 사용하려면, gpus = "0"

      • 멀티gpu 학습 지원, 디바이스 id 지정

    • num_test_image = 1000 // 테스트 셋 개수 지정

• 실행 : python examples/ssd/ssd_x.py