경구약제 이미지 분류(프로젝트 실패)

프로젝트 진행 현황

클래스 축소를 위해 전문의약품 제외
데이터 불균형 해결을 위한 데이터 제거
- 위도 75, 90도 만 남기고 모두 제거
- 각도 0, 80, 180, 260도 만 남기고 모두 제거
- rembg 로 배경 제거
550x550px 로 배경 크롭하여 용량문제 해결
Resnet, 50epoch 학습하여 테스트
- 같은 shape 인것은 잘 분류해내지만, 정확한 클래스명으로 분류하지는 못함
- VGGnet으로도 학습시켜보았지만 여전히 같은 증상
첫 분류로 shape 확인 후, 다시 정확한 클래스명으로 분류해내는 이중구조로 구현
- 여전히 shape 는 잘 분류해내지만, 정확한 클래스명으로 분류하지는 못함
MSAI School 강사와 멘토링 후 분류 불가능으로 결론내고, 프로젝트 종료
- 클래스 개수가 너무 많고, 약품마다 데이터 불균형이 너무 심해서 정확한 분류가 불가능하다고 판단내림

Name		Name	Last commit message	Last commit date
history		history
pt_files		pt_files
.gitignore		.gitignore
README.md		README.md
class_to_drug_name.py		class_to_drug_name.py
color_aug.py		color_aug.py
color_change 2.py		color_change 2.py
count_images.py		count_images.py
csv.xlsx		csv.xlsx
delete_camera_la.py		delete_camera_la.py
drug_Classifier_gradio.py		drug_Classifier_gradio.py
drug_Classifier_train.py		drug_Classifier_train.py
drug_dataset.py		drug_dataset.py
drug_shape.py		drug_shape.py
file_finder.py		file_finder.py
image_filter_traing.py		image_filter_traing.py
image_resize.py		image_resize.py
lisettxt.txt		lisettxt.txt
remove_background_Training.py		remove_background_Training.py
second_to_first.py		second_to_first.py
train_resnet18.py		train_resnet18.py

Navigation Menu