Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
maskrcnn
Commits
5c012b7f
Commit
5c012b7f
authored
May 06, 2025
by
zhanggezhong
Browse files
Update README.md
parent
d57cc392
Pipeline
#2688
failed with stages
in 0 seconds
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
61 additions
and
0 deletions
+61
-0
README.md
README.md
+61
-0
No files found.
README.md
View file @
5c012b7f
# <div align="center"><strong>Mask-RCNN</strong></div>
## 简介
Mask R-CNN 是在 Faster R-CNN 架构的基础上发展而来的,Faster R-CNN 是一种经典的两阶段目标检测算法,主要由骨干网络(Backbone)、区域建议网络(Region Proposal Network, RPN)和检测头(Detection Head)三部分组成。Mask R-CNN 在 Faster R-CNN 的基础上添加了一个用于生成目标掩码(mask)的分支,形成了一个多任务学习框架。
## 安装
### 适用环境
-
ubuntu20.04 或 rocky8.6
-
Python==3.10
-
PyTorch==2.4.1 DTK=25.04
### 使用pip方式安装
```
shell
pip
install
maskrcnn
*
```
### 源码编译安装
#### 编译环境准备
-
拉取maskrcnn代码
```
shell
git clone
-b
maskrcnn-hcu https://developer.sourcefind.cn/codes/OpenDAS/maskrcnn.git
```
-
导入环境变量以及安装必要依赖库
安装fastpt-2.1.0-torch2.4.1版本
```
shell
source
/usr/local/bin/fastpt_cuda_init
--compile
```
使用audio时执行
```
shell
source
/usr/local/bin/fastpt_cuda_init
--execute
```
#### 编译安装
-
执行编译命令并安装
```
shell
cd
NVIDIA/benchmarks/maskrcnn/implementations/pytorch/maskrcnn
python3 setup.py bdist_wheel
pip
install
dist/maskrcnn
*
```
## Known Issue
-
无
## 其他参考
-
[
README_ORIGIN
](
README_ORIGIN.md
)
-
[
GitHub
](
https://github.com/mlcommons/training_results_v3.1
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment