Skip to content

Commit c2cc43e

Browse files
committed
add requirements file
1 parent 28165a4 commit c2cc43e

File tree

3 files changed

+25
-1
lines changed

3 files changed

+25
-1
lines changed

.gitignore

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,4 +94,5 @@ ENV/
9494

9595

9696
# Project files
97-
.vscode
97+
.vscode
98+
.idea

README.md

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,17 @@
1+
# Fork of [ZHKKKe/MODNet](https://github.com/ZHKKKe/MODNet)
2+
3+
Differences between original repository and fork:
4+
5+
* Installation with [requirements.txt](requirements.txt) file.
6+
7+
# Installation
8+
9+
```shell
10+
pip install -r requirements.txt
11+
```
12+
13+
14+
115
<h2 align="center">MODNet: Trimap-Free Portrait Matting in Real Time</h2>
216

317
<div align="center"><i>MODNet: Real-Time Trimap-Free Portrait Matting via Objective Decomposition (AAAI 2022)</i></div>

requirements.txt

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
torch>=2.0.0
2+
torchvision>=0.15.0
3+
opencv-python>=4.7.0
4+
tqdm>=4.65.0
5+
scipy>=1.10.0
6+
7+
### Optional
8+
# onnx>=1.14.0
9+
# onnxruntime>=1.15.0

0 commit comments

Comments
 (0)