opencv-python의 8bit 이미지 읽기 및 저장

선언:
최근cityscapes 데이터 집합의 데이터 사전 처리에 바빴는데, 그 label 그림 집합은 모두 8bit 기반의 단일 채널 데이터라는 것을 발견했다.이전에 데이터 처리를 할 때 이 점을 주의하지 않았기 때문에 생성된 label 도면집은 예측된 결과 도면집과miou계산을 할 때 오류가 발생했습니다.
코드 구현:
from cv2 import cv2 as cv2
import numpy as np

# img = np.zeros((1024,2048),np.int8)  #     8bit   
# img = cv2.imwrite("./1.png",img)     #          8bit   

# img = cv2.imread("./frankfurt_000000_000294_gtFine_labelIds.png",0)  #          
# img = cv2.imwrite("./2.png",img)                                     #          

img1 = cv2.imread("./frankfurt_000000_000294_gtFine_labelIds.png",0)
img2 = cv2.imread("./2.png",0)
# print(img1.shape[0])  # hight
for i in range(img1.shape[0]):
    for j in range(img1.shape[1]):
        if img1[i][j] != img2[i][j]:
            print("bug!")
        else:
            print("The point of "+str(i)+"_"+str(j)+" row:"+"is ok!")     #            

좋은 웹페이지 즐겨찾기