8. pre-trained 모델 다운로드 Object Detection zoo에서 다운받으시면 됩니다. 저는 EfficientDet D0 512x512을 다운받겠습니다. object_detection 폴더에서 압축을 해제합니다. 9. pre-trained 모델 config 수정 object_detection/configs/tf2 경로에 있는 ssd_efficientdet_d0_512x512_coco17_tpu-8.config 파일을 복사해서 images 폴더에 붙여넣기 해줍니다. 방금 복사한 파일을 images 폴더에서 편집합니다. 1. num_classes -> 원하는 Object Detection 클래스 수 model { ssd { inplace_batchnorm_update: true freeze_..
4. 제너레이트 파일 다운 github_url에서 파일을 클론 받아서 README.md 파일을 제외하고 object_detection 폴더에 넣으시면 됩니다. git clone https://github.com/hojihun5516/object_detection_setting.git 그러면 해당 폴더에 아래와 같이 추가가 됩니다. 5. xml_to_csv 방금 내려받은 파일중 xml_to_csv.py를 실행해줍니다. xml로 되어있는 내용들을 csv로 컨버팅 해주는 작업입니다. #object_detection 폴더 python xml_to_csv.py 잘 실행이 되었다면 images폴더에 test_labels.csv, train_labels.csv 파일들이 생성됩니다. 6. generate tf reco..
안녕하세요 오늘은 Tensorflow를 활용한 Object Detection를 fine tuning해보겠습니다. 저는 조카가 좋아하는 뽀로로에 등장하는 캐릭터를 해보겠습니다. 🖐 기존 python이 설치되어 있고 tensorflow v2.3이 설치되어 있다는 가정 하에 진행하겠습니다. 목차 1. 텐서플로우 API 설치 2. Object detection 이미지 수집 3. 이미지 라벨링 4. 제너레이트 파일 다운 5. xml_to_csv 6. generate tf record 7. generate labelmap 8. pre-trained 모델 다운로드 9. pre-trained 모델 config 수정 10. Object detection 폴더 추가 생성 11. training 폴더 생성 12.이미지 학습..
