最近用mmsegmentation的时候,需要将label转为HxW的格式。
代码:
import os
import cv2
import numpy as np
from matplotlib import pyplot as plt
%matplotlib inline
imglist = os.listdir(os.getcwd())
for imgname in imglist:
img = cv2
最近用mmsegmentation的时候,需要将label转为HxW的格式。
代码:
import os
import cv2
import numpy as np
from matplotlib import pyplot as plt
%matplotlib inline
imglist = os.listdir(os.getcwd())
for imgname in imglist:
img = cv2