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
相关推荐
九年义务漏网鲨鱼36 分钟前
【大模型学习 | MINIGPT-4原理】元宇宙时间1 小时前
Playfun即将开启大型Web3线上活动,打造沉浸式GameFi体验生态开发者工具分享1 小时前
文本音频违规识别工具排行榜(12选)产品经理独孤虾1 小时前
人工智能大模型如何助力电商产品经理打造高效的商品工业属性画像老任与码1 小时前
Spring AI Alibaba(1)——基本使用蹦蹦跳跳真可爱5892 小时前
Python----OpenCV(图像増强——高通滤波(索贝尔算子、沙尔算子、拉普拉斯算子),图像浮雕与特效处理)nananaij2 小时前
【Python进阶篇 面向对象程序设计(3) 继承】雷羿 LexChien2 小时前
从 Prompt 管理到人格稳定:探索 Cursor AI 编辑器如何赋能 Prompt 工程与人格风格设计(上)两棵雪松2 小时前
如何通过向量化技术比较两段文本是否相似?heart000_12 小时前
128K 长文本处理实战:腾讯混元 + 云函数 SCF 构建 PDF 摘要生成器