Skip to content

Commit b3feff0

Browse files
committed
Cleanup python implementation from obsolette OMZ wrappers
1 parent 16818b0 commit b3feff0

17 files changed

+8
-3734
lines changed

model_api/python/README.md

Lines changed: 7 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -53,24 +53,19 @@ python -c "from openvino.model_zoo import model_api"
5353

5454
## Model API Wrappers
5555

56-
The Model API package provides model wrappers, which implement standardized preprocessing/postprocessing functions per "task type" and incapsulate model-specific logic for usage of different models in a unified manner inside the application.
56+
The Model API package provides model wrappers, which implement standardized preprocessing/postprocessing functions per "task type" and encapsulate model-specific logic for usage of different models in a unified manner inside the application.
5757

5858
The following tasks can be solved with wrappers usage:
5959

6060
| Task type | Model API wrappers |
6161
|----------------------------|--------------------|
62-
| Background Matting | <ul><li>`VideoBackgroundMatting`</li><li>`ImageMattingWithBackground`</li><li>`PortraitBackgroundMatting`</li></ul> |
6362
| Classification | <ul><li>`ClassificationModel`</li></ul> |
64-
| Deblurring | <ul><li>`Deblurring`</li></ul> |
65-
| Human Pose Estimation | <ul><li>`HpeAssociativeEmbedding`</li><li>`OpenPose`</li></ul> |
66-
| Instance Segmentation | <ul><li>`MaskRCNNModel`</li><li>`YolactModel`</li></ul> |
67-
| Monocular Depth Estimation | <ul><li> `MonoDepthModel`</li></ul> |
68-
| Named Entity Recognition | <ul><li>`BertNamedEntityRecognition`</li></ul> |
69-
| Object Detection | <ul><li>`CenterNet`</li><li>`DETR`</li><li>`CTPN`</li><li>`FaceBoxes`</li><li>`NanoDet`</li><li>`NanoDetPlus`</li><li>`RetinaFace`</li><li>`RetinaFacePyTorch`</li><li>`SSD`</li><li>`UltraLightweightFaceDetection`</li><li>`YOLO`</li><li>`YoloV3ONNX`</li><li>`YoloV4`</li><li>`YOLOF`</li><li>`YOLOX`</li></ul> |
70-
| Question Answering | <ul><li>`BertQuestionAnswering`</li></ul> |
71-
| Salient Object Detection | <ul><li>`SalientObjectDetectionModel`</li></ul> |
72-
| Semantic Segmentation | <ul><li>`SegmentationModel`</li></ul> |
73-
| Action Classification | <ul><li>`ActionClassificationModel`</li></ul> |
63+
| Human Pose Estimation | <ul><li>`KeypointDetectionModel`</li> |
64+
| Instance Segmentation | <ul><li>`MaskRCNNModel`</li></ul> |
65+
| Object Detection | <ul><li>`SSD`</li><li>`YOLO`</li><li>`YoloV3ONNX`</li><li>`YoloV4`</li><li>`YOLOF`</li><li>`YOLOX`</li></ul> |
66+
| Semantic Segmentation | <ul><li>`SegmentationModel`</li></ul> |
67+
| Visual Prompting | <ul><li>`SAMDecoder`</li><li>`SAMImageEncoder`</li></ul> |
68+
| Action Classification | <ul><li>`ActionClassificationModel`</li></ul> |
7469

7570
## Model API Adapters
7671

model_api/python/model_api/models/__init__.py

Lines changed: 1 addition & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -16,32 +16,15 @@
1616

1717
from .action_classification import ActionClassificationModel
1818
from .anomaly import AnomalyDetection
19-
from .background_matting import (
20-
ImageMattingWithBackground,
21-
PortraitBackgroundMatting,
22-
VideoBackgroundMatting,
23-
)
24-
from .bert import BertEmbedding, BertNamedEntityRecognition, BertQuestionAnswering
25-
from .centernet import CenterNet
2619
from .classification import ClassificationModel
27-
from .ctpn import CTPN
28-
from .deblurring import Deblurring
2920
from .detection_model import DetectionModel
30-
from .detr import DETR
31-
from .faceboxes import FaceBoxes
32-
from .hpe_associative_embedding import HpeAssociativeEmbedding
3321
from .image_model import ImageModel
34-
from .instance_segmentation import MaskRCNNModel, YolactModel
22+
from .instance_segmentation import MaskRCNNModel
3523
from .keypoint_detection import KeypointDetectionModel, TopDownKeypointDetectionPipeline
3624
from .model import Model
37-
from .monodepth import MonoDepthModel
38-
from .nanodet import NanoDet, NanoDetPlus
39-
from .open_pose import OpenPose
40-
from .retinaface import RetinaFace, RetinaFacePyTorch
4125
from .sam_models import SAMDecoder, SAMImageEncoder
4226
from .segmentation import SalientObjectDetectionModel, SegmentationModel
4327
from .ssd import SSD
44-
from .ultra_lightweight_face_detection import UltraLightweightFaceDetection
4528
from .utils import (
4629
AnomalyResult,
4730
ClassificationResult,
@@ -87,19 +70,10 @@
8770
"ActionClassificationModel",
8871
"AnomalyDetection",
8972
"AnomalyResult",
90-
"BertEmbedding",
91-
"BertNamedEntityRecognition",
92-
"BertQuestionAnswering",
93-
"CenterNet",
9473
"ClassificationModel",
9574
"Contour",
96-
"CTPN",
97-
"Deblurring",
9875
"DetectionModel",
9976
"DetectionWithLandmarks",
100-
"DETR",
101-
"FaceBoxes",
102-
"HpeAssociativeEmbedding",
10377
"ImageMattingWithBackground",
10478
"ImageModel",
10579
"ImageResultWithSoftPrediction",
@@ -113,20 +87,12 @@
11387
"TopDownKeypointDetectionPipeline",
11488
"MaskRCNNModel",
11589
"Model",
116-
"MonoDepthModel",
117-
"NanoDet",
118-
"NanoDetPlus",
119-
"OpenPose",
12090
"OutputTransform",
12191
"PortraitBackgroundMatting",
122-
"RetinaFace",
123-
"RetinaFacePyTorch",
12492
"SalientObjectDetectionModel",
12593
"SegmentationModel",
12694
"SSD",
127-
"UltraLightweightFaceDetection",
12895
"VideoBackgroundMatting",
129-
"YolactModel",
13096
"YOLO",
13197
"YoloV3ONNX",
13298
"YoloV4",

model_api/python/model_api/models/background_matting.py

Lines changed: 0 additions & 194 deletions
This file was deleted.

0 commit comments

Comments
 (0)