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
相关推荐
charley.layabox6 小时前
8月1日ChinaJoy酒会 | 游戏出海高端私享局 | 平台 × 发行 × 投资 × 研发精英畅饮畅聊DFRobot智位机器人6 小时前
AIOT开发选型:行空板 K10 与 M10 适用场景与选型深度解析想成为风筝8 小时前
从零开始学习深度学习—水果分类之PyQt5AppF_D_Z8 小时前
MMaDA:多模态大型扩散语言模型江沉晚呤时8 小时前
在 C# 中调用 Python 脚本:实现跨语言功能集成大知闲闲哟9 小时前
深度学习G2周:人脸图像生成(DCGAN)飞哥数智坊9 小时前
Coze实战第15讲:钱都去哪儿了?Coze+飞书搭建自动记账系统wenzhangli79 小时前
低代码引擎核心技术:OneCode常用动作事件速查手册及注解驱动开发详解电脑能手9 小时前
如何远程访问在WSL运行的Jupyter NotebookEdward-tan10 小时前
CCPD 车牌数据集提取标注,并转为标准 YOLO 格式