zejunyang commited on
Commit
b79bc32
1 Parent(s): 938daee
Files changed (2) hide show
  1. NTED/NTED_module.py +1 -1
  2. 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
- mediapipe
 
 
 
 
 
 
 
 
 
 
 
 
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