关于mask_rcnn 实现的几点说明

感觉GitHub上Star数最多的竟然不是官方的实现,估计是官方刚开始放出的是基于caffe2 实现的,而当时TensorFlow和keras 大火,所以大家都在寻找基于TensorFlow 和keras 的实现了。后来Facebook官方又放出了基于pytorch的开源实现。确实不错,速度和精度都有很大的提升,之前详细了解了faster rcnn 的一种pytorch实现的版本,就是借鉴了官方pytorch的版本的一些代码。

在win10 上进行配置的时候,需要修改几个cuda 文件。主要的问题是在win10 的pytorch 的API中可能有一两个函数 没有实现。造成编译错误。修改之后,完美编译并且运行。

官网的demo是跑通了,接下来,看下训练和测试的情况怎么样。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值