Dataset 구축 및 Training Code Follow Up

[Zerohertz]

🔨 Dead Line 🔨: 2023년 3월 20일

Training Data 및 Test Data 디렉토리 구축하고 모델 학습까지 해오기 !

데이터 디렉토리 구축 및 모델 학습은 이거 참고 ~

[Zerohertz]

STD 모델의 한계? -> YOLO 사용?

[c0olcode]

데이터 전처리할 때 train/test img/gt 개수 맞춰도 계속 파일 못찾는대서,,, 파일 이름 비교해서 데이터 삭제시키고 다시 훈련 돌리겠슴다

[woo-ara]

epoch 10으로 학습

[c0olcode]

epoch 10으로 학습 -> 성능은,,, gurim

[seohl16]

학습 돌리면서 생겼던 이슈

1. zip 파일 업로드 -> 해결 : wget & unzip 조합으로 해결
2. ModuleNotFoundError: No module named 'models.post_processing.pa.pa'에러 발생 -> 해결 : https://github.com/whai362/pan_pp.pytorch/issues/108 -> cpython compile은 안해서 생겼던 에러

   cd ./models/post_processing/pa/
   python setup.py build_ext --inplace
   cd ../boxgen/
   python setup.py build_ext --inplace
   cd ../../../
   echo Done!

   코드를 실행해서 해결

Epoch10 돌린 결과 스크린샷

Epoch10 loss 0.420, epoch 1 때 0.610에 비해 줄어들었다.