모션 키포인트 검출 AI 경진대회

운동 | Keypoint Detection | 비전 | RMSE

  • Prize : 400만원
  • 650명
  • 마감
Join Competition

 

[DICE] | Priavate 9등 | 7.70853 | HRNet | 경로/코드 수정

손성빈 Follower 3 Following 0
2021.04.09 13:09 947 Views language
7
comments 1
3
4

[수정] 
Test.py의 78번째 linedls "Model Init"부분에서 Model.eval()부분이 빠져있습니다. 
model.load_state_dict(torch.load(model_state_file) 아래에 model.eval() 코드를 추가하여 Test를 진행하시면 되겠습니다.
이 경우 Priviate 기준 6.9054가 나오게 됩니다.

[학습]
1. Learning rate Scheduler 사용
2. Optimizer : Adam, Lr : 1e-3
3. TrainValide Split : X

[전처리]
WarpAffine 대신 Center Crop 이용 / Albumentation의 ShiftScaleRotate 사용

[후처리]
1. Detector를 사용하였으나 WarpAffine을 사용하지 못해 사람의 몸 비율이 바뀌는 현상이 존재해서 CenterCrop으로 대체
2. Heatmap의 Size가 1/4이므로 원본 이미지로 Heatmap를 바꾸기 위해 interpoltation 사용

[데이터 경로]
Test.py 파일의 testDataset의 root_dir에  자신의 데이터폴더경로를 기입하여 수정하시면 됩니다.
데이터 폴더 안에는 train폴더, test폴더, train.csv 파일이 있습니다. train폴더안에 train 이미지를 test폴더에 test이미지를 넣어주시면 됩니다.

※ VSCode 및 Linux  Ubuntu를 사용하여 이번 대회를 진행하였습니다. (Zip파일은 Data는 제외한 Code/Best.pth/Pretrain.pth로 이루어져 있습니다.)
Zip 파일 : https://drive.google.com/file/d/1q1d-3i8CYU10MYKlyrb7pT8vDkXljMxB/view?usp=sharing

Code
손성빈
2021.04.12 12:02

일전에 코드공유를 해놓았으나, 사용자분들이 코드를 보고 사용하시기에 코드상에서 누락된 부분이 존재하였습니다.
따라서 누락된부분에 대해 데이콘 측의 허락을 받고 재업로드 하였습니다. 

로그인이 필요합니다
0 / 1000
이전 글
이전 글이 존재하지 않습니다.
현재 글
[DICE] | Priavate 9등 | 7.70853 | HRNet | 경로/코드 수정
Competition - 모션 키포인트 검출 AI 경진대회
Likes 7
Views 947
Comments 1
4 months ago
다음 글
[에르모팀] | Priavate 2등 | 5.59514 | HRNet & Yolov5
Competition - 모션 키포인트 검출 AI 경진대회
Likes 17
Views 1,796
Comments 11
4 months ago