``資料所在的目錄只有 50 個檔案,但在分配影像標簽并調整大小后,當我列印資料的 len 時,我得到 750。我開始懷疑影像中的陣列是否被分離并分配了自己的標簽. 影像被賦予標簽并調整大小的代碼
import tensorflow as tf
from tensorflow import keras
from keras.models import Sequential
from keras.layers import Dense, Flatten, Conv2D, MaxPooling2D, Dropout
from tensorflow.keras import layers
from tensorflow.keras.utils import to_categorical
import numpy as np
import matplotlib.pyplot as plt
import os
import cv2
import random
Directory='../input/hotdog'
Directory1='../input/nothotdog'
image_size=100
data=[]
for directory in Directory:
folder=os.path.join(Directory)
label=0
for img in os.listdir(Directory):
img_path=os.path.join(Directory, img)
img_arr= cv2.imread(img_path)
img_arr= cv2.resize(img_arr, dsize=(image_size, image_size))
data.append([img_arr, label])
print(len(data))
for directory in Directory1:
folder=os.path.join(Directory1)
label=1
for img in os.listdir(Directory1):
img_path=os.path.join(Directory1, img)
img_arr= cv2.imread(img_path)
img_arr= cv2.resize(img_arr, dsize=(image_size, image_size))
data.append([img_arr, label])
uj5u.com熱心網友回復:
你的線路
for directory in Directory:
回圈Directory包含 is的字符'../input/hotdog'。
您應該完全洗掉for回圈:
import tensorflow as tf
from tensorflow import keras
from keras.models import Sequential
from keras.layers import Dense, Flatten, Conv2D, MaxPooling2D, Dropout
from tensorflow.keras import layers
from tensorflow.keras.utils import to_categorical
import numpy as np
import matplotlib.pyplot as plt
import os
import cv2
import random
Directory='../input/hotdog'
Directory1='../input/nothotdog'
image_size=100
data=[]
label=0
for img in os.listdir(Directory):
img_path=os.path.join(Directory, img)
img_arr= cv2.imread(img_path)
img_arr= cv2.resize(img_arr, dsize=(image_size, image_size))
data.append([img_arr, label])
print(len(data))
label=1
for img in os.listdir(Directory1):
img_path=os.path.join(Directory1, img)
img_arr= cv2.imread(img_path)
img_arr= cv2.resize(img_arr, dsize=(image_size, image_size))
data.append([img_arr, label])
轉載請註明出處,本文鏈接:https://www.uj5u.com/ruanti/396763.html
