Skip to content

Commit

Permalink
added documentation on how to launch training
Browse files Browse the repository at this point in the history
  • Loading branch information
danielgehrig18 committed Sep 3, 2024
1 parent 3c9c717 commit b8a5c03
Showing 1 changed file with 21 additions and 0 deletions.
21 changes: 21 additions & 0 deletions readme.md
Original file line number Diff line number Diff line change
Expand Up @@ -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/
```

0 comments on commit b8a5c03

Please sign in to comment.