import cv2
import numpy as np
def get_image_input(img):
h, w = img.shape[:2]
inputs = []
for y in range(4:h-4):
for x in range(4:w-4):
inputs.append(img[y-4:y+5, x-4:x+5])
return inputs
# img is numpy 2d array
img = cv2.imread(SAMPLE_IMG, cv2.IMREAD_GRAYSFCALE)
inputs = get_image_input(img)
我不擅长python和numpy 是否有更好的方式获得“输入”?(输入=返回值)
抱歉我的英语不好。