ICnet模型复现

数据集存放路径 datasets-Cityscapes- 

 

 

 

train:
  specific_gpu_num: "0"     # for example: "0", "1" or "0, 1" 单显卡默认是0
  train_batch_size: 1      # adjust according to gpu resources 笔记本显存一般是4g，基本上是设置最小为1，可以试
  valid_batch_size: 1
  cityscapes_root: "/home/s/Desktop/ICNet-pytorch-master/ICNet-pytorch-master/datasets/Cityscapes" #数据集文件目录
  epochs: 200               #数据集全部训练一次为一个epoch
  log_iter: 10        # print log every log-iter 迭代10次记录一下
  val_epoch: 1        # run validation every val-epoch
  ckpt_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文件，里面是需要安装的库。最后运行指令，训练。