Skip to content

Commit 549e4da

Browse files
authored
Update model_zoo.md
1 parent a20ba28 commit 549e4da

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

doc/model_zoo.md

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,16 @@ This document contains links to some pre-trained models. See [here](https://gith
33
for how to load and visualize a pre-trained network. Prior to evaluating make sure you have downloaded the appropriate train
44
and test data as outlined in the tutorial.
55

6-
1. [Caterpillar](https://data.csail.mit.edu/labelfusion/pdccompressed/trained_models/stable/caterpillar_standard_params_3.tar.gz)
7-
2. [Shoes Class Consistent](https://data.csail.mit.edu/labelfusion/pdccompressed/trained_models/stable/shoes_consistent_M_background_0.500_3.tar.gz)
8-
3. [Hats Class Consistent](https://data.csail.mit.edu/labelfusion/pdccompressed/trained_models/stable/hats_consistent_M_background_0.500_3.tar.gz)
6+
Inside of the folders for each of these pre-trained networks you will find:
7+
- a `dataset.yaml`, which documents which dataset configuration was used
8+
- a `training.yaml`, which documents which training configuration was used
9+
- an `analysis.yaml`, which contains quantitative analysis and is plottable using the `evaluation_quantitative_tutorial.ipynb` notebook
10+
- the `tensorboard`subdirectory containing all info needed to plot the results of the training
11+
- a handful of other files, including network weights at various iterations
12+
13+
14+
Models
15+
16+
1. [Caterpillar](https://data.csail.mit.edu/labelfusion/pdccompressed/trained_models/stable/caterpillar_standard_params_3.tar.gz) -- note that this network was trained with M_background = 2.0 and M_masked = 0.5, this means there is a larger contrastive loss with the background than there is with the rest of the object. While we didn't find this to improve quantitative visualizations, it can improve qualitative visualizations of the descriptor space.
17+
2. [Shoes Class Consistent](https://data.csail.mit.edu/labelfusion/pdccompressed/trained_models/stable/shoes_consistent_M_background_0.500_3.tar.gz)
18+
3. [Hats Class Consistent](https://data.csail.mit.edu/labelfusion/pdccompressed/trained_models/stable/hats_consistent_M_background_0.500_3.tar.gz)

0 commit comments

Comments
 (0)