Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

Commit

Permalink
Merge branch 'master' into bugfix/instance_segmentation_example
Browse files Browse the repository at this point in the history
  • Loading branch information
mergify[bot] authored Dec 14, 2021
2 parents a5b71ea + 924aee8 commit 8b8b0b0
Showing 1 changed file with 12 additions and 9 deletions.
21 changes: 12 additions & 9 deletions flash/image/detection/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,7 @@ def from_coco(
**data_module_kwargs: Any,
) -> "ObjectDetectionData":
"""Creates a :class:`~flash.image.detection.data.ObjectDetectionData` object from the given data folders
and annotation files in the COCO format.
and annotation files in the `COCO JSON format <https://cocodataset.org/#format-data>`_.
Args:
train_folder: The folder containing the train data.
Expand Down Expand Up @@ -152,15 +152,16 @@ def from_voc(
**data_module_kwargs: Any,
) -> "ObjectDetectionData":
"""Creates a :class:`~flash.image.detection.data.ObjectDetectionData` object from the given data folders
and annotation files in the VOC format.
and annotation files in the `PASCAL VOC (Visual Obect Challenge)
<http://host.robots.ox.ac.uk/pascal/VOC/>`_ XML format.
Args:
train_folder: The folder containing the train data.
train_ann_file: The COCO format annotation file.
train_ann_file: The VOC format annotation file.
val_folder: The folder containing the validation data.
val_ann_file: The COCO format annotation file.
val_ann_file: The VOC format annotation file.
test_folder: The folder containing the test data.
test_ann_file: The COCO format annotation file.
test_ann_file: The VOC format annotation file.
predict_folder: The folder containing the predict data.
train_transform: The dictionary of transforms to use during training which maps
:class:`~flash.core.data.io.input_transform.InputTransform` hook names to callable transforms.
Expand Down Expand Up @@ -211,15 +212,17 @@ def from_via(
**data_module_kwargs: Any,
) -> "ObjectDetectionData":
"""Creates a :class:`~flash.image.detection.data.ObjectDetectionData` object from the given data folders
and annotation files in the VIA format.
and annotation files in the VIA (`VGG Image Annotator <https://www.robots.ox.ac.uk/~vgg/software/via/>`_)
`JSON format <https://gitlab.com/vgg/via/-/blob/via-3.x.y/via-3.x.y/CodeDoc.md#structure-of-via-project-
json-file>`_.
Args:
train_folder: The folder containing the train data.
train_ann_file: The COCO format annotation file.
train_ann_file: The VIA format annotation file.
val_folder: The folder containing the validation data.
val_ann_file: The COCO format annotation file.
val_ann_file: The VIA format annotation file.
test_folder: The folder containing the test data.
test_ann_file: The COCO format annotation file.
test_ann_file: The VIA format annotation file.
predict_folder: The folder containing the predict data.
train_transform: The dictionary of transforms to use during training which maps
:class:`~flash.core.data.io.input_transform.InputTransform` hook names to callable transforms.
Expand Down

0 comments on commit 8b8b0b0

Please sign in to comment.