Spaces:
Runtime error
Runtime error
zejunyang
commited on
Commit
•
b79bc32
1
Parent(s):
938daee
update
Browse files- NTED/NTED_module.py +1 -1
- requirements.txt +11 -2
NTED/NTED_module.py
CHANGED
@@ -3,7 +3,7 @@ import numpy as np
|
|
3 |
import torch
|
4 |
import random
|
5 |
|
6 |
-
import mediapipe as mp
|
7 |
from lite_openpose.body_bbox_detector import BodyPoseEstimator
|
8 |
from NTED.extraction_distribution_model import Generator
|
9 |
from NTED.demo_dataset import DemoDataset
|
|
|
3 |
import torch
|
4 |
import random
|
5 |
|
6 |
+
# import mediapipe as mp
|
7 |
from lite_openpose.body_bbox_detector import BodyPoseEstimator
|
8 |
from NTED.extraction_distribution_model import Generator
|
9 |
from NTED.demo_dataset import DemoDataset
|
requirements.txt
CHANGED
@@ -1,4 +1,13 @@
|
|
1 |
-
opencv-python
|
2 |
torch>=1.12
|
3 |
torchvision
|
4 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
torch>=1.12
|
2 |
torchvision
|
3 |
+
gradio
|
4 |
+
collections
|
5 |
+
imageio==2.10.3
|
6 |
+
lmdb==1.2.1
|
7 |
+
opencv-python==4.5.4.58
|
8 |
+
Pillow==8.3.2
|
9 |
+
PyYAML==5.4.1
|
10 |
+
scikit-image==0.17.2
|
11 |
+
scipy==1.5.4
|
12 |
+
tensorboard==2.6.0
|
13 |
+
tqdm==4.62.3
|