tensorflow读取img,csv

import tensorflow as tf
import cv2 as cv


image_add_list=[]
image_label_list=[]
with open("train_list.csv") as fid:
    for image in fid.readlines():
        image_add_list.append(image.strip().split(",")[0])
        image_label_list.append(image.strip().split(",")[1])
#用tensor来存储img
def get_image(path):
    return tf.image.convert_image_dtype(tf.image.decode_jpeg(tf.read_file(path),channels=1),dtype=tf.float32)
    


with tf.Session() as sess:
    img=get_image('jpg/image_0000.jpg')
    cv2img=sess.run(img)
    img2=cv.resize(cv2img,(200,200))
    cv.imshow('image',cv2img)
    cv.waitKey()
    
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
检查下述代码并修改错误import tensorflow as tf from tensorflow.keras.models import Sequential from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense import pandas as pd import numpy as np import cv2 import os 构建模型 model = Sequential() model.add(Conv2D(32, (3, 3), activation='relu', input_shape=(80, 160, 3))) # (None, 80, 160, 3) model.add(MaxPooling2D((2, 2))) model.add(Conv2D(64, (3, 3), activation='relu')) model.add(MaxPooling2D((2, 2))) model.add(Flatten()) model.add(Dense(64, activation='relu')) model.add(Dense(62, activation='softmax')) # 36表示0-9数字和A-Z(a-z)字母的类别数 编译模型 model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy']) 验证码图片加载 定义训练数据和标签的文件夹路径 train_data_folder = r'C:\Users\CXY\PycharmProjects\pythonProject\data\train' train_labels_folder = r'C:\Users\CXY\PycharmProjects\pythonProject\data' 加载训练数据 train_data = [] train_labels = pd.read_csv(r'C:\Users\CXY\PycharmProjects\pythonProject\data\traincodes.csv')['code'].values 遍历训练数据文件夹,读取每个图片并添加到训练数据列表 for filename in os.listdir(train_data_folder): img_path = os.path.join(train_data_folder, filename) img = cv2.imread(img_path) train_data.append(img) # 遍历训练标签文件夹,读取每个标签并添加到训练标签列表 for filename in os.listdir(train_labels_folder): label_path = os.path.join(train_labels_folder, filename) label = cv2.imread(label_path, 0) # 读取灰度图像 train_labels.append(label) 转换训练数据和标签为NumPy数组 train_data = np.array(train_data) train_labels = np.array(train_labels) 训练模型 model.fit(train_data, train_labels, epochs=10, batch_size=32) 保存模型 model.save('captcha_model.h5')
最新发布
07-09

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值