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
95a3fbc4
Commit
95a3fbc4
authored
May 27, 2025
by
zhanggzh
Browse files
add maskrcnn src code and change readme
parent
5c012b7f
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
20 additions
and
18 deletions
+20
-18
NVIDIA/benchmarks/maskrcnn/implementations/pytorch/maskrcnn/maskrcnn_benchmark/__init__.py
...entations/pytorch/maskrcnn/maskrcnn_benchmark/__init__.py
+1
-0
NVIDIA/benchmarks/maskrcnn/implementations/pytorch/maskrcnn/maskrcnn_benchmark/__pycache__/__init__.cpython-310.pyc
...n/maskrcnn_benchmark/__pycache__/__init__.cpython-310.pyc
+0
-0
README.md
README.md
+19
-18
No files found.
NVIDIA/benchmarks/maskrcnn/implementations/pytorch/maskrcnn/maskrcnn_benchmark/__init__.py
View file @
95a3fbc4
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
__version__
=
0.1
NVIDIA/benchmarks/maskrcnn/implementations/pytorch/maskrcnn/maskrcnn_benchmark/__pycache__/__init__.cpython-310.pyc
0 → 100644
View file @
95a3fbc4
File added
README.md
View file @
95a3fbc4
...
...
@@ -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
...
...
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