MaixPy 图像基础操作

2024-04-03

更新历史

日期	版本	作者	更新内容
2024-04-03	1.0.0	neucrack	初版文档

简介

视觉应用中图像占据非常重要的位置，不管是图片还是视频，因为视频本质上就是一帧帧的图像，所以图像处理是视觉应用的基础。

API 文档

本文介绍常用方法，更多 API 参考 maix.image 模块的文档。

图像格式

MaixPy 提供基础图像模块image，里面最重要的就是image.Image类，用于图像的创建以及各种图像基础操作，以及图像加载和保存等。

图像格式有很多，一般我们用image.Format.FMT_RGB888 或者 image.Format.FMT_RGBA8888 或者 image.Format.FMT_GRAYSCALE或者image.Format.FMT_BGR888等。

大家知道 RGB 三色可以合成任意颜色，所以一般情况下我们使用 image.Format.FMT_RGB888就足够， RGB888 在内存中是 RGB packed 排列，即在内存中的排列：
像素1_红色, 像素1_绿色, 像素1_蓝色, 像素2_红色, 像素2_绿色, 像素2_蓝色, ... 依次排列。

创建图像

创建图像很简单，只需要指定图像的宽度和高度以及图像格式即可：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
print(img)
print(img.width(), img.height(), img.format())

320 是图像的宽度，240 是图像的高度，image.Format.FMT_RGB888 是图像的格式，格式参数可以省略，默认是image.Format.FMT_RGB888。

这里通过img.width()、img.height()、img.format()可以获取图像的宽度、高度和格式。

显示到屏幕

MaixPy 提供了maix.display.Display类，可以方便的显示图像：

from maix import image, display

disp = display.Display()

img = image.Image(320, 240, image.Format.FMT_RGB888)
disp.show(img)

注意这里因为没有图像数据，所以显示的是黑色的图像，修改画面看后文。

从文件系统读取图像

MaixPy 提供了maix.image.load方法，可以从文件系统读取图像：

from maix import image

img = image.load("/root/image.jpg")
print(img)

注意这里/root/image.jpg 是提前传输到了板子上的，方法可以看前面的教程。
可以支持 jpg 和 png 格式的图像。

保存图像到文件系统

MaixPy 的maix.image.Image提供了save方法，可以保存图像到文件系统：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)

# do something with img
img.save("/root/image.jpg")

画框

image.Image提供了draw_rect方法，可以在图像上画框：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
img.draw_rect(10, 10, 100, 100, image.Color.from_rgb(255, 0, 0))

这里的参数依次是：x, y, w, h, color，x 和 y 是框的左上角坐标，w 和 h 是框的宽度和高度，color 是框的颜色，可以使用image.Color.from_rgb方法创建颜色。
可以用thickness指定框的线宽，默认是1，

也可以画实心框，传参 thickness=-1 即可：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
img.draw_rect(10, 10, 100, 100, (255, 0, 0), thickness=-1)

写字符串

image.Image提供了draw_string方法，可以在图像上写字：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
img.draw_string(10, 10, "Hello MaixPy", image.Color.from_rgb(255, 0, 0))

这里的参数依次是：x, y, text, color，x 和 y 是文字的左上角坐标，text 是要写的文字，color 是文字的颜色，可以使用image.Color.from_rgb方法创建颜色。

还可以放大字体，传参 scale 即可：

img.draw_string(10, 10, "Hello MaixPy", image.Color.from_rgb(255, 0, 0), scale=2)

获取字体的宽度和高度：

w, h = img.string_size("Hello MaixPy", scale=2)
print(w, h)

注意这里scale是放大倍数，默认是1，和draw_string应该保持一致。

画线

image.Image提供了draw_line方法，可以在图像上画线：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
img.draw_line(10, 10, 100, 100, image.Color.from_rgb(255, 0, 0))

这里的参数依次是：x1, y1, x2, y2, color，x1 和 y1 是线的起点坐标，x2 和 y2 是线的终点坐标，color 是线的颜色，可以使用image.Color.from_rgb方法创建颜色。

画圆

image.Image提供了draw_circle方法，可以在图像上画圆：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
img.draw_circle(100, 100, 50, image.Color.from_rgb(255, 0, 0))

