README.md

# <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)