马春杰杰博客
致力于深度学习经验分享!

Detectron2使用自定义数据集进行训练

首先找到包含Trainer这个类的train.py文件,然后在这个类前面定义数据集:

这里的__background__记得留着。

上面就把数据集定义好了,接下来要设置后面训练的数据集。找到该文件中的setup方法,加入:

需要注意的是,这三行需要在cfg.freeze()这句之前加入。

最后,在开始训练之前,执行上面我们定义的plain_register_dataset方法。

这样就可以了。

 

如果你对这篇文章有什么疑问或建议,欢迎下面留言提出,我看到会立刻回复!

打赏
未经允许不得转载:马春杰杰 » Detectron2使用自定义数据集进行训练
蝴蝶PT招人啦

留个评论吧~ 抢沙发 评论前登陆可免验证码!

私密评论
  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址(选填,便于回访^_^)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

微信扫一扫打赏

切换注册

登录

忘记密码 ?

切换登录

注册