import tensorflow as tf from tensorflow import keras from keras import layers import xml.etree.ElementTree as ET import pathlib from pathlib import Path file_path = Path('C:/1') def net_init(): model = keras.Sequential([layers.Input(shape=(1200, 1600, 3))]) model.add(layers.Conv2D(filters=3, activation="relu", kernel_size=(3, 3), padding="same", strides=2)) model.add(layers.MaxPool2D(pool_size=(2, 2))) model.add(layers.Conv2D(filters=3, activation="relu", kernel_size=(3, 3), padding="same", strides=2)) model.add(layers.MaxPool2D(pool_size=(2, 2))) model.add(layers.Conv2D(filters=1, activation="relu", kernel_size=(3, 3), padding="same", strides=2)) model.add(layers.MaxPool2D(pool_size=(2, 2))) model.add(layers.Dense(48, activation='relu')) model.add(layers.Dense(2, activation='softmax')) return model def load_xml(folder_path: Path) -> list: feature_list = [] file_list = [] label_list = [] for file_name in folder_path.glob('*.xml'): xml_tree = ET.parse(file_name) root = xml_tree.getroot() feature = ( int(root.find('object/bndbox/xmin').text), int(root.find('object/bndbox/ymin').text), int(root.find('object/bndbox/xmax').text), int(root.find('object/bndbox/ymax').text) ) feature_list.append(feature) file_list.append(file_name) label_list.append(root.find('object/name').text) return feature_list, file_list, label_list def load_img(folder_path : Path, xml_list : list): img_list = [] print(xml_list) for img_name in folder_path.glob('*.jpg'): print(img_name) xml_name = img_name.with_suffix('.xml') print(xml_name) if xml_name in xml_list: print("yes") img = tf.io.read_file(img_name.as_posix()) img = tf.image.decode_image(img, channels=3) img = tf.image.per_image_standardization(img) img_list.append(img) return img_list def main(): feature_list, file_list, label_list = load_xml(file_path) img_list = load_img(file_path, file_list) model = net_init() model.compile(optimizer='adam', loss=tf.keras.losses.mse, metrics=['accuracy']) model.fit(img_list, feature_list, epochs=1) main()这段程序有什么问题
时间: 2024-02-05 07:04:12 浏览: 146
解决import tensorflow as tf 出错的原因
5星 · 资源好评率100%
这段程序可能会有以下问题:
1. 程序中没有明确指定输入的图像尺寸,而模型中的第一层需要接受一个 (1200, 1600, 3) 的输入,可能会导致输入图像的尺寸不符合要求而导致程序出错。
2. 程序中使用了标准化函数 tf.image.per_image_standardization() 对输入图像进行标准化处理,但这个函数会将图像像素值减去均值再除以标准差,这有可能会导致图像像素值超出了合理范围,需要使用其他的标准化方法。
3. 程序中编译模型时,损失函数使用的是均方误差(MSE),但这个损失函数通常用于回归问题,对于分类问题来说,应该使用交叉熵损失函数。
4. 程序中训练模型时,只训练了一个 epoch,这可能会导致模型未能充分学习特征,需要增加训练 epochs 的数量。
阅读全文