Open gustavofuhr opened 1 week ago
I have the same problem.
Hello, @gustavofuhr
I solved this issue in #7 by modifying mmdet_custom/models/co_detr_heads/init.py as shown below:
# @Time : 29/08/2023 13:35
# @Author : BubblyYi
# @FileName: __init__.py
# @Software: PyCharm
from .co_atss_head import CoATSSHead
from .detr_head import DETRHead_Fix
from .co_deformable_detr_head import CoDeformDETRHead
from .co_dino_head import CoDINOCoodHead, CoDINOHead
from .query_denoising import build_dn_generator
from .transformer import (BaseTransformerLayer_CP, CoDeformableDetrTransformer,
CoDeformableDetrTransformerDecoder,
CoDinoTransformer, DinoTransformerDecoder)
__all__ = [
'CoATSSHead', 'CoDeformDETRHead', 'CoDINOCoodHead', 'CoDINOHead',
'build_dn_generator', 'BaseTransformerLayer_CP',
'CoDeformableDetrTransformer', 'CoDeformableDetrTransformerDecoder',
'CoDinoTransformer', 'DinoTransformerDecoder', 'DETRHead_Fix'
]
When I try to import
mmdet_custom
I got the following error:Also, if possible, can you provide how to run inference without slurm?