Skip to content

Commit

Permalink
Add link to pre-trained models to README.txt
Browse files Browse the repository at this point in the history
  • Loading branch information
javiribera authored and Javi Ribera committed May 8, 2019
1 parent e6837b3 commit 5e0ed00
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -73,8 +73,16 @@ python -m object-locator.train \
--save saved_model.ckpt
</pre>

## Pre-trained models
Models are trained separately for each of the four datasets, as described in the paper:
1. [Mall dataset](https://lorenz.ecn.purdue.edu/~cvpr2019/pretrained_models/mall,lambdaa=1,BS=32,Adam,LR1e-4.ckpt)
2. [Pupil dataset](https://lorenz.ecn.purdue.edu/~cvpr2019/pretrained_models/pupil,lambdaa=1,BS=64,SGD,LR1e-3,p=-1,ultrasmallNet.ckpt)
3. [Plant dataset](https://lorenz.ecn.purdue.edu/~cvpr2019/pretrained_models/plants_20160613_F54,BS=32,Adam,LR1e-5,p=-1.ckpt)
4. [ShanghaiTechB dataset](https://lorenz.ecn.purdue.edu/~cvpr2019/pretrained_models/shanghai,lambdaa=1,p=-1,BS=32,Adam,LR=1e-4.ckpt)

The [COPYRIGHT](COPYRIGHT.txt) of the pre-trained models is the same as in this repository.

## Uninstall

<pre>
conda deactivate object-locator
conda env remove --name object-locator
Expand Down

0 comments on commit 5e0ed00

Please sign in to comment.