mblt-vision-python 0.0.0__py3-none-any.whl

mblt_vision/datasets/__init__.py

@@ -0,0 +1,15 @@
+"""YAML-backed definitions for vision validation datasets."""
+
+from .registry import (
+    get_dataset_category_ids,
+    get_dataset_class_names,
+    get_dataset_config,
+    get_dataset_config_for_task,
+)
+
+__all__ = [
+    "get_dataset_category_ids",
+    "get_dataset_class_names",
+    "get_dataset_config",
+    "get_dataset_config_for_task",
+]

mblt_vision/datasets/ade20k.yaml

@@ -0,0 +1,164 @@
+# ADE20K validation dataset. Organized paths are relative to `path`.
+name: ade20k
+path: ~/.mblt_model_zoo/datasets/ADEChallengeData2016
+val: images
+tasks:
+  - semantic_segmentation
+names:
+  0: wall
+  1: building
+  2: sky
+  3: floor
+  4: tree
+  5: ceiling
+  6: road
+  7: bed
+  8: windowpane
+  9: grass
+  10: cabinet
+  11: sidewalk
+  12: person
+  13: earth
+  14: door
+  15: table
+  16: mountain
+  17: plant
+  18: curtain
+  19: chair
+  20: car
+  21: water
+  22: painting
+  23: sofa
+  24: shelf
+  25: house
+  26: sea
+  27: mirror
+  28: rug
+  29: field
+  30: armchair
+  31: seat
+  32: fence
+  33: desk
+  34: rock
+  35: wardrobe
+  36: lamp
+  37: bathtub
+  38: railing
+  39: cushion
+  40: base
+  41: box
+  42: column
+  43: signboard
+  44: chest of drawers
+  45: counter
+  46: sand
+  47: sink
+  48: skyscraper
+  49: fireplace
+  50: refrigerator
+  51: grandstand
+  52: path
+  53: stairs
+  54: runway
+  55: case
+  56: pool table
+  57: pillow
+  58: screen door
+  59: stairway
+  60: river
+  61: bridge
+  62: bookcase
+  63: blind
+  64: coffee table
+  65: toilet
+  66: flower
+  67: book
+  68: hill
+  69: bench
+  70: countertop
+  71: stove
+  72: palm
+  73: kitchen island
+  74: computer
+  75: swivel chair
+  76: boat
+  77: bar
+  78: arcade machine
+  79: hovel
+  80: bus
+  81: towel
+  82: light
+  83: truck
+  84: tower
+  85: chandelier
+  86: awning
+  87: streetlight
+  88: booth
+  89: television receiver
+  90: airplane
+  91: dirt track
+  92: apparel
+  93: pole
+  94: land
+  95: bannister
+  96: escalator
+  97: ottoman
+  98: bottle
+  99: buffet
+  100: poster
+  101: stage
+  102: van
+  103: ship
+  104: fountain
+  105: conveyor belt
+  106: canopy
+  107: washer
+  108: plaything
+  109: swimming pool
+  110: stool
+  111: barrel
+  112: basket
+  113: waterfall
+  114: tent
+  115: bag
+  116: minibike
+  117: cradle
+  118: oven
+  119: ball
+  120: food
+  121: step
+  122: tank
+  123: trade name
+  124: microwave
+  125: pot
+  126: animal
+  127: bicycle
+  128: lake
+  129: dishwasher
+  130: screen
+  131: blanket
+  132: sculpture
+  133: hood
+  134: sconce
+  135: vase
+  136: traffic light
+  137: tray
+  138: ashcan
+  139: fan
+  140: pier
+  141: crt screen
+  142: plate
+  143: monitor
+  144: bulletin board
+  145: shower
+  146: radiator
+  147: glass
+  148: clock
+  149: flag
+download:
+  url: https://data.csail.mit.edu/places/ADEchallenge/ADEChallengeData2016.zip
+  sha256: 7ff1be44964418441f542a7cc1e1a650e7dc0fc275f5d23252bc9bbdbc977b29
+nc: 150
+source_ignore_label: 0
+ignore_label: 255
+label_offset: 1

mblt_vision/datasets/cityscapes.yaml

@@ -0,0 +1,54 @@
+# Cityscapes validation taxonomy from the official train/validation/test archives.
+# Organized paths are relative to `path`.
+name: cityscapes
+path: ~/.mblt_model_zoo/datasets/cityscapes
+val: images
+tasks:
+  - semantic_segmentation
+nc: 19
+names:
+  0: road
+  1: sidewalk
+  2: building
+  3: wall
+  4: fence
+  5: pole
+  6: traffic light
+  7: traffic sign
+  8: vegetation
+  9: terrain
+  10: sky
+  11: person
+  12: rider
+  13: car
+  14: truck
+  15: bus
+  16: train
+  17: motorcycle
+  18: bicycle
+category_ids: [7, 8, 11, 12, 13, 17, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 31, 32, 33]
+palette:
+  0: [128, 64, 128]
+  1: [244, 35, 232]
+  2: [70, 70, 70]
+  3: [102, 102, 156]
+  4: [190, 153, 153]
+  5: [153, 153, 153]
+  6: [250, 170, 30]
+  7: [220, 220, 0]
+  8: [107, 142, 35]
+  9: [152, 251, 152]
+  10: [70, 130, 180]
+  11: [220, 20, 60]
+  12: [255, 0, 0]
+  13: [0, 0, 142]
+  14: [0, 0, 70]
+  15: [0, 60, 100]
+  16: [0, 80, 100]
+  17: [0, 0, 230]
+  18: [119, 11, 32]
+download:
+  type: cityscapes
+  source: https://github.com/mcordts/cityscapesScripts
+  images_archive: leftImg8bit_trainvaltest.zip
+  annotations_archive: gtFine_trainvaltest.zip

