Commit 95a3fbc4 authored by zhanggzh's avatar zhanggzh
Browse files

add maskrcnn src code and change readme

parent 5c012b7f
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
__version__ = 0.1
......@@ -14,11 +14,6 @@ Mask R-CNN 是在 Faster R-CNN 架构的基础上发展而来的,Faster R-CNN
- PyTorch==2.4.1 DTK=25.04
### 使用pip方式安装
```shell
pip install maskrcnn*
```
### 源码编译安装
#### 编译环境准备
......@@ -29,26 +24,32 @@ pip install maskrcnn*
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
pip3 install fastpt-2.0.1+das.dtk2504-py3-none-any.whl #以torch2.4.1,dtk2504为例
source /usr/local/bin/fastpt -c
git checkout v0.1-fastpt
cd NVIDIA/benchmarks/maskrcnn/implementations/pytorch/maskrcnn
python3 setup.py bdist_wheel
pip install dist/maskrcnn*
pip3 install dist/maskrcnn*
```
- 验证安装
使用时执行
```shell
source /usr/local/bin/fastpt -e
```
```shell
python3
Python 3.10.12 (main, Feb 4 2025, 14:57:36) [GCC 11.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import maskrcnn_benchmark
>>> maskrcnn_benchmark.__version__
0.1
>>>
```
## Known Issue
......
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