liangfeng commited on
Commit
07bb7b3
1 Parent(s): e66e6e4
Files changed (1) hide show
  1. requirements.txt +4 -1
requirements.txt CHANGED
@@ -6,4 +6,7 @@ h5py
6
  wandb
7
  fire
8
  opencv-python
9
- pandas
 
 
 
 
6
  wandb
7
  fire
8
  opencv-python
9
+ pandas
10
+ torch==1.10.1
11
+ torchvision==0.11.2
12
+ detectron2==0.6