Download
Contributors
- Happy623623
- 0-kyun
- gill1026
- chang-kyu
- ssh9871
현재 학습한 음식 데이터
-
UECFOOD100 데이터셋
- 밥
- 비빔밥
- 햄버거
- 피자
- 스파게티
- 돈까스
- 계란후라이
- 소시지
-
한국 음식 이미지
- 김치찌개
- 된장찌개
- 라면
- 삼겹살
- 짜장면
- 짬뽕
사용방법
-
data.zip을 다운받고 압축해제
-
data_process.py를 실행하여 food_dataset 폴더 생성
-
model_train.py를 실행하여 yolo모델 학습
-
test_images 폴더에 bbox를 표시할 사진들을 넣기
-
test.py를 실행하면 result_images 폴더가 생성됨
-
model_validate.py를 실행하면 best.pt의 성능을 출력함
실행 결과
