Spaces:
Runtime error
Runtime error
Update opencv_zoo/models/__init__.py
Browse files
opencv_zoo/models/__init__.py
CHANGED
@@ -3,23 +3,7 @@ import glob
|
|
3 |
import os
|
4 |
|
5 |
from .face_detection_yunet.yunet import YuNet
|
6 |
-
from .text_detection_db.db import DB
|
7 |
-
from .text_recognition_crnn.crnn import CRNN
|
8 |
-
from .face_recognition_sface.sface import SFace
|
9 |
-
from .image_classification_ppresnet.ppresnet import PPResNet
|
10 |
-
from .human_segmentation_pphumanseg.pphumanseg import PPHumanSeg
|
11 |
-
from .person_detection_mediapipe.mp_persondet import MPPersonDet
|
12 |
-
from .pose_estimation_mediapipe.mp_pose import MPPose
|
13 |
-
from .qrcode_wechatqrcode.wechatqrcode import WeChatQRCode
|
14 |
-
from .object_tracking_dasiamrpn.dasiamrpn import DaSiamRPN
|
15 |
-
from .person_reid_youtureid.youtureid import YoutuReID
|
16 |
-
from .image_classification_mobilenet.mobilenet import MobileNet
|
17 |
-
from .palm_detection_mediapipe.mp_palmdet import MPPalmDet
|
18 |
-
from .handpose_estimation_mediapipe.mp_handpose import MPHandPose
|
19 |
from .license_plate_detection_yunet.lpd_yunet import LPD_YuNet
|
20 |
-
from .object_detection_nanodet.nanodet import NanoDet
|
21 |
-
from .object_detection_yolox.yolox import YoloX
|
22 |
-
from .facial_expression_recognition.facial_fer_model import FacialExpressionRecog
|
23 |
|
24 |
class ModuleRegistery:
|
25 |
def __init__(self, name):
|
@@ -77,20 +61,4 @@ class ModuleRegistery:
|
|
77 |
|
78 |
MODELS = ModuleRegistery('Models')
|
79 |
MODELS.register(YuNet)
|
80 |
-
MODELS.register(
|
81 |
-
MODELS.register(CRNN)
|
82 |
-
MODELS.register(SFace)
|
83 |
-
MODELS.register(PPResNet)
|
84 |
-
MODELS.register(PPHumanSeg)
|
85 |
-
MODELS.register(MPPersonDet)
|
86 |
-
MODELS.register(MPPose)
|
87 |
-
MODELS.register(WeChatQRCode)
|
88 |
-
MODELS.register(DaSiamRPN)
|
89 |
-
MODELS.register(YoutuReID)
|
90 |
-
MODELS.register(MobileNet)
|
91 |
-
MODELS.register(MPPalmDet)
|
92 |
-
MODELS.register(MPHandPose)
|
93 |
-
MODELS.register(LPD_YuNet)
|
94 |
-
MODELS.register(NanoDet)
|
95 |
-
MODELS.register(YoloX)
|
96 |
-
MODELS.register(FacialExpressionRecog)
|
|
|
3 |
import os
|
4 |
|
5 |
from .face_detection_yunet.yunet import YuNet
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
from .license_plate_detection_yunet.lpd_yunet import LPD_YuNet
|
|
|
|
|
|
|
7 |
|
8 |
class ModuleRegistery:
|
9 |
def __init__(self, name):
|
|
|
61 |
|
62 |
MODELS = ModuleRegistery('Models')
|
63 |
MODELS.register(YuNet)
|
64 |
+
MODELS.register(LPD_YuNet)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|