기존 데이터를 다음과 같이 변경하여 저장하였습니다.
Normal chest PA -> 1 (618장)
EF_abnormal_기타제외 -> 2 (816장)
이제 1, 2로 이름이 바뀐 데이터 폴더는 다음과 같이 train, test로 분리하였습니다.
normal - train: 588장, test: 30장
abnormal - train: 786장, test: 30장
기본적인 모델은
https://pytorch.org/docs/stable/torchvision/models.html
에서 각 네트워크에서의 [SOURCE] 부분을 참조하여
main.py의 Net Class 대신에 사용하세요(이미지 사이즈 및 피처 크기 조절 필수)