Commit 5c012b7f authored by zhanggezhong's avatar zhanggezhong
Browse files

Update README.md

parent d57cc392
Pipeline #2688 failed with stages
in 0 seconds
# <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)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment