from dataset import generate_data
import numpy as np
import os
from model import enhancednet
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
image_rows = 128
image_cols = 256
filename = 'detached_data.mat'
train_data, train1_data, label_data = generate_data(filename)
train_data = np.array(train_data, dtype=float)
train_data = np.reshape(train_data, (len(train_data), image_rows, image_cols, 1))
# print(train_data.shape)
train1_data = np.array(train1_data, dtype=float)
train1_data = np.reshape(train1_data, (len(train1_data), image_rows, image_cols, 1))
model = enhancednet()
model.fit(train_data, train1_data, batch_size=32, epochs=100, verbose=2, shuffle=True, validation_split=0.1)
model.save('enhanced_model.h5')
demo-tensorflow
yyfhq2024-10-24 11:25
相关推荐
用户372157426135几秒前
Python 高效实现 Excel 与 TXT 文本文件之间的数据转换这张生成的图像能检测吗5 分钟前
(论文速读)超像素引导低光图像增强与特征恢复机器之心15 分钟前
吴恩达关注的Ling-1T背后,蚂蚁Ling 2.0技术报告解密万亿模型开源配方空白到白28 分钟前
NLP相关面试题嵌入式-老费31 分钟前
Easyx图形库应用(和opencv协同)放羊郎34 分钟前
基于三维点云图的路径规划算家计算35 分钟前
英伟达亮出最强AI芯片!性能暴增3倍,算力竞赛迎来新节点AndrewHZ43 分钟前
【图像处理基石】图像滤镜的算法原理:从基础到进阶的技术解析lxmyzzs1 小时前
【图像算法 - 30】基于深度学习的PCB板缺陷检测系统: YOLOv11 + UI界面 + 数据集实现Patrick_Wilson1 小时前
AI 时代下的工程师核心竞争力思考