Skip to content

Commit

Permalink
Fix broken links to pretrained models
Browse files Browse the repository at this point in the history
  • Loading branch information
peteanderson80 authored Nov 12, 2019
1 parent 151cea1 commit 4bb21fb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -122,11 +122,11 @@ Any NVIDIA GPU with 12GB or larger memory is OK for training Faster R-CNN ResNet

### Demo

1. Download [pretrained model](https://www.dropbox.com/s/tr24q7h0zm2wnjv/resnet101_faster_rcnn_final.caffemodel?dl=1), and put it under `data\faster_rcnn_models`.
1. Download [pretrained model](https://www.dropbox.com/s/wqada4qiv1dz9dk/resnet101_faster_rcnn_final.caffemodel?dl=1), and put it under `data\faster_rcnn_models`.

2. Run `tools/demo.ipynb` to show object and attribute detections on demo images.

3. Run `tools/genenerate_tsv.py` to extract bounding box features to a tab-separated-values (tsv) file. This will require modifying the `load_image_ids` function to suit your data locations. To recreate the pretrained feature files with 10 to 100 features per image, set MIN_BOXES=10 and MAX_BOXES=100. To recreate the pretrained feature files with 36 features per image, set MIN_BOXES=36 and MAX_BOXES=36 use this [alternative pretrained model](https://www.dropbox.com/s/bacig173qnxddvz/resnet101_faster_rcnn_final_iter_320000.caffemodel?dl=1) instead. The alternative pretrained model was trained for fewer iterations but performance is similar.
3. Run `tools/genenerate_tsv.py` to extract bounding box features to a tab-separated-values (tsv) file. This will require modifying the `load_image_ids` function to suit your data locations. To recreate the pretrained feature files with 10 to 100 features per image, set MIN_BOXES=10 and MAX_BOXES=100. To recreate the pretrained feature files with 36 features per image, set MIN_BOXES=36 and MAX_BOXES=36 use this [alternative pretrained model](https://www.dropbox.com/s/nu6jwhc88ujbw1v/resnet101_faster_rcnn_final_iter_320000.caffemodel?dl=1) instead. The alternative pretrained model was trained for fewer iterations but performance is similar.


### Training
Expand Down

0 comments on commit 4bb21fb

Please sign in to comment.