python代码在gpu上面运行_如何将python代码转换为在GPU上而不是CPU上运行?
·
我得到了一个python代码,它接收两个图像作为输入,并使用Gabor过滤器来查找两个图像的RGB的相关性,并将其保存在csv文件中。所以我需要使用GPU来执行程序,因为它需要很多时间和CPU利用率。我有一个geforcegtx1050ti和一个完全的编程初学者。在
我做了一些研究,了解了CUDA和Tensorflow,但是我真的不确定如何继续实现它,以及在不改变代码的情况下实现它的最佳方法是什么。在#Gabor Filter
def build_filters():
filters = []
#tesing phrase filter - reduce
for ksize in range(9, 19, 5):
for theta in np.arange(45, 225, 45):
for sigma in range(2,6,2):
kern = cv2.getGaborKernel((ksize, ksize), sigma, theta, 5.0, 0.5, 0, ktype=cv2.CV_32F)
kern /= 1.5*kern.sum()
filters.append(kern)
return filters
#Apply filter into the image
def process(images, f):
accum = np.zeros_like(images)
for kern in f:
fimg = cv2.filter2D(images, cv2.CV_8UC3, kern)
np.maximum(accum, fimg, accum)
return accum
谢谢你!在
更多推荐



所有评论(0)