Skip to content

Latest commit

 

History

History
60 lines (50 loc) · 3.2 KB

README.md

File metadata and controls

60 lines (50 loc) · 3.2 KB
Raw

Check our paper here

How to use

First time

Create ./data folder

This step might not be necessary because this folder will be tracked by git. I leave this snnipet as a reference

  1. Create symbolic link to the dataset in PascalVOC format. The script will look for root at ../data/face-mask-detection
  2. Run python create_data_lists.py inside the repo

Other files

Check you have the following files:

  • calibril.ttf: Used to render text in PIL images (used in detect.py)
  • ssd300_PascalVOC.pth: State dict from the tutorial's pretrained model on PascalVOC

Initialize a screen

This will save all the outputs of the screen to a log file

  1. Make sure you have a logs/ dir
  2. Choose a exp_name (or any name you wish for the log file). Make sure it doesn't already exist.
  3. Run screen -S mask-SSD -L -Logfile logs/exp_name.txt
  4. Activate environment: conda activate pytorch
  5. Go to repo cd a-PyTorch-Tutorial-to-Object-Detection

Run scripts:

Train

The models obtained during training will be saved in a folder with the same exp_name. Models will be saved with the syntax model_{epoch}.pth, where epoch="final" for the last epoch.

  1. Define a new exp_name or continue an interrupted experiment (it must not contain a model_final.pth file)
  2. Change the parameters with which you want to experiment. The available options for learning are:
    • lr: Learning rate
    • decay_lr_at: Each time that training reaches one of these epochs, learning rate will decay
    • decay_lr_to: Learning rate will decay to this fraction of the existing learning rate
    • momentum: Momentum
    • weight_decay: Weight decay
    • grad_clip: Wheter to clip gradients in the range (-grad_clip, grad_clip) to avoid going to inf
  3. Change other running parameters if necessary. Options include:
    • print_freq: How many batches per epoch to print training losses
    • save_freq: How many epochs to save the model and evaluate on validation
    • batch_size: How many images per batch in training and evaluating
    • epochs: Total number of epochs to train
    • workers: Number of workers for loading data
    • split: Should only be changed at the end from 'val' to 'test'
  4. If there was a directory named exp_name/, the script will look for the last model. To use a specific checkpoint, set checkpoint = "model_name.pth" where appropiate
  5. Specify GPUs and cores. For example: CUDA_VISIBLE_DEVICES=0 taskset -c 0-7
  6. Run python train.py --exp exp_name with your experiment name
  7. The losses for training and validation will be appended to the file losses.txt

Evaluate

The main function from this script is used in the final epoch of training. However, it can still be used with a saved model

  1. Specify in filename the path to the model state dict
  2. Optionally, adjust the parameters batch_size, workers and split
  3. Run python eval.py

Visualize

This script creates a new folder inside your exp_name/ directory which will contain all the validation images with their predictions

  1. Specify in filename the path to the model state dict
  2. Like before, you can adjust parameters like batch_size, workers and split
  3. Run python detect.py