mblt_vision/datasets/coco.yaml

@@ -0,0 +1,178 @@
+# COCO 2017 validation dataset. Paths are relative to `path` unless absolute.
+name: coco
+path: ~/.mblt_model_zoo/datasets/coco
+tasks:
+  - object_detection
+  - instance_segmentation
+  - pose_estimation
+val: val2017
+download:
+  # The official COCO object-store hostname has no valid TLS certificate. Use a
+  # pinned HTTPS mirror and verify each archive digest before extraction.
+  images: https://huggingface.co/datasets/pcuenq/coco-2017-mirror/resolve/5200d2cffae8121713bec767b4693bdafc0eeb0b/val2017.zip
+  images_sha256: 4f7e2ccb2866ec5041993c9cf2a952bbed69647b115d0f74da7ce8f4bef82f05
+  annotations: https://huggingface.co/datasets/pcuenq/coco-2017-mirror/resolve/5200d2cffae8121713bec767b4693bdafc0eeb0b/annotations_trainval2017.zip
+  annotations_sha256: 113a836d90195ee1f884e704da6304dfaaecff1f023f49b6ca93c4aaae470268
+
+names:
+  0: person
+  1: bicycle
+  2: car
+  3: motorcycle
+  4: airplane
+  5: bus
+  6: train
+  7: truck
+  8: boat
+  9: traffic light
+  10: fire hydrant
+  11: stop sign
+  12: parking meter
+  13: bench
+  14: bird
+  15: cat
+  16: dog
+  17: horse
+  18: sheep
+  19: cow
+  20: elephant
+  21: bear
+  22: zebra
+  23: giraffe
+  24: backpack
+  25: umbrella
+  26: handbag
+  27: tie
+  28: suitcase
+  29: frisbee
+  30: skis
+  31: snowboard
+  32: sports ball
+  33: kite
+  34: baseball bat
+  35: baseball glove
+  36: skateboard
+  37: surfboard
+  38: tennis racket
+  39: bottle
+  40: wine glass
+  41: cup
+  42: fork
+  43: knife
+  44: spoon
+  45: bowl
+  46: banana
+  47: apple
+  48: sandwich
+  49: orange
+  50: broccoli
+  51: carrot
+  52: hot dog
+  53: pizza
+  54: donut
+  55: cake
+  56: chair
+  57: couch
+  58: potted plant
+  59: bed
+  60: dining table
+  61: toilet
+  62: tv
+  63: laptop
+  64: mouse
+  65: remote
+  66: keyboard
+  67: cell phone
+  68: microwave
+  69: oven
+  70: toaster
+  71: sink
+  72: refrigerator
+  73: book
+  74: clock
+  75: vase
+  76: scissors
+  77: teddy bear
+  78: hair drier
+  79: toothbrush
+category_ids:
+- 1
+- 2
+- 3
+- 4
+- 5
+- 6
+- 7
+- 8
+- 9
+- 10
+- 11
+- 13
+- 14
+- 15
+- 16
+- 17
+- 18
+- 19
+- 20
+- 21
+- 22
+- 23
+- 24
+- 25
+- 27
+- 28
+- 31
+- 32
+- 33
+- 34
+- 35
+- 36
+- 37
+- 38
+- 39
+- 40
+- 41
+- 42
+- 43
+- 44
+- 46
+- 47
+- 48
+- 49
+- 50
+- 51
+- 52
+- 53
+- 54
+- 55
+- 56
+- 57
+- 58
+- 59
+- 60
+- 61
+- 62
+- 63
+- 64
+- 65
+- 67
+- 70
+- 72
+- 73
+- 74
+- 75
+- 76
+- 77
+- 78
+- 79
+- 80
+- 81
+- 82
+- 84
+- 85
+- 86
+- 87
+- 88
+- 89
+- 90

mblt_vision/datasets/dotav1.yaml

@@ -0,0 +1,28 @@
+# DOTAv1 validation dataset. Paths are relative to `path` unless absolute.
+name: dotav1
+path: ~/.mblt_model_zoo/datasets/dotav1
+tasks:
+  - obb
+val: images
+labels: labels/val_original
+names:
+  0: plane
+  1: ship
+  2: storage-tank
+  3: baseball-diamond
+  4: tennis-court
+  5: basketball-court
+  6: ground-track-field
+  7: harbor
+  8: bridge
+  9: large-vehicle
+  10: small-vehicle
+  11: helicopter
+  12: roundabout
+  13: soccer-ball-field
+  14: swimming-pool
+download:
+  type: google_drive_folder
+  url: https://drive.google.com/drive/folders/1n5w45suVOyaqY84hltJhIZdtVFD9B224
+  images_archive: images/part1.zip
+  labels_archive: labelTxt-v1.0/labelTxt.zip