demo-tensorflow

复制代码
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')
相关推荐
用户372157426135几秒前
Python 高效实现 Excel 与 TXT 文本文件之间的数据转换
python
这张生成的图像能检测吗5 分钟前
(论文速读)超像素引导低光图像增强与特征恢复
图像处理·人工智能·目标检测·计算机视觉·低照度图像增强
机器之心15 分钟前
吴恩达关注的Ling-1T背后,蚂蚁Ling 2.0技术报告解密万亿模型开源配方
人工智能·openai
空白到白28 分钟前
NLP相关面试题
人工智能·自然语言处理
嵌入式-老费31 分钟前
Easyx图形库应用(和opencv协同)
人工智能·opencv·计算机视觉
放羊郎34 分钟前
基于三维点云图的路径规划
人工智能·动态规划·slam·点云·路径规划·激光slam
算家计算35 分钟前
英伟达亮出最强AI芯片!性能暴增3倍,算力竞赛迎来新节点
人工智能·nvidia·芯片
AndrewHZ43 分钟前
【图像处理基石】图像滤镜的算法原理:从基础到进阶的技术解析
图像处理·python·opencv·算法·计算机视觉·滤镜·cv
lxmyzzs1 小时前
【图像算法 - 30】基于深度学习的PCB板缺陷检测系统: YOLOv11 + UI界面 + 数据集实现
人工智能·深度学习·算法·yolo·缺陷检测
Patrick_Wilson1 小时前
AI 时代下的工程师核心竞争力思考
人工智能·ai编程·求职