You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+31-2
Original file line number
Diff line number
Diff line change
@@ -33,14 +33,43 @@ We convert the pretrained weights from Caffe so that we can load them using Kera
33
33
* For CIFAR10, we convert the numpy formats provided in the [official website](https://www.cs.toronto.edu/~kriz/cifar.html) into the png formats.
34
34
Download the [converted CIFAR10](https://drive.google.com/open?id=1Zy72S74AGDAX-OjLqR5NrVUmMCy653pT), extract the .zip file and place `cifar/` in `data/`.
35
35
36
-
## COMMING SOON ...
37
-
38
36
### Training
39
37
38
+
Use the following command to train VGG-CNN-F on CIFAR10.
39
+
40
+
```bash
41
+
$ bash scripts/run_train.sh '12 24 32 48' 0
42
+
```
43
+
44
+
The first argument is a string of integers separated by whitespaces to indicate the hash code lengths for training.
45
+
In this case, we train 4 VGG-CNN-F models and, they have 12, 24, 32, 48 bits of hash codes, respectively.
46
+
The second argument is an integer indicates the gpu_id.
47
+
The training parameters such as learning rates, epochs, early stopping can be set in `setting.cfg`.
48
+
40
49
### Testing
41
50
51
+
Use the following command to compute hash codes for CIFAR10.
52
+
53
+
```bash
54
+
$ bash scripts/run_test.sh '12 24 32 48' 0
55
+
```
56
+
57
+
The arguments here are the same as those in the training script.
58
+
This script takes the list of images stored in `sample_files/cifar10_supB_query.txt` and `sample_files/cifar10_supB_query.txt`, computes binary codes for them, and store the results under `experiments/`.
59
+
We represent binary codes as numpy arrays and store them as .npy formats.
60
+
42
61
### Evaluation (matlab required)
43
62
63
+
Use the following command for computing mAPs on CIFAR10.
64
+
65
+
```bash
66
+
$ cd scripts/
67
+
$ CODE_LEN=12 ## Use differnt code lengths for eavaluation
0 commit comments