han-tae-ho / image-classification-level1-32 Goto Github PK

부스트캠프 AI Tech 2기 32조의 마스크 착용 상태 분류 프로젝트 입니다.

• 김종현
• 김준섭
• 유관식
• 이윤영
• 조성욱
• 한태호

• torch==1.6.0
• torchvision==0.7.0
• tensorboard==2.4.1
• pandas==1.1.5
• opencv-python==4.5.1.48
• scikit-learn~=0.24.1
• matplotlib==3.2.1

• pip install -r requirements.txt

• python train.py
• output : ./lab/[PersonalId or anonymous]/model/**

• python inference.py
• output : ./lab/[PersonalId or anonymous]/output/output.csv

• vi /etc/bash.bashrc
• export MYNAME=[PersonalId]
• source /etc/bash.bashrc

• pip install wandb
• wandb login
• https://wandb.ai/authorize 에서 api key 복사 후 터미널 입력

• chmod 711 some_task.sh
• nohup some_task.sh 1>/dev/null 2>&1 &