my_mask: 自实现的pytoch版本的.对mask-rcnn调用修改配置文件即可运行内含一个自实现的mask-rcnn模型

1月18日，北京，聊聊2025如何加入技术开发？

forked from Tomcat/my_mask

加入 Gitee

与超过 1200万开发者一起发现、参与优秀开源项目，私有仓库也完全免费：）

免费加入

该仓库未声明开源许可证文件（LICENSE），使用请关注具体项目描述及其代码上游依赖。

克隆/下载

分支 1

标签 0

1.修正错误 3b25732

15 次提交

model

check.py

coco_eval.py

coco_utils.py

config.py

engine.py

group_by_aspect_ratio.py

my_transforms.py

readme.md

run.py

utils.py

README

自定义调用torch_version的mask-rcnn项目

这是自实现的文档说明: [pytorch中创建maskrcnn模型](https://blog.csdn.net/qq_37293230/article/details/138127095)

注意!!!
启动方法有2个,一个在run模块,一个在main模块
run模块是调用pytorch官方实现的maskrcnn
main模块调用的是model目录下自实现的maskrcnn(没写完呢)(学习参考使用)
二者使用同一个config(使用main运行时会忽略大部分的参数,因为参数写在模型上了,为了阅读学习清晰)

启动报错可能是缺一个out目录,手动建立一下就好

注意!!!!
可以直接修改
torchvision\models\detection\roi_heads.py 716行的代码,使之nms与分类无关
keep = box_ops.batched_nms(boxes, scores, labels, self.nms_thresh)->
keep = box_ops.batched_nms(boxes, scores, torch.ones_like(labels), self.nms_thresh)
默认调用实现了一个默认的nms之后再进行一次类型无关的nms

自实现的pytoch版本的.对mask-rcnn调用修改配置文件即可运行内含一个自实现的mask-rcnn模型

取消

暂无发行版

马建仓 AI 助手

尝试更多

代码解读

代码找茬

代码优化

https://gitee.com/pottedplant/my_mask.git

[email protected]:pottedplant/my_mask.git

pottedplant

my_mask

master

周盆栽/my_mask

简介

发行版

贡献者

近期动态

周盆栽/my_mask .gitee-modal { width: 500px !important; }

简介

发行版

贡献者

近期动态

搜索帮助

周盆栽/my_mask