DarkNet53_CatDog

dataset

ImageFolder를 사용하면 데이터셋 폴더의 구조를 기반으로 자동으로 라벨링된 데이터셋을 생성
각 클래스 폴더 내에는 해당 클래스에 속하는 이미지들이 저장
폴더의 이름을 클래스로 인식하여 이미지와 해당 클래스의 라벨을 매칭

train data cat dog 갯수 파악

test data는 파일명이 숫자로 표기되어 파악 불가

train : val = 9 : 1 = 22500 : 2500
val 생성
val 내부에 dog cat 생성 후 각 파일을 폴더 안으로 이동
나머지 데이터 dog cat 생성 후 각 파일을 폴더 안으로 이동

mkdir val val/dog val/cat
mv train/dog.{0..1249}.jpg val/dog/
mv train/cat.{0..1249}.jpg val/cat/

mkdir train/dog train/cat
mv train/*dog* train/dog/ 
mv train/*cat* train/cat/ 

train

batch 32

python main.py --mode train --output_dir ./output

tensorboard

tensorboard --logdir=./tensorboard/DOGS_CATS/

val

python main.py --mode eval --output_dir ./output 
			   --checkpoint ./output/DOGS_CATS/model_epoch57.pt

test

python main.py --mode test --output_dir ./output 
			   --checkpoint ./output/DOGS_CATS/model_epoch21.pt

ref

YOLOv3: An Incremental Improvement
dogs-vs-cats