Jittor / deeplab-jittor

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Jittor DeepLab

  1. 为了更好的运行这份代码,建议先对这份 教程 进行阅读。

  2. 你可以在下方链接处下载 ImageNet pretrain model。

resnet_101

resnet_50

由于我们暂时没有 ImageNet 的 Pre-train model 所以暂时先使用的是 pytorch 的预训练模型。这里也同时体现了我们和 pytorch 是兼容的。模型参数来源

  1. 把数据集的路径和模型的路径分别替换成本地的数据集路径和模型路径。
  2. 运行下面脚本,即可运行。
sh train.sh

About


Languages

Language:Python 99.8%Language:Shell 0.2%