Centernet训练自己的数据集--真(的)菜

之前没有什么经验,前前后后搞了两周.实惨

一 环境搭建

参考1  

本人系统:ubantu16.04 + cuda9.0+cudnn7.6.4

 鉴于参考博文以及真大佬的详细讲解,依照本人经验做以下建议:

1. 强烈建议安装Anaconda!创建虚拟环境!
2. 依照参考1的命令执行安装环境.
3. 遇到的问题:

• 1.  执行conda activate 时
• 报错: CommandNotFoundError: Your shell has not been properly configured to use 'conda activate'.
      If your shell is Bash or a Bourne variant, enable conda for the current user with 
• 解决方法:执行以下两个命令:source activate
                                               source deactivate  即可
• 2.pytorch 版本为1.0以上会出现问题,强烈建议按照大佬环境配置.
• 3.cuda的版本问题,同上

二 训练自己的数据集

    参考:参考1

    补充:

    1.看到下图22行将原来的COCO改为自己的数据集名.