Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
chilo-ms authored Oct 4, 2024
1 parent 081679b commit e9ffdef
Showing 1 changed file with 35 additions and 6 deletions.
41 changes: 35 additions & 6 deletions quantization/image_classification/trt/resnet50/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ wget https://image-net.org/data/ILSVRC/2012/ILSVRC2012_devkit_t12.tar.gz --no-ch
```
Untar the tarballs to `val` and `ILSVRC2012_devkit_t12` folder separately.

The dataset layout should look like this:
The dataset layout should look like this: Following sample code expects this dataset layout.

```
|-- ILSVRC2012_devkit_t12
Expand All @@ -38,15 +38,44 @@ The dataset layout should look like this:
| | `-- make_hash.m
| `-- readme.txt
|-- meta.bin
|-- synset_words.txt
`-- val
|-- n01440764
| |-- ILSVRC2012_val_00000293.JPEG
| |-- ILSVRC2012_val_00002138.JPEG
| |-- ILSVRC2012_val_00003014.JPEG
|-- ILSVRC2012_val_00000001.JPEG
|-- ILSVRC2012_val_00000002.JPEG
|-- ILSVRC2012_val_00000003.JPEG
...
```
Note: If the data in `val` folder is not grouped by class, please run following command to reconstruct the layout

However, if you are using ImageNet, then please run following command to reconstruct the layout to be grouped by class.
```shell
cd val/
wget -qO- https://raw.githubusercontent.com/soumith/imagenetloader.torch/master/valprep.sh | bash
```
```
|-- ILSVRC2012_devkit_t12
| |-- COPYING
| |-- data
| | |-- ILSVRC2012_validation_ground_truth.txt
| | `-- meta.mat
| |-- evaluation
| | |-- VOCreadrecxml.m
| | |-- VOCreadxml.m
| | |-- VOCxml2struct.m
| | |-- compute_overlap.m
| | |-- demo.val.pred.det.txt
| | |-- demo.val.pred.txt
| | |-- demo_eval.m
| | |-- eval_flat.m
| | |-- eval_localization_flat.m
| | |-- get_class2node.m
| | `-- make_hash.m
| `-- readme.txt
|-- meta.bin
`-- val
|-- n01440764
| |-- ILSVRC2012_val_00000293.JPEG
| |-- ILSVRC2012_val_00002138.JPEG
| |-- ILSVRC2012_val_00003014.JPEG
...
```

0 comments on commit e9ffdef

Please sign in to comment.