Skip to content

Commit

Permalink
Add ImageNet to documentation
Browse files Browse the repository at this point in the history
  • Loading branch information
yasakova-anastasia committed Oct 30, 2020
1 parent d172497 commit a38b10e
Show file tree
Hide file tree
Showing 2 changed files with 33 additions and 1 deletion.
33 changes: 33 additions & 0 deletions cvat/apps/dataset_manager/formats/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
- [PASCAL VOC and mask](#voc)
- [YOLO](#yolo)
- [TF detection API](#tfrecord)
- [ImageNet](#imagenet)

## How to add a new annotation format support<a id="how-to-add"></a>

Expand Down Expand Up @@ -802,3 +803,35 @@ taskname.zip/
```

- supported annotations: Rectangles, Polygons, Masks (as polygons)

### [ImageNet](http://www.image-net.org)<a id="imagenet" />

#### ImageNet Dumper

Downloaded file: a zip archive of the following structure:

```bash
# if we save images:
taskname.zip/
└── label1/
├── label1_image1.jpg
└── label1_image2.jpg
└── label2/
├── label2_image1.jpg
├── label2_image3.jpg
└── label2_image4.jpg

# if we keep only annotation:
taskname.zip/
└── <any_subset_name>.txt
└── synsets.txt

```

- supported annotations: Labels

#### ImageNet Loader

Uploaded file: a zip archive of the structure above

- supported annotations: Labels
1 change: 0 additions & 1 deletion cvat/apps/engine/tests/test_rest_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -3358,7 +3358,6 @@ def _get_initial_annotation(annotation_format):
elif annotation_format == "ImageNet 1.0":
annotations["tags"] = tags_wo_attrs


else:
raise Exception("Unknown format {}".format(annotation_format))

Expand Down

0 comments on commit a38b10e

Please sign in to comment.