ICnet模型复现

数据集存放路径 datasets-Cityscapes- 

 

 

 

train:specific_gpu_num: "0"     # for example: "0", "1" or "0, 1" 单显卡默认是0train_batch_size: 1      # adjust according to gpu resources 笔记本显存一般是4g,基本上是设置最小为1,可以试valid_batch_size: 1cityscapes_root: "/home/s/Desktop/ICNet-pytorch-master/ICNet-pytorch-master/datasets/Cityscapes" #数据集文件目录epochs: 200               #数据集全部训练一次为一个epochlog_iter: 10        # print log every log-iter 迭代10次记录一下val_epoch: 1        # run validation every val-epochckpt_dir: "./ckpt/" # ckpt and trainning log will be saved here 模型训练完成后的保存路径
4.Test
test:ckpt_path: "./ckpt/icnet_resnet50-19c8e357.pth"  #加载模型,可以是网上训练好的,也可以是自己训练的,路径和名字要准确

首先,下载工程代码文件夹,然后将数据集放入合适位置,一般看代码的read.md文件,有上述的脚本文件,在里面更改自己的路径,然后安装对应的依赖库,这个看remd.md文件或者代码文件夹里有requirement文件,里面是需要安装的库。最后运行指令,训练。


本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!

相关文章

立即
投稿

微信公众账号

微信扫一扫加关注

返回
顶部