numpy数组迭代和切片的最快方法是什么

时间:2017-09-22 07:36:03

标签: python arrays numpy iteration slice

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 是否有更好的方式获得“输入”?(输入=返回值)

抱歉我的英语不好。

0 个答案:

没有答案