这里的参数依次是：x, y, r, color，x 和 y 是圆心坐标，r 是半径，color 是圆的颜色，可以使用image.Color.from_rgb方法创建颜色。

缩放图像

image.Image提供了resize方法，可以缩放图像：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
img_new = img.resize(160, 120)
print(img, img_new)

注意这里resize方法返回一个新的图像对象，原图像不变。

剪裁图像

image.Image提供了crop方法，可以剪裁图像：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
img_new = img.crop(10, 10, 100, 100)
print(img, img_new)

注意这里crop方法返回一个新的图像对象，原图像不变。

旋转图像

image.Image提供了rotate方法，可以旋转图像：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
img_new = img.rotate(90)
print(img, img_new)

注意这里rotate方法返回一个新的图像对象，原图像不变。

拷贝图像

image.Image提供了copy方法，可以拷贝一份独立的图像：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
img_new = img.copy()
print(img, img_new)

仿射变换

image.Image提供了affine方法，可以进行仿射变换，即提供当前图中三个及以上的点坐标，以及目标图中对应的点坐标，可以自动进行图像的旋转、缩放、平移等操作变换到目标图像：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
img_new = img.affine([(10, 10), (100, 10), (10, 100)], [(10, 10), (100, 20), (20, 100)])
print(img, img_new)

更多参数和用法请参考 API 文档。

画关键点

image.Image提供了draw_keypoints方法，可以在图像上画关键点：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)

keypoints = [(10, 10), (100, 10), (10, 100)]
img.draw_keypoints(keypoints, image.Color.from_rgb(255, 0, 0), size=10, thickness=1, fill=False)

在坐标(10, 10)、(100, 10)、(10, 100)画三个红色的关键点，关键点的大小是10，线宽是1，不填充。

画十字

image.Image提供了draw_cross方法，可以在图像上画十字：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
img.draw_cross(100, 100, image.Color.from_rgb(255, 0, 0), size=5, thickness=1)

在坐标(100, 100)画一个红色的十字，十字的延长大小是5，所以线段长度为2 * size + thickness, 线宽是1。

画箭头

image.Image提供了draw_arrow方法，可以在图像上画箭头：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
img.draw_arrow(10, 10, 100, 100, image.Color.from_rgb(255, 0, 0), thickness=1)

在坐标(10, 10)画一个红色的箭头，箭头的终点是(100, 100)，线宽是1。

画图

image.Image提供了draw_image方法，可以在图像上画图：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
img2 = image.Image(100, 100, image.Format.FMT_RGB888)
img2.draw_rect(10, 10, 90, 90, image.Color.from_rgb(255, 0, 0))
img.draw_image(10, 10, img2)

转换格式

image.Image提供了to_format方法，可以转换图像格式：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
img_new = img.to_format(image.Format.FMT_BGR888)
print(img, img_new)
img_jpg = img.to_format(image.Format.FMT_JPEG)
print(img, img_new)

注意这里to_format方法返回一个新的图像对象，原图像不变。

和 Numpy/OpenCV 格式互相转换

另外还可以转换成numpy数组，这样就能给numpy和opencv等库使用：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
img_np = image.image2cv(img)
img2   = image.cv2image(img_np)
print(type(img_np), img_np, img_np.shape)
print(type(img2), img2)

和 bytes 数据互相转换

image.Image提供了to_bytes方法，可以转换图像为bytes数据：

from maix import image

img = image.Image(320, 240, image.Format.FMT_RGB888)
data = img.to_bytes()
print(type(data), len(data), img.data_size())

img2 = image.Image(320, 240, image.Format.FMT_RGB888, data)
print(img2)

这里to_bytes获得一个新的bytes对象，是独立的内存，不会影响原图。
image.Image构造函数中传入data参数可以直接从bytes数据构造图像对象，注意新的图像也是独立的内存，不会影响到data。

因为涉及到内存拷贝，所以这个方法比较耗时，不建议频繁使用。

如果你想用不拷贝的方式优化程序（不建议轻易使用，写不好代码会导致程序容易崩溃，），请看 API 文档。

MaixPy