FSCE 运行COCO2017数据集环境搭建

把FSCE克隆到本地

git clone https://github.com/megvii-research/FSCE.git

准备好COCO2017数据集

这一步就不多说了，直接去下载，记得路径在哪就行

修改几个文件

主要是修改这几个文件

1.修改builtin.py文件

把原来35行开始的代码注释掉，换成自己的数据命名，自己的数据集地址和json（从55行的代码展示）
接下来是这部分，还是这个文件

把这部分换成下面这部分

还是和上面换的差不多，自己模仿看看就行了

我自己就没用1，2，3，5的了，因为是COCO数据集，可以不改，后面改成2017
这样这个文件就改完了

2.接下来就是修改meta_coco.py文件

这部分就是把目录换成140行这样，自己的COCO数据集保存地址。

3.修改prepare_coco_few_shot.py文件

就是生成一堆txt文件

修改成自己的数据集路径

可选同上

修改完后点击运行该文件，生成txt文件

然后任选一个种子{}，如seed9，复制到上级目录，即cocosplit

这样的话，基本修改完毕，剩下的就是yaml文件修改了。

第一阶段训练

比较简单，就是修改base_trainning.yml文件

把数据集这部分换成自己的数据集命名

好像我设置的名字没有变化，和原来一样。

可以运行第一阶段的训练了

python -m tools/train_net.py --num-gpus 1
–config-file configs/COCO/R-101/base_training.yml
如果出现以下错误：
(dzpEnvs) sh-4.3$ python tools/train_net.py --num-gpus 1 --config-file configs/COCO/R-101/base_training.yml
Traceback (most recent call last):
File “tools/train_net.py”, line 23, in
import fsdet.utils.comm as comm
ModuleNotFoundError: No module named ‘fsdet.utils.comm’
就需要添加
import sys
sys.path.append(“/home/data/dzp/FSCE”)
在报错的文件中添加这两句
路径是你这项目的根目录/home/data/dzp/FSCE，自己修改