--- a +++ b/features/rgb_gr.py @@ -0,0 +1,49 @@ +import os +import cv2 + +# data_path = 'test/' +# out_path = 'res/' +# filelist = os.listdir(data_path) + +def trans(img): + ht = img.shape[0] + wd = img.shape[1] + for i in range(0,ht): + for j in range(0,wd): + B = img[i][j][0] + G = img[i][j][1] + R = img[i][j][2] + if B <= 51: + img[i][j][0] = 255 + elif B <= 102: + img[i][j][0] = 255 - (B-51)*5 + elif B <= 153: + img[i][j][0] = 0 + else : img[i][j][0] = 0 + + if G <= 51: + img[i][j][1] = G*5 + elif G <= 102: + img[i][j][1] = 255 + elif G <= 153: + img[i][j][1] = 255 + elif G <= 204: + img[i][j][1] = 255 - int(128.0*(G-153.0)/51.0 + 0.5) + else : img[i][j][1] = 127 - int(127.0*(G-204.0)/51.0 + 0.5) + + if R <= 51: + img[i][j][2] = 0 + elif R <= 102: + img[i][j][2] = 0 + elif R <= 153: + img[i][j][2] = (R-102)*5 + elif G <= 204: + img[i][j][2] = 255 + else : img[i][j][2] = 255 + # cv2.imwrite(os.path.join(out_path,file_name),img) + return img + + +# for onefile in filelist: +# img = cv2.imread(data_path + onefile) +# trans(onefile, img) \ No newline at end of file