분석시각화 대회 코드 공유 게시물은
내용 확인 후
좋아요(투표) 가능합니다.
[DICE] | Priavate 9등 | 7.70853 | HRNet | 경로/코드 수정
[수정]
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
데이콘(주) | 대표 김국진 | 699-81-01021
통신판매업 신고번호: 제 2021-서울영등포-1704호
서울특별시 영등포구 은행로 3 익스콘벤처타워 901호
이메일 dacon@dacon.io | 전화번호: 070-4102-0545
Copyright ⓒ DACON Inc. All rights reserved
일전에 코드공유를 해놓았으나, 사용자분들이 코드를 보고 사용하시기에 코드상에서 누락된 부분이 존재하였습니다.
따라서 누락된부분에 대해 데이콘 측의 허락을 받고 재업로드 하였습니다.