nn_calc.py 文件源码

def image_postprocessing(img, t_size_y, t_size_x, feedback, t):

    # resize image
    img = cv2.resize(img, (t_size_y, t_size_x))

    # cut image
    img = img[t_size_y/2-1:-1,:]

    # filter image
    ret,img = cv2.threshold(img,160,255,cv2.THRESH_BINARY)

    #store if flag is set
    if feedback:
        cv2.imwrite('feedback/image_' + str(t) + '_filter.png', img)

    return img

# add image and depth image
# will store all stage of processing if flag for feedback is set