K210学习笔记(七)——MAIX BIT(K210)拍照

前言

MAIX BIT使用的是OV5642,像素为500w,但实际使用只有30w,但对于物体检测,人脸识别是够用的,用maix bit进行拍照,所得照片可以直接用于后面的数据集训练,减少了图片格式转换这一步骤。
K210学习笔记(七)——MAIX BIT(K210)拍照_第1张图片

一、初始化摄像头

sensor.reset([, freq=24000000[, set_regs=True[, dual_buff=False]]])

参数说明:
freq: 设置摄像头时钟频率,频率越高帧率越高,但是画质可能更差。默认 24MHz, 如果摄像头有彩色斑点(ov7740),可以适当调低比如 20MHz
set_regs: 允许程序写摄像头寄存器,默认为 True。 如果需要自定义复位序列,可以设置为False,然后使用sensor.__write_reg(addr, value) 函数自定义写寄存器序列
dual_buff: 默认为False。允许使用双缓冲,会增高帧率,但是内存占用也会增加(大约为384KiB)
choice: 指定需要搜索的摄像头类型,ov类型(1),gc类型(2),mt类型(3),不传入该参数则搜索全部类型摄像头

二、使用方法

1.设置帧大小

用于设置摄像头输出帧大小,k210最大支持VGA格式,大于VGA将无法获取图像,Maix Bit开发板配置的屏幕是320*240分辨率,推荐设置为QVGA格式

sensor.set_framesize(framesize[, set_regs=True])

参数说明:
framesize: 帧大小
set_regs: 允许程序写摄像头寄存器,默认为 True。 如果需要自定义设置帧大小的序列,可以设置为False,然后使用sensor.__write_reg(addr, value) 函数自定义写寄存器序列
返回值:
True : 设置成功
False: 设置错误

2.设置帧格式

用于设置摄像头输出格式,Maix Bit开发板配置的屏幕使用的是RGB565,推荐设置为RGB565格式

sensor.set_pixformat(format[, set_regs=True])

format: 帧格式
set_regs: 允许程序写摄像头寄存器,默认为 True。 如果需要自定义设置像素格式的序列,可以设置为False,然后使用sensor.__write_reg(addr, value) 函数自定义写寄存器序列

可选的帧格式有GRAYSCALE, RGB565, YUV422
返回值:
True : 设置成功
False: 设置错误

3.图像捕捉控制

图像捕捉功能控制

sensor.run(enable)

参数说明:
enable: 1 表示开始抓取图像 0 表示停止抓取图像
返回值:
True : 设置成功
False: 设置错误

4.拍摄图像

使用摄像头拍摄一张照片

sensor.snapshot()

返回值:
img: 返回的图像对象

5.分辨率

获取摄像头分辨率宽度,高度

sensor.width()
sensor.height()

返回值:
int类型的摄像头分辨率宽度,高度

6. 设置彩条测试模式

将摄像头设置为彩条测试模式,开启彩条测试模式后,摄像头会输出一彩条图像,常用来检测摄像机总线是否连接正确。

sensor.set_colorbar(enable)

参数说明:
enable: 1 表示开启彩条测试模式 0 表示关闭彩条测试模式

7.设置对比度

设置摄像头对比度

sensor.set_contrast(contrast)

参数:
constrast: 摄像头对比度,范围为[-2,+2]
返回值:
True : 设置成功
False: 设置错误

8. 设置亮度

设置摄像头亮度

sensor.set_brightness(brightness)

参数:
brightness: 摄像头亮度,范围为[-2,+2]
返回值:
True : 设置成功
False: 设置错误

9. 设置饱和度

设置摄像头饱和度

sensor.set_saturation(saturation)

constrast: 摄像头饱和度,范围为[-2,+2]
返回值:
True : 设置成功
False: 设置错误

10. 设置自动增益

设置摄像自动增益模式,如果需要追踪颜色,需要关闭自动增益

sensor.set_auto_gain(enable,gain_db)
sensor.get_gain_db()#获取摄像头增益值,返回float类型的增益值

参数:
enable: 1 表示开启自动增益 0 表示关闭自动增益
gain_db: 关闭自动增益时,设置的摄像头固定增益值,单位为dB

11. 设置摄像头

设置摄像头水平镜像

sensor.set_hmirror(enable)

参数:
enable: 1 表示开启水平镜像 0 表示关闭水平镜像
设置摄像头垂直翻转

sensor.set_vflip(enable)

参数:
enable: 1 表示开启垂直翻转 0 表示关闭垂直翻转

12. 设置IDE质量

设置传送给 IDE 图像的质量

sensor.set_jb_quality(quality)

quality:int 类型,图像质量百分比(0~100),数字越大质量越好

三、用MAIX BIT拍照

使用 这个 脚本来进行采集图片
按照 图片采集脚本使用说明 采集图片到SD卡
将SD卡中的图片拷贝到电脑, 整理成上面的目录结构, 所有图片的分辨率为224x224
注意 SD卡需要硬件支持SPI通信, 并使用MBR(msdos)分区方案,并格式化为FAT32格式
代码:

