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
相关推荐
时空无限13 小时前
说说transformer 中的掩码矩阵以及为什么能掩盖住词语HAH-HAH13 小时前
【Python 入门】(2)Python 语言基础(变量)查里王13 小时前
AI 3D 生成工具知识库:当前产品格局与测评总结武子康13 小时前
AI-调查研究-76-具身智能 当机器人走进生活:具身智能对就业与社会结构的深远影响小鹿清扫日记13 小时前
从蛮力清扫到 “会看路”:室外清洁机器人的文明进阶SunnyDays101114 小时前
Python 轻松实现替换或修改 PDF 文字fanstuck14 小时前
Prompt提示工程上手指南(六):AI避免“幻觉”(Hallucination)策略下的PromptJust_Paranoid14 小时前
【Settings】恢复出厂设置密码校验zhangfeng113314 小时前
win7 R 4.4.0和RStudio1.25的版本兼容性以及系统区域设置有关 导致Plots绘图面板被禁用,但是单独页面显示DogDaoDao15 小时前
神经网络稀疏化设计构架方法和原理深度解析