Ejemplo n.º 1
0
def preprocess(fname):
    # return fname
    image = cv2.imread(fname)
    im_data = np.expand_dims(
        yolo_utils.preprocess_test((image, None, cfg.multi_scale_inp_size),
                                   0)[0], 0)
    return image, im_data
Ejemplo n.º 2
0
def preprocess(fname):
    # return fname
    # read the pic  fname is the path of pic
#    ipdb.set_trace()
    image = cv2.imread(fname)
    im_data = np.expand_dims(
        yolo_utils.preprocess_test((image, None, cfg.multi_scale_inp_size), 0)[0], 0)
    return image, im_data
Ejemplo n.º 3
0
 def preprocess(self, content, method='nparray'):
     image = 0
     if method == 'path':
         image = cv2.imread(content)
     elif method == 'nparray':
         image = content
     else:
         print('\n\nError in preprocess\n\n')
     #yolo_utils.preprocess_test((image, none, cfg.multi_scale_inp_size), 0)将图像转换到RGB空间,resize到(320, 320),值归一化到(0,1)返回的0号元素为处理后图像,4号元素为原图拷贝
     im_data = np.expand_dims(
         yolo_utils.preprocess_test((image, None, cfg.multi_scale_inp_size),
                                    0)[0], 0)
     #im_data被转换为(1,3,320,320)归一化np矩阵
     return image, im_data
Ejemplo n.º 4
0
def preprocess(fname):
    # return fname
    image = cv2.imread(fname)
    im_data = np.expand_dims(yolo_utils.preprocess_test(image, cfg.inp_size),
                             0)
    return image, im_data
Ejemplo n.º 5
0
def preprocess(filename):
    image = cv2.imread(filename)
    im_data = np.expand_dims(
        yolo_utils.preprocess_test((image, None, cfg.inp_size))[0], 0)
    return image, im_data
Ejemplo n.º 6
0
def preprocess(camera):
    _, image = camera.read()
    im_data = np.expand_dims(
        yolo_utils.preprocess_test((image, None, cfg.inp_size))[0], 0)
    return image, im_data
Ejemplo n.º 7
0
def get_ImData(fname):
    im_data = np.expand_dims(
        yolo_utils.preprocess_test((fname, None, cfg.inp_size))[0], 0)
    return im_data
Ejemplo n.º 8
0
def preprocess(frame):
    # return fname
    image = frame
    im_data = np.expand_dims(
        yolo_utils.preprocess_test((frame, None, cfg.inp_size))[0], 0)
    return image, im_data
Ejemplo n.º 9
0
def preprocess(fname):
    # return fname
    image = cv2.imread(fname)
    im_data = np.expand_dims(
        yolo_utils.preprocess_test((image, None, cfg.multi_scale_inp_size), 0)[0], 0)
    return image, im_data