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
相关推荐
ACP广源盛1392462567317 分钟前
(ACP广源盛)GSV1175---- MIPI/LVDS 转 Type-C/DisplayPort 1.2 转换器产品说明及功能分享胡耀超21 分钟前
隐私计算技术全景:从联邦学习到可信执行环境的实战指南—数据安全——隐私计算 联邦学习 多方安全计算 可信执行环境 差分隐私Turnsole_y2 小时前
pycharm自动化测试初始化停停的茶2 小时前
深度学习(目标检测)Y200309162 小时前
基于 CIFAR10 数据集的卷积神经网络(CNN)模型训练与集成学习老兵发新帖2 小时前
主流神经网络快速应用指南weixin-a153003083163 小时前
[数据抓取-1]beautifulsoupAI量化投资实验室3 小时前
15年122倍,年化43.58%,回撤才20%,Optuna机器学习多目标调参backtrader,附python代码java_logo3 小时前
vllm-openai Docker 部署手册倔强青铜三3 小时前
苦练Python第67天:光速读取任意行,linecache模块解锁文件处理新姿势