Object-Detection-Manga109

See:

• haunt98/Object-Detection-Manga109
• Manga109

SSD

# Train
python object-detection-main.py \
  -mode=1 \
  -fn=model_ssd.pth \
  -model=ssd \
  -bb=resnet50v2 \
  -min_ep=0 \
  -early_stopping=1 \
  -num_epochs=8 \
  -lr=0.0001 \
  -opt=SGD \
  -add_auth=0 \
  -bs=16 \
  -res=512 \
  -size32=0 \
  -size64=0 \
  -ar05=0 \
  -ar2=0 \
  -frame=0 \
  -text=0 \
  -dataset /datasets/Manga109_released_2023_12_07

# Inference
python object-detection-main.py \
  -mode=2 \
  -fn=model_ssd.pth \
  -inference_path=custom_inference_images \
  -model=ssd \
  -frame=0 \
  -text=0 \
  -det_thresh=0.50

Faster R-CNN

# Train
python object-detection-main.py \
  -mode=0 \
  -model=fasterrcnn \
  -fn=model_fasterrcnn.pth \
  -bb=resnet50v2 \
  -min_ep=0 \
  -early_stopping=1 \
  -num_epochs=4 \
  -lr=0.001 \
  -opt=SGD \
  -add_auth=0 \
  -bs=8 \
  -res=512 \
  -size32=0 \
  -size64=0 \
  -ar05=0 \
  -ar2=0 \
  -frame=0 \
  -text=0 \
  -dataset /content/drive/MyDrive/datasets/Manga109_released_2023_12_07

# Inference
python object-detection-main.py \
  -mode=2 \
  -fn=model_fasterrcnn.pth \
  -inference_path=custom_inference_images \
  -model=fasterrcnn \
  -frame=0 \
  -text=0 \
  -det_thresh=0.50