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

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

2021.04.09 13:09 6,903 조회 language

[수정] 
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

코드
로그인이 필요합니다
0 / 1000
구미사는공대생
2021.04.12 12:02

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