PyCenterNetDetector is not in the models registry

在mmdet中增加了新模型，没有注册。

打开mmdet/models/backbones/_init_.py。
将PyCenterNetDetector 添加进去。
修改完成后记着再安装一次

python setup.py install

# Copyright (c) OpenMMLab. All rights reserved.
from .atss import ATSS
from .autoassign import AutoAssign
from .base import BaseDetector
from .cascade_rcnn import CascadeRCNN
from .centernet import CenterNet
from .cornernet import CornerNet
from .deformable_detr import DeformableDETR
from .detr import DETR
from .fast_rcnn import FastRCNN
from .faster_rcnn import FasterRCNN
from .fcos import FCOS
from .fovea import FOVEA
from .fsaf import FSAF
from .gfl import GFL
from .grid_rcnn import GridRCNN
from .htc import HybridTaskCascade
from .kd_one_stage import KnowledgeDistillationSingleStageDetector
from .lad import LAD
from .mask2former import Mask2Former
from .mask_rcnn import MaskRCNN
from .mask_scoring_rcnn import MaskScoringRCNN
from .maskformer import MaskFormer
from .nasfcos import NASFCOS
from .paa import PAA
from .panoptic_fpn import PanopticFPN
from .panoptic_two_stage_segmentor import TwoStagePanopticSegmentor
from .point_rend import PointRend
from .queryinst import QueryInst
from .reppoints_detector import RepPointsDetector
from .retinanet import RetinaNet
from .rpn import RPN
from .scnet import SCNet
from .single_stage import SingleStageDetector
from .solo import SOLO
from .sparse_rcnn import SparseRCNN
from .tood import TOOD
from .trident_faster_rcnn import TridentFasterRCNN
from .two_stage import TwoStageDetector
from .vfnet import VFNet
from .yolact import YOLACT
from .yolo import YOLOV3
from .yolof import YOLOF
from .yolox import YOLOX
from .pycenternet_detector import PyCenterNetDetector__all__ = ['ATSS', 'BaseDetector', 'SingleStageDetector', 'TwoStageDetector', 'RPN','KnowledgeDistillationSingleStageDetector', 'FastRCNN', 'FasterRCNN','MaskRCNN', 'CascadeRCNN', 'HybridTaskCascade', 'RetinaNet', 'FCOS','GridRCNN', 'MaskScoringRCNN', 'RepPointsDetector', 'FOVEA', 'FSAF','NASFCOS', 'PointRend', 'GFL', 'CornerNet', 'PAA', 'YOLOV3', 'YOLACT','VFNet', 'DETR', 'TridentFasterRCNN', 'SparseRCNN', 'SCNet', 'SOLO','DeformableDETR', 'AutoAssign', 'YOLOF', 'CenterNet', 'YOLOX','TwoStagePanopticSegmentor', 'PanopticFPN', 'QueryInst', 'LAD', 'TOOD','MaskFormer', 'Mask2Former','PyCenterNetDetector'
]

PyCenterNetDetector is not in the models registry相关推荐

mmdetection 报错 KeyError: ‘xxx is not in the models registry‘
我的经验适用于直接下载解压别人源码后无法运行的,如果是自己改写的代码,还是检查一下__init__.py里有没有import xxx,或者 __all__里有没有注册xxx. 直入主题,先说解决方案: ...
SCRFD is not in the models registry
mmdet 解决方法: 在mmdet model detectors的__init__.py中手动加上你的类代码: from .atss import ATSS from .autoassign im ...
【Transformer】TNT: Transformer iN Transformer
文章目录一.背景二.动机三.方法 3.1 Transformer in Transformer 3.2 Network Architecture 四.效果五.代码论文链接:https://a ...
【python 9】python注册器
文章目录一.什么是 python 注册器二.python 注册器怎么用一.什么是 python 注册器 python 的 register 类可以理解为一个字典,这个字典里边会存储很多相同系列的 ...
用于MLOps的MLflow简介第1部分：Anaconda环境
在这三部分的博客中跟随了演示之后,您将能够: (After following along with the demos in this three part blog you will be abl ...
L1/L2/smooth_l1_loss/center_loss+Dice Loss+focal loss+各种IOU loss+kl散度
一．L1/L2/smooth_l1_loss/center_loss公式与代码 1.公式 L2公式: smooth_L1公式: 2.三种函数numpy代码实现 import numpy as np i ...
【语义分割】2021-PVT ICCV
文章目录 [语义分割]2021-PVT ICCV 1. 简介 1.1 简介 1.2 ViT在语义分割上存在的问题 2. 网络 2.1 整体架构 2.2 Patch embedding 2.3 Spat ...
MMDet踩坑与使用体会
1.报错module 'torch' has no attribute 'count_nonzero' 我的Pytorch版本是1.6.0,可以在pytorch网页上搜一下,count_nonzero ...
【点云3D目标检测】跑通CIA-SSD过程中的一些报错集锦
前言本文主要是介绍了在跑通CIA-SSD系列算法过程中遇到的一些bug,下面是对CIASSD算法的一个简单的介绍. CIA-SSD其基本思想是校准单步目标检测中分类和定位两个任务,提出Confide ...

PyCenterNetDetector is not in the models registry

PyCenterNetDetector is not in the models registry相关推荐

最新文章

热门文章