This is the case with the 00 sequence

[ztx991028]

[aofrancani]

This is definitely not the expected result. Could you give me more details about how you're running the code? You may have missed something. Have you downloaded the pre-trained weights from the DPT?

[ztx991028]

This is definitely not the expected result. Could you give me more details about how you're running the code? You may have missed something. Have you downloaded the pre-trained weights from the DPT?

[ztx991028]

The DPT model is

[ZOUrz]

@ztx991028 你好！你环境中的sklearn是不是高于作者要求的版本呢

[aofrancani]

I just ran the code here and got the following result:

You probably have an outdated package in your virtual environment. I tested scikit-learn with versions 0.23.0 and 1.1.1. Here are the other versions in my venv:

colorama==0.4.6
cycler==0.12.1
filelock==3.13.1
fonttools==4.55.3
fsspec==2024.2.0
Jinja2==3.1.3
joblib==1.4.2
kiwisolver==1.4.7
MarkupSafe==2.1.5
matplotlib==3.5.2
mpmath==1.3.0
networkx==3.0
numpy==1.23.1
opencv-python==4.5.2.54
packaging==24.2
pillow==10.4.0
pyparsing==3.1.4
python-dateutil==2.9.0.post0
scikit-learn==0.23.0
scipy==1.10.1
six==1.17.0
sklearn==0.0
sympy==1.13.1
threadpoolctl==3.5.0
timm==0.4.5
torch==2.4.1+cu124
torchvision==0.19.1+cu124
tqdm==4.64.0
typing_extensions==4.12.2