Skip to content

Commit

Permalink
Updated README to incorporate the description of the sketch-based ima…
Browse files Browse the repository at this point in the history
…ge retrieval and shape matching script.
  • Loading branch information
filipradenovic committed Oct 9, 2018
1 parent 0db13cf commit 0b34ed9
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ In order to run this toolbox you will need:
1. MatConvNet MATLAB toolbox version [1.0-beta25](http://www.vlfeat.org/matconvnet/download/matconvnet-1.0-beta25.tar.gz)
1. All the rest (data + networks) is automatically downloaded with our scripts

## Image retrieval (training and testing)
## Image retrieval

This code implements:

Expand Down Expand Up @@ -53,21 +53,22 @@ We provide the pretrained networks trained using the same parameters as in our E

**Note**: Data and networks used for training and testing are automatically downloaded when using the example scripts.

## Sketch-based image retrieval and shape matching (training coming soon, testing available)
## Sketch-based image retrieval and shape matching

This code implements:

1. Training (fine-tuning) CNN for sketch-based image retrieval and shape matching (coming soon)
1. Training (fine-tuning) CNN for sketch-based image retrieval and shape matching
1. Testing CNN sketch-based image retrieval on Flickr15k dataset

Run the following script in MATLAB:

```
>> run [MATCONVNET_ROOT]/matlab/vl_setupnn;
>> run [CNNIMAGERETRIEVAL_ROOT]/setup_cnnimageretrieval;
>> train_cnnsketch2imageretrieval;
>> test_cnnsketch2imageretrieval;
```
See ```[CNNIMAGERETRIEVAL_ROOT]/examples/test_sketch2cnnimageretrieval``` for additional details.
See ```[CNNIMAGERETRIEVAL_ROOT]/examples/train_cnnsketch2imageretrieval``` and ```[CNNIMAGERETRIEVAL_ROOT]/examples/test_cnnsketch2imageretrieval``` for additional details.

We provide the pretrained networks trained using the same parameters as in our ECCV 2018 paper. The Flickr15k dataset used in the paper is slightly outdated compared to the latest one that is automatically downloaded when using this code (0.1 difference in mAP), so we report results here:

Expand Down

0 comments on commit 0b34ed9

Please sign in to comment.