import sensor, lcd
from Maix import GPIO
from fpioa_manager import fm
from board import board_info
import os, sys
import time
import image

#### image size ####
set_windowing = (224, 224)

#### sensor config ####

sensor.reset(freq=22000000, dual_buff=False)
sensor.set_pixformat(sensor.RGB565)
sensor.set_framesize(sensor.QVGA) # 320x240
try:
    sensor.set_jb_quality(95)         # for IDE display quality
except Exception:
    pass # no IDE support
if set_windowing:
    sensor.set_windowing(set_windowing)



sensor.skip_frames()

#### lcd config ####
lcd.init(type=1, freq=15000000)
lcd.rotation(2)

#### boot key ####
boot_pin = 16 # board_info.BOOT_KEY
fm.register(boot_pin, fm.fpioa.GPIOHS0)
key = GPIO(GPIO.GPIOHS0, GPIO.PULL_UP)

######################################################

#### main ####
def capture_main(key):
    def draw_string(img, x, y, text, color, scale, bg=None , full_w = False):
        if bg:
            if full_w:
                full_w = img.width()
            else:
                full_w = len(text)*8*scale+4
            img.draw_rectangle(x-2,y-2, full_w, 16*scale, fill=True, color=bg)
        img = img.draw_string(x, y, text, color=color,scale=scale)
        return img

    def del_all_images():
        os.chdir("/sd")
        images_dir = "cap_images"
        if images_dir in os.listdir():
            os.chdir(images_dir)
            types = os.listdir()
            for t in types:
                os.chdir(t)
                files = os.listdir()
                for f in files:
                    os.remove(f)
                os.chdir("..")
                os.rmdir(t)
            os.chdir("..")
            os.rmdir(images_dir)

    # del_all_images()
    os.chdir("/sd")
    dirs = os.listdir()
    images_dir = "cap_images"
    last_dir = 0
    for d in dirs:
        if d.startswith(images_dir):
            if len(d) > 11:
                n = int(d[11:])
                if n > last_dir:
                    last_dir = n
    images_dir = "{}_{}".format(images_dir, last_dir+1)
    print("save to ", images_dir)
    if images_dir in os.listdir():
        img = image.Image()
        img = draw_string(img, 2, 200, "please del cap_images dir", color=lcd.WHITE,scale=1, bg=lcd.RED)
        lcd.display(img)
        sys.exit(1)
    os.mkdir(images_dir)
    last_cap_time = 0
    last_btn_status = 1
    save_dir = 0
    save_count = 0
    os.mkdir("{}/{}".format(images_dir, save_dir))
    while(True):
        img0 = sensor.snapshot()
        if set_windowing:
            img = image.Image()
            img = img.draw_image(img0, (img.width() - set_windowing[0])//2, img.height() - set_windowing[1])
        else:
            img = img0.copy()
        # img = img.resize(320, 240)
        if key.value() == 0:
            time.sleep_ms(30)
            if key.value() == 0 and (last_btn_status == 1) and (time.ticks_ms() - last_cap_time > 500):
                last_btn_status = 0
                last_cap_time = time.ticks_ms()
            else:
                if time.ticks_ms() - last_cap_time > 5000:
                    img = draw_string(img, 2, 200, "release to change type", color=lcd.WHITE,scale=1, bg=lcd.RED)
                else:
                    img = draw_string(img, 2, 200, "release to capture", color=lcd.WHITE,scale=1, bg=lcd.RED)
                    if time.ticks_ms() - last_cap_time > 2000:
                        img = draw_string(img, 2, 160, "keep push to change type", color=lcd.WHITE,scale=1, bg=lcd.RED)
        else:
            time.sleep_ms(30)
            if key.value() == 1 and (last_btn_status == 0):
                if time.ticks_ms() - last_cap_time > 5000:
                    img = draw_string(img, 2, 200, "change object type", color=lcd.WHITE,scale=1, bg=lcd.RED)
                    lcd.display(img)
                    time.sleep_ms(1000)
                    save_dir += 1
                    save_count = 0
                    dir_name = "{}/{}".format(images_dir, save_dir)
                    os.mkdir(dir_name)
                else:
                    draw_string(img, 2, 200, "capture image {}".format(save_count), color=lcd.WHITE,scale=1, bg=lcd.RED)
                    lcd.display(img)
                    f_name = "{}/{}/{}.jpg".format(images_dir, save_dir, save_count)
                    img0.save(f_name, quality=95)
                    save_count += 1
                last_btn_status = 1
        img = draw_string(img, 2, 0, "will save to {}/{}/{}.jpg".format(images_dir, save_dir, save_count), color=lcd.WHITE,scale=1, bg=lcd.RED, full_w=True)
        lcd.display(img)
        del img
        del img0


def main():
    try:
        capture_main(key)
    except Exception as e:
        print("error:", e)
        import uio
        s = uio.StringIO()
        sys.print_exception(e, s)
        s = s.getvalue()
        img = image.Image()
        img.draw_string(0, 0, s)
        lcd.display(img)
main()

总结

用MAIX BIT(K210)拍出来的照片可以直接当作数据集来使用。

你可能感兴趣的:(K210学习笔记,人工智能)