|
16 | 16 |
|
17 | 17 | from .action_classification import ActionClassificationModel |
18 | 18 | 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 |
26 | 19 | from .classification import ClassificationModel |
27 | | -from .ctpn import CTPN |
28 | | -from .deblurring import Deblurring |
29 | 20 | from .detection_model import DetectionModel |
30 | | -from .detr import DETR |
31 | | -from .faceboxes import FaceBoxes |
32 | | -from .hpe_associative_embedding import HpeAssociativeEmbedding |
33 | 21 | from .image_model import ImageModel |
34 | | -from .instance_segmentation import MaskRCNNModel, YolactModel |
| 22 | +from .instance_segmentation import MaskRCNNModel |
35 | 23 | from .keypoint_detection import KeypointDetectionModel, TopDownKeypointDetectionPipeline |
36 | 24 | 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 |
41 | 25 | from .sam_models import SAMDecoder, SAMImageEncoder |
42 | 26 | from .segmentation import SalientObjectDetectionModel, SegmentationModel |
43 | 27 | from .ssd import SSD |
44 | | -from .ultra_lightweight_face_detection import UltraLightweightFaceDetection |
45 | 28 | from .utils import ( |
46 | 29 | AnomalyResult, |
47 | 30 | ClassificationResult, |
|
87 | 70 | "ActionClassificationModel", |
88 | 71 | "AnomalyDetection", |
89 | 72 | "AnomalyResult", |
90 | | - "BertEmbedding", |
91 | | - "BertNamedEntityRecognition", |
92 | | - "BertQuestionAnswering", |
93 | | - "CenterNet", |
94 | 73 | "ClassificationModel", |
95 | 74 | "Contour", |
96 | | - "CTPN", |
97 | | - "Deblurring", |
98 | 75 | "DetectionModel", |
99 | 76 | "DetectionWithLandmarks", |
100 | | - "DETR", |
101 | | - "FaceBoxes", |
102 | | - "HpeAssociativeEmbedding", |
103 | 77 | "ImageMattingWithBackground", |
104 | 78 | "ImageModel", |
105 | 79 | "ImageResultWithSoftPrediction", |
|
113 | 87 | "TopDownKeypointDetectionPipeline", |
114 | 88 | "MaskRCNNModel", |
115 | 89 | "Model", |
116 | | - "MonoDepthModel", |
117 | | - "NanoDet", |
118 | | - "NanoDetPlus", |
119 | | - "OpenPose", |
120 | 90 | "OutputTransform", |
121 | 91 | "PortraitBackgroundMatting", |
122 | | - "RetinaFace", |
123 | | - "RetinaFacePyTorch", |
124 | 92 | "SalientObjectDetectionModel", |
125 | 93 | "SegmentationModel", |
126 | 94 | "SSD", |
127 | | - "UltraLightweightFaceDetection", |
128 | 95 | "VideoBackgroundMatting", |
129 | | - "YolactModel", |
130 | 96 | "YOLO", |
131 | 97 | "YoloV3ONNX", |
132 | 98 | "YoloV4", |
|
0 commit comments