基于pytorch实现的ICCV2019HarDNet: A Low Memory Traffic Network算法

时间:2022-03-03 17:19:53
【文件属性】:

文件名称:基于pytorch实现的ICCV2019HarDNet: A Low Memory Traffic Network算法

文件大小:526.95MB

文件格式:ZIP

更新时间:2022-03-03 17:19:53

语义分割 FCHarDNet 深度学习 pytorch

1、 训练数据准备 所有数据均放置于Sample\Build\下训练数据按照: Sample\Build\train\ IMG_T1 -----------------------\ IMG_LABEL -----------------\val\ IMG_T1 -----------------------\IMG_LABEL 2、 训练参数设置 参数和运行皆在dp0_train.py中,主要参数包括: (1) num_epochs,训练批次 (2) learning_rate,学习率 (3) dataset,步骤一自己构建的数据集名称 (4) band,输入数据通道数(波段数) (5) n_class, 模型输出通道数(类别) 3、 训练模型位置 模型结构放与net文件中,生成模型文件pt位于Checkpointspath\ 4、 预测使用 填写参数 Checkpointspath,模型位置名称 Dataset,待预测数据文件夹 Outputpath,输出数据文件夹


网友评论