rbanfield commited on
Commit
8aafd7c
1 Parent(s): 5ed15ed

Update opencv_zoo/models/__init__.py

Browse files
Files changed (1) hide show
  1. opencv_zoo/models/__init__.py +1 -33
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(DB)
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)