Save serialized pictures into redis

img-to-redis.py

import pickle
import redis
 
class CACHE:
    def __init__(self, host='127.0.0.1',password='123456'):
        pool = redis.ConnectionPool(host=host, password=password)
        self.conn = redis.Redis(connection_pool=pool)
 
    def insert_image(self, frame_id, frame):
        # 将图片序列化存入redis中
        b = pickle.dumps(frame)  # frame is numpy.ndarray
        self.conn.set(frame_id,b)
 
    def get_image(self, frame_id):
        # 从redis中取出序列化的图片并进行反序列化
        return pickle.loads(self.conn.get(frame_id))