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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|