image-classifier

🖼️ 분류용 이미지 레이블링 GUI 툴

* PyTorch를 통한 쓰레기 이미지 분류 AI 개발을 위해 시작된 프로젝트입니다.

PyQt5를 통한 가벼운 분류용 이미지 레이블링 GUI 툴입니다. 대량의 이미지를 20개 이내의 클래스로 수동으로 분류하는데에 적합합니다. 하위 폴더 사용, 클래스별 분류 결과 확인, JSON 자동 저장 기능을 지원합니다.

분류할 이미지가 포함된 폴더를 선택 후, 최대 20개의 클래스 목록을 수정합니다. 결과를 저장할 JSON 파일을 선택하면 분류를 진행할 수 있습니다. 뒤로 가기, 다시 실행을 사용할 수 있으며, 언제든지 분류할 폴더, JSON 파일과 클래스 목록을 수정할 수 있습니다. 좌측에서 현재 이미지의 미리보기를, 우측에서 클래스별 파일 목록을 통해 이미지 경로를 확인할 수 있습니다.

$ pip3 install -r requirements.txt

$ python3 app.py

* Windows 11 의 Python 3.8.8 환경에서 테스트되었습니다.