Обработка изображений и просмотр изображений в папках для поиска SPN - PullRequest
0 голосов
/ 26 апреля 2020

Я пытался выучить python, поэтому я новичок в этом, я пытаюсь сделать следующее: набор изображений в папке 'original_images' обрезается и сохраняется в папке 'Cropped_Pics', откуда я смотрю только зеленый канал изображения и сохраните его в папке «Cropped_green_Image». сейчас код работает только для одного изображения, все процессы выполняются, однако я не уверен, как l oop проходить через каждое изображение в каждой папке, поэтому в итоге у меня есть все обрезанные и зеленые изображения каналов в папке 'Cropped_green_Image'.

код следующий:

# find image#
img = Image.open(r"C:\Users\danma\Desktop\\Project\Original_Images/pic2.jpg")
# crop image#
b = (0, 0, 3, 3)
c_i = img.crop(box=b)
# save#
c_i.save(r"C:\Users\danma\Desktop\\Project\Cropped_Pics\pic1C.jpg")

# find image#
a = Image.open(r"C:\Users\danma\Desktop\\Project\Cropped_Pics/pic1C.jpg")
a = numpy.array(a)
# show green channel#
a[:, :, 0] *= 0
a[:, :, 2] *= 0
a = Image.fromarray(a)
# save#
a.save(r"C:\Users\danma\Desktop\Project\Cropped_Green_Image/pic1.jpg")

# import an image#
im1 = Image.open(r"C:\Users\danma\Desktop\Project\Cropped_Green_Image/pic1.jpg")
# set as array#
ar = array(im1)
# print text#
print("wiener filter matrix:")


# wiener filter function called#
def wiener_filter(signal):
    """ wiener filter. """
    output = wiener(array(im1), 2 ** 5 - 1)
    return output


# print array for filtered image#
print(wiener(array(im1), 2 ** 5 - 1))
...