diff --git a/readme.md b/readme.md index c5c2174..68257e5 100644 --- a/readme.md +++ b/readme.md @@ -147,3 +147,24 @@ python scripts/visualize_detections.py --detections_folder $LOG_DIR/$WANDB_DIR \ ``` This will start a visualization window showing the detections over a given sequence. If you want to save the detections to a video, use the `--write_to_output` flag, which will create a video in the folder `$LOG_DIR/$WANDB_DIR/visualization}`. + +## Training +To train on N-Caltech101, download the files with + +```bash +wget https://download.ifi.uzh.ch/rpg/dagr/data/ncaltech101.zip -P $DAGR_DIR/data/ +cd $DAGR_DIR/data/ +unzip ncaltech101.zip +rm -rf ncaltech101.zip +``` + +Then run training with + +```bash + +python scripts/train_ncaltech101.py --config config/dagr-l-ncaltech.yaml \ + --exp_name ncaltech_l \ + --dataset_directory $DAGR_DIR/data/ \ + --output_directory $DAGR_DIR/logs/ +``` +