From 29df8001287dec5de25294e9ab5878ee55ee365d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E5=A7=9C=E5=A5=94?= Date: Wed, 23 Mar 2022 15:51:52 +0800 Subject: [PATCH] =?UTF-8?q?=E8=A7=84=E9=81=BFlvis=E5=AF=BC=E5=85=A5?= =?UTF-8?q?=E9=94=99=E8=AF=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../detectron2/data/datasets/builtin.py | 5 ++++- .../Cascade_RCNN/detectron2/data/datasets/builtin.py | 5 ++++- .../CascadedMaskRCNN/detectron2/data/datasets/builtin.py | 5 ++++- .../detection/RetinaNet/detectron2/data/datasets/builtin.py | 5 ++++- .../PointRend/detectron2/data/datasets/builtin.py | 5 ++++- .../detectron2/data/datasets/builtin.py | 5 ++++- .../detectron2/data/datasets/builtin.py | 5 ++++- .../detectron2/data/datasets/builtin.py | 5 ++++- .../detectron2/detectron2/data/datasets/builtin.py | 5 ++++- 9 files changed, 36 insertions(+), 9 deletions(-) diff --git a/PyTorch/built-in/cv/detection/Faster_Mask_RCNN_for_PyTorch/detectron2/data/datasets/builtin.py b/PyTorch/built-in/cv/detection/Faster_Mask_RCNN_for_PyTorch/detectron2/data/datasets/builtin.py index 28aba61851..d08b135914 100755 --- a/PyTorch/built-in/cv/detection/Faster_Mask_RCNN_for_PyTorch/detectron2/data/datasets/builtin.py +++ b/PyTorch/built-in/cv/detection/Faster_Mask_RCNN_for_PyTorch/detectron2/data/datasets/builtin.py @@ -234,6 +234,9 @@ def register_all_pascal_voc(root): # Register them all under "./datasets" _root = os.getenv("DETECTRON2_DATASETS", "datasets") register_all_coco(_root) -register_all_lvis(_root) + +# lvis dataset is not supported in this repo, please refer to origin repo if used.. +# register_all_lvis(_root) + register_all_cityscapes(_root) register_all_pascal_voc(_root) diff --git a/PyTorch/contrib/cv/detection/Cascade_RCNN/detectron2/data/datasets/builtin.py b/PyTorch/contrib/cv/detection/Cascade_RCNN/detectron2/data/datasets/builtin.py index 28aba61851..d08b135914 100644 --- a/PyTorch/contrib/cv/detection/Cascade_RCNN/detectron2/data/datasets/builtin.py +++ b/PyTorch/contrib/cv/detection/Cascade_RCNN/detectron2/data/datasets/builtin.py @@ -234,6 +234,9 @@ def register_all_pascal_voc(root): # Register them all under "./datasets" _root = os.getenv("DETECTRON2_DATASETS", "datasets") register_all_coco(_root) -register_all_lvis(_root) + +# lvis dataset is not supported in this repo, please refer to origin repo if used.. +# register_all_lvis(_root) + register_all_cityscapes(_root) register_all_pascal_voc(_root) diff --git a/PyTorch/contrib/cv/detection/CascadedMaskRCNN/detectron2/data/datasets/builtin.py b/PyTorch/contrib/cv/detection/CascadedMaskRCNN/detectron2/data/datasets/builtin.py index 28aba61851..d08b135914 100644 --- a/PyTorch/contrib/cv/detection/CascadedMaskRCNN/detectron2/data/datasets/builtin.py +++ b/PyTorch/contrib/cv/detection/CascadedMaskRCNN/detectron2/data/datasets/builtin.py @@ -234,6 +234,9 @@ def register_all_pascal_voc(root): # Register them all under "./datasets" _root = os.getenv("DETECTRON2_DATASETS", "datasets") register_all_coco(_root) -register_all_lvis(_root) + +# lvis dataset is not supported in this repo, please refer to origin repo if used.. +# register_all_lvis(_root) + register_all_cityscapes(_root) register_all_pascal_voc(_root) diff --git a/PyTorch/contrib/cv/detection/RetinaNet/detectron2/data/datasets/builtin.py b/PyTorch/contrib/cv/detection/RetinaNet/detectron2/data/datasets/builtin.py index 28aba61851..d08b135914 100644 --- a/PyTorch/contrib/cv/detection/RetinaNet/detectron2/data/datasets/builtin.py +++ b/PyTorch/contrib/cv/detection/RetinaNet/detectron2/data/datasets/builtin.py @@ -234,6 +234,9 @@ def register_all_pascal_voc(root): # Register them all under "./datasets" _root = os.getenv("DETECTRON2_DATASETS", "datasets") register_all_coco(_root) -register_all_lvis(_root) + +# lvis dataset is not supported in this repo, please refer to origin repo if used.. +# register_all_lvis(_root) + register_all_cityscapes(_root) register_all_pascal_voc(_root) diff --git a/PyTorch/contrib/cv/semantic_segmentation/PointRend/detectron2/data/datasets/builtin.py b/PyTorch/contrib/cv/semantic_segmentation/PointRend/detectron2/data/datasets/builtin.py index 067e9377fb..95c460321f 100644 --- a/PyTorch/contrib/cv/semantic_segmentation/PointRend/detectron2/data/datasets/builtin.py +++ b/PyTorch/contrib/cv/semantic_segmentation/PointRend/detectron2/data/datasets/builtin.py @@ -270,7 +270,10 @@ if __name__.endswith(".builtin"): # Assume pre-defined datasets live in `./datasets`. _root = os.getenv("DETECTRON2_DATASETS", "datasets") register_all_coco(_root) - register_all_lvis(_root) + + # lvis dataset is not supported in this repo, please refer to origin repo if used.. + # register_all_lvis(_root) + register_all_cityscapes(_root) register_all_cityscapes_panoptic(_root) register_all_pascal_voc(_root) diff --git a/PyTorch/dev/cv/detection/FasterRCNN-Resnet50-FPN_ID1552_for_PyTorch/detectron2/data/datasets/builtin.py b/PyTorch/dev/cv/detection/FasterRCNN-Resnet50-FPN_ID1552_for_PyTorch/detectron2/data/datasets/builtin.py index 28aba61851..d08b135914 100644 --- a/PyTorch/dev/cv/detection/FasterRCNN-Resnet50-FPN_ID1552_for_PyTorch/detectron2/data/datasets/builtin.py +++ b/PyTorch/dev/cv/detection/FasterRCNN-Resnet50-FPN_ID1552_for_PyTorch/detectron2/data/datasets/builtin.py @@ -234,6 +234,9 @@ def register_all_pascal_voc(root): # Register them all under "./datasets" _root = os.getenv("DETECTRON2_DATASETS", "datasets") register_all_coco(_root) -register_all_lvis(_root) + +# lvis dataset is not supported in this repo, please refer to origin repo if used.. +# register_all_lvis(_root) + register_all_cityscapes(_root) register_all_pascal_voc(_root) diff --git a/PyTorch/dev/cv/image_classification/FasterRCNN_ID0100_for_PyTorch/detectron2/data/datasets/builtin.py b/PyTorch/dev/cv/image_classification/FasterRCNN_ID0100_for_PyTorch/detectron2/data/datasets/builtin.py index 28aba61851..d08b135914 100644 --- a/PyTorch/dev/cv/image_classification/FasterRCNN_ID0100_for_PyTorch/detectron2/data/datasets/builtin.py +++ b/PyTorch/dev/cv/image_classification/FasterRCNN_ID0100_for_PyTorch/detectron2/data/datasets/builtin.py @@ -234,6 +234,9 @@ def register_all_pascal_voc(root): # Register them all under "./datasets" _root = os.getenv("DETECTRON2_DATASETS", "datasets") register_all_coco(_root) -register_all_lvis(_root) + +# lvis dataset is not supported in this repo, please refer to origin repo if used.. +# register_all_lvis(_root) + register_all_cityscapes(_root) register_all_pascal_voc(_root) diff --git a/PyTorch/dev/cv/image_classification/MaskRCNN_ID0101_for_PyTorch/detectron2/data/datasets/builtin.py b/PyTorch/dev/cv/image_classification/MaskRCNN_ID0101_for_PyTorch/detectron2/data/datasets/builtin.py index 28aba61851..d08b135914 100644 --- a/PyTorch/dev/cv/image_classification/MaskRCNN_ID0101_for_PyTorch/detectron2/data/datasets/builtin.py +++ b/PyTorch/dev/cv/image_classification/MaskRCNN_ID0101_for_PyTorch/detectron2/data/datasets/builtin.py @@ -234,6 +234,9 @@ def register_all_pascal_voc(root): # Register them all under "./datasets" _root = os.getenv("DETECTRON2_DATASETS", "datasets") register_all_coco(_root) -register_all_lvis(_root) + +# lvis dataset is not supported in this repo, please refer to origin repo if used.. +# register_all_lvis(_root) + register_all_cityscapes(_root) register_all_pascal_voc(_root) diff --git a/PyTorch/dev/cv/image_classification/SlowFast_ID0646_for_PyTorch/detectron2/detectron2/data/datasets/builtin.py b/PyTorch/dev/cv/image_classification/SlowFast_ID0646_for_PyTorch/detectron2/detectron2/data/datasets/builtin.py index 91b9de5562..bead28477c 100644 --- a/PyTorch/dev/cv/image_classification/SlowFast_ID0646_for_PyTorch/detectron2/detectron2/data/datasets/builtin.py +++ b/PyTorch/dev/cv/image_classification/SlowFast_ID0646_for_PyTorch/detectron2/detectron2/data/datasets/builtin.py @@ -290,7 +290,10 @@ if __name__.endswith(".builtin"): # Assume pre-defined datasets live in `./datasets`. _root = os.getenv("DETECTRON2_DATASETS", "datasets") register_all_coco(_root) - register_all_lvis(_root) + + # lvis dataset is not supported in this repo, please refer to origin repo if used.. + # register_all_lvis(_root) + register_all_cityscapes(_root) register_all_cityscapes_panoptic(_root) register_all_pascal_voc(_root) -- Gitee