Skip to content

Commit

Permalink
Update README
Browse files Browse the repository at this point in the history
  • Loading branch information
xuanbinh-nguyen96 committed Sep 7, 2020
1 parent 5e43db8 commit c9b8893
Showing 1 changed file with 20 additions and 3 deletions.
23 changes: 20 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,19 @@ AIOZ BMVC 2020: "Deep Metric Learning Meets DeepClustering: An Novel Unsupervise

This repository is the implementation of `CBSwR` for Unsupervised Deep Metric Learning task. Our model achieved **55.9**, **37.6** on NMI, and **47.5**, **45.6** on R@1 over CUB200-2001 and Car196 datasets, respectively. For the detail, please refer to [link](http://arxiv.xxx).

### Summary

* [The proposed framework](#the-proposed-framework)
* [Prerequisites](#prerequisites)
* [Preprocessing](#preprocessing)
* [Training](#training)
* [Testing](#testing)
* [Citation](#citation)
* [License](#license)
* [More information](#more-information)

### The proposed framework

![Illustration of the proposed framework](misc/best_model.png)

### Prerequisites
Expand Down Expand Up @@ -48,8 +61,8 @@ $ python train.py --dataset car196 --model_name CBSwR_Car196
The training scores will be printed every epoch.


### Pretrained models and Testing
In this repo, we include the pre-trained weight of CBSwR_CUB200 and CBSwR_Car196.
### Testing
In this repo, we include the pre-trained weight of CBSwR_CUB200 and CBSwR_Car196 models.

For `CBSwR_CUB200` pretrained model. Please download the [link](https://drive.google.com/file/d/1PsBSICUabh7NLNXoDZMvxj_6oGu_m7KO/view?usp=sharing) and move to `new_checkpoint`. The trained `CBSwR_CUB200` model can be tested in CUB200 validation set via:
```
Expand All @@ -71,4 +84,8 @@ Updating

### License

MIT License
MIT License

### More information

AIOZ AI Homepage: https://ai.aioz.io

0 comments on commit c9b8893

Please sign in to comment.