Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 12:10:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix matplotlib.cm compatible issue

Browse Source
This commit is contained in:
mulin.lyh
2024-07-15 16:28:40 +08:00
parent e21c91ae35
commit de782a5a9e
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modelscope/models/cv/video_depth_estimation/utils/depth.py
View File

@@ -3,7 +3,9 @@
import numpy as np
import torch
import torchvision.transforms as transforms
from matplotlib.cm import get_cmap
# from matplotlib.cm import get_cmap
# compatible with matplotlib 3.9.0
from matplotlib.pyplot import get_cmap
from modelscope.models.cv.video_depth_estimation.utils.image import (
flip_lr, gradient_x, gradient_y, interpolate_image, load_image)