search

LiheYoung / Depth-Anything

[CVPR 2024] Depth Anything: Unleashing the Power of Large-Scale Unlabeled Data. Foundation Model for Monocular Depth Estimation
https://depth-anything.github.io
Apache License 2.0
6.99k stars 537 forks source link

mmseg::model registry error #219

Closed Bion7232 closed 3 months ago

Bion7232 commented 3 months ago

Hi, I tried to test segmentation by following README, but i have some error like this, KeyError: 'DINOv2 is not in the mmseg::model registry. Please check whether the value ofDINOv2is correct or it was registered as expected. More details can be found at https://mmengine.readthedocs.io/en/latest/advanced_tutorials/config.html#import-the-custom-module' how can i solve it? Is it correct to make a 'checkpoints' folder in the 'mmsegmentation' folder and put it in it? i have already added dinov2.py and changed 'Depth-Anything/mmsegmentation/mmseg/models/backbones/init.py' like this, ' from .dinov2 import DINOv2

all = [ 'ResNet', 'ResNetV1c', 'ResNetV1d', 'ResNeXt', 'HRNet', 'FastSCNN', 'ResNeSt', 'MobileNetV2', 'UNet', 'CGNet', 'MobileNetV3', 'VisionTransformer', 'SwinTransformer', 'MixVisionTransformer', 'BiSeNetV1', 'BiSeNetV2', 'ICNet', 'TIMMBackbone', 'ERFNet', 'PCPVT', 'SVT', 'STDCNet', 'STDCContextPathNet', 'BEiT', 'MAE', 'PIDNet', 'MSCAN', 'DDRNet', 'VPD', 'DINOv2' ] '

Bion7232 commented 3 months ago

I solved this problem.

In Depth-Anything/mmsegmentation,

python setup.py install
pip install -v -e .