python - 用于图像过滤器的 Python - TypeError:“NoneType”对象不可下标
问题描述
我试图用 python、opencv、skimage 等在图像上放置不同的过滤器。这个操作将帮助我进行神经训练程序。但我现在的过滤器代码有问题。
import numpy as np
import os
import skimage.io as io
import matplotlib.pyplot as plt
from PIL import ImageFilter
from skimage.util import random_noise
from skimage.transform import rotate, AffineTransform, warp
from skimage.filters import gaussian
import shutil
from PIL import Image
import warnings
import time
import os.path
warnings.filterwarnings('ignore')
print("hello") # Affiche "hello"
# Localisation du dossier source
source_images = ''
# Localisation du dossier destination
desti = ''
source_labels = ''
#filtre mexican_hat (find edges)/ Affiche un filtre sombre et fais ressortir les contours de l'objet sur l'integralité de l'image
def Mexican(img): #fonction Mexican
filter = np.array([[0,0,-1,0,0],[0,-1,-2,-1,0],[-1,-2,16,-2,-1],[0,-1,-2,-1,0],[0,0,-1,0,0]])
mexican_hat_img1=cv2.filter2D(img,-1,filter) #paramètres du filtre
return mexican_hat_img1 #renvoi de l'image filtrée
#filtre couleur (0=Blue,1=Green,2=Red)/ Affiche du Rouge, du Bleu ou du Vert sur l'integralité de l'image
def exponential_function(channel, exp):
channel_color =channel
table = np.array([min((i**exp), 255) for i in np.arange(0, 256)]).astype("uint8") # creating table for exponent
channel_color = cv2.LUT(channel_color, table)
return channel_color
def tone(img, number): #fonction Blue/Green/Red
img_color = img
for i in range(3):
if i == number:
img_color[:, :, i] = exponential_function(img_color[:, :, i], 1.10) # applying exponential function on slice
else:
img_color[:, :, i] = 0 # setting values of all other slices to 0 to blue 1to green 2to red
return img_color
def desti_labels():
count=1
for filename in sorted(os.listdir(source_labels)):
param0 = str(count) + "Blur" + '.txt'
param1 = str(count) + "Noise" + '.txt'
param2 = str(count) + "Sharpen" + '.txt'
param3 = str(count) + "Blue" + '.txt'
param4 =str(count) + "Green" + '.txt'
param5 =str(count) + "Red" + '.txt'
param6 =str(count) + "Sepia" + '.txt'
param7 =str(count) + "Mexican" + '.txt'
param8 =str(count) + '.txt'
src ="/home/melanie/Desktop/test-filtre/crease/final-crease"+filename
#dst0 ="/Desktop/Quality_Inspection/Defects_location_for_metal_surface/Classe 1 crease/test-augmentor"+param0
#dst1 ="/Desktop/Quality_Inspection/Defects_location_for_metal_surface/Classe 1 crease/test-augmentor"+param1
#dst2 ="/home/alternant2/augment_env_TP_folder/Test_data100/Final/"+param2
dst3 ="/home/melanie/Desktop/test-filtre/crease/final-crease"+param3
dst4 ="/home/melanie/Desktop/test-filtre/crease/final-crease"+param4
dst5 ="/home/melanie/Desktop/test-filtre/crease/final-crease"+param5
#dst6 ="/Desktop/Quality_Inspection/Defects_location_for_metal_surface/Classe 1 crease/test-augmentor"+param6
dst7 ="/home/melanie/Desktop/test-filtre/crease/final-crease"+param7
dst8 ="/home/melanie/Desktop/test-filtre/crease/final-crease"+param8
shutil.copyfile(src,dst3)
shutil.copyfile(src,dst4)
shutil.copyfile(src,dst5)
shutil.copyfile(src,dst7)
shutil.copyfile(src,dst8)
count+=1
print("count", count)
return
lst_img=os.listdir(source_images)
print(lst_img)
def dest_images():
count_img=1
for filename in sorted(lst_img): #Pour tous les fichiers dans source
img=cv2.imread(source_images+filename) #choix du fichier
print(type(img)) #Affiche le type de l'image(dans notre cas ndarray)
img=cv2.imread(source_images+filename) #choix du fichier
img_blue = tone(img,0) #appel fonction Blue
img=cv2.imread(source_images+filename) #choix du fichier
img_green =tone(img,1) #appel fonction Green
img=cv2.imread(source_images+filename) #choix du fichier
img_red = tone(img,2) #appel fonction Red
img=cv2.imread(source_images+filename) #choix du fichier
img_mexican = Mexican(img) #appel fonction Mexican
#CONFIG 1
new_filename=str(count_img) # Remove the 4 last character ==> .PNG i.e.
print("A few moments later...") # Afficher "A few moments later..."
time.sleep(3) # Delai d'attente
cv2.imwrite(new_filename+'Blue.JPG',img_blue) #Enregistre le filtre Blue
cv2.imwrite(new_filename+'Green.JPG',img_green) #Enregistre le filtre Green
cv2.imwrite(new_filename+'Red.JPG',img_red) #Enregistre le filtre Red
cv2.imwrite(new_filename+'Mexican.JPG',img_mexican) #Enregistre le filtre Mexican
cv2.imwrite(new_filename+'.JPG',img)
count_img+=1 #Enregistre l'image originale
return
os.chdir(desti) # "Dans le fichier appelé desti"
dest_images()
desti_labels()
当我运行此代码时,我在终端上看到此错误:
hello
['final-crease', 'img_01_425382900_00002.jpg']
<class 'NoneType'>
Traceback (most recent call last):
File "/home/melanie/Desktop/test-de-merde-3.py", line 115, in <module>
dest_images()
File "/home/melanie/Desktop/test-de-merde-3.py", line 88, in dest_images
img_blue = tone(img,0) #appel fonction Blue
File "/home/melanie/Desktop/test-de-merde-3.py", line 42, in tone
img_color[:, :, i] = exponential_function(img_color[:, :, i], 1.10) # applying exponential function on slice
TypeError: 'NoneType' object is not subscriptable
我不得不承认,我真的不知道该怎么办。不幸的是,我没有找到解决方案。如果你有想法,我会接受的:)
解决方案
print(type(img))
<class 'NoneType'>
该行img=cv2.imread(source_images+filename)
没有找到文件,因此返回 None。当您尝试在第 42 行建立索引时,您会收到错误,因为您无法对 None 进行切片。
--尝试img=cv2.imread(source_images+"/"+filename)
更改目录。--这将不起作用source_images=0
推荐阅读
- python - 想要从输出中获取最大值并将它们应用于方程
- sql-server - 单个 SQL 查询选择所有客户以及属于每个客户的所有订单?
- powershell - 带有用户名和密码的powershell soaprequest
- powershell - 使用 PowerShell 脚本映射主目录时出错
- html - 如何在 CSS 文件中使用 Flask url_for?
- apache-kafka - kafka.tools.GetOffsetShell 在 Kafka 1.0 上启用 Kerberos 安全性
- angularjs - 使用 ng-if-start 而不是多个 ng-if 时有什么性能提升?
- python - Numpy索引:二维数组中每一行的第一个(变化的)元素数
- python - 如何加入大量数据框
- sql - 将表中的行转换为星期几