今天学习ctf遇到了从rgb数据构造图片的题，顺便学一下python的Image模块吧 具体文档

属性

# 图片大小 width, height = im.size # 文件名包含路径 print(im.filename) # 文件类型png/jpg/bmt print(im.format) # 模式rgb/rgba/cmyk print(im.mode) # im.width # im.height

方法

from PIL import Image from io import BytesIO # 对于彩色图像，不管其图像格式是PNG，还是BMP，或者JPG，在PIL中，使用Image模块的open()函数打开后，返回的图像对象的模式都是“RGB”。 # 而对于灰度图像，不管其图像格式是PNG，还是BMP，或者JPG，打开后，其模式为“L” im = Image.open("misc/tp/txt.png") # 从二进制打开图片 im = Image.open(BytesIO("misc/tp/txt.bin")) # 构造图像 width, height = 128, 128 color = (0, 235, 255) im = Image.new('RGB', (width, height), color) #RGB im = Image.new('1', (width, height),1) #1是白0是黑 im = Image.new('L', (width, height), 120) # 128位灰度图 im = Image.new('RGBA', (width, height), (255,255,128,128)) # RGBA # 得到i,j位置的像素值 im.getpixel((i, j)) # 通过像素修改图像 x, y = (1, 2) color = (0, 0, 0) im.putpixel((x, y), color) # 展示图片 im.show() # 保存图片 im.save("misc/tp/txt.png") #得到png,jpg,bmp格式图片的每一帧的RGB width,height=im.size pix = [im.getpixel((i, j)) for i in range(width) for j in range(height)] print(pix) # 旋转45度 im = im.rotate(45) # 等比缩放 size = 128, 128 im.thumbnail(size) # 按给定值缩放 im = im.resize((120, 12)) # 图片类型转换 im.convert("1") im.convert("L") im.convert("RGB") 类型转换这个博客写的挺好的: https://blog.csdn.net/icamera0/article/details/50843172 # split分割RGB时，按照RGB的三原色，返回三个图像。其他类似 imm=im.split() imm[0].show() # 裁剪矩形 left,top,right,bottm=(1,2,100,100) im=im.crop((left,top,right,bottm)) # 转换成比特流 im.tobytes()

ctf 练习

练习一下从RGB数据恢复成图片 https://pan.baidu.com/s/1KP9BiRn8EaSnB9s9iSEJlg 密码: awic

答案

from PIL import Image file_txt = open("misc/tp/pic.txt", "r") str_pix = file_txt.read().split("\n") width, height = 0, 0 # Get the width and height of the image for i in str_pix: if int(i.split(" ")[0]) > width: width = int(i.split(" ")[0]) if int(i.split(" ")[1]) > height: height = int(i.split(" ")[1]) # Output the width and height print ("Width: ", width, " Height: ", height) img = Image.new("RGB", (width,height)) # Write pixel to new image file: # pixTuple(R, G, B, A) # x -> width, y -> height, R -> red, G -> green, B -> blue, A -> transparency for i in str_pix: x = int(i.split(" ")[0]) y = int(i.split(" ")[1]) R = int(i.split(" ")[2]) G = int(i.split(" ")[3]) B = int(i.split(" ")[4]) pixTuple = (R, G, B, 0) img.putpixel((x-1, y-1), pixTuple) # Save the image file img.save("output.png")