Skip to content

boostcampaitech7/level1-imageclassification-cv-02

Repository files navigation

Sketch Image Classification

Boostcamp 내에서 ImageNet-Sketch 데이터셋을 기반으로 한 이미지 분류 대회를 진행하였습니다.

git 설치

apt-get update -y
apt-get install -y libgl1-mesa-glx libglib2.0-0
apt install -y git
git clone https://github.com/boostcampaitech7/level1-imageclassification-cv-02.git
cd level1-imageclassification-cv-02

Python package 설치

이 프로젝트는 Poetry를 사용하여 의존성을 관리합니다. 설치 방법은 다음과 같습니다:

# Poetry 설치
python -m venv competition1
source competition1/bin/activate
pip install poetry

1. poetry 의존성 update

poetry lock

2. 설치

poetry install
  • 추가로 설치한 패키지들도 있어서, pip도 해주어야 합니다. (lock으로 설정 X)
pip install -r requirements.txt

사용 방법

Train

python train.py --config configs/train_configs/train/config.yaml

Test (Inference)

python test.py --config configs/train_configs/test/config.yaml

Wandb

터미널에서 wandb login 실행 후, 본인의 API를 입력합니다.

  • Wandb 쓰는 방법 👉 --use_wandb 추가
python train.py --config configs/train_configs/train/config.yaml --use_wandb

sweep

  • sweep을 사용할 때는, configs/train_configs/train/config.yaml 내에 use_sweep을 True로 변경합니다.

sweep 사용 방법

  1. agent 생성
wandb sweep configs/train_configs/train/sweep.yaml
  1. agent 적용
wandb agent <your wandb agent> -- count 5

About

level1-imageclassification-cv-02 created by GitHub Classroom

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages