noelshin commited on
Commit
f10bd9c
1 Parent(s): 35188e4

add requirements.txt

Browse files
Files changed (3) hide show
  1. .idea/vcs.xml +6 -0
  2. bilateral_solver.py +0 -12
  3. requirements.txt +5 -0
.idea/vcs.xml ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ <?xml version="1.0" encoding="UTF-8"?>
2
+ <project version="4">
3
+ <component name="VcsDirectoryMappings">
4
+ <mapping directory="$PROJECT_DIR$" vcs="Git" />
5
+ </component>
6
+ </project>
bilateral_solver.py CHANGED
@@ -1,20 +1,8 @@
1
  from scipy.sparse import diags
2
  from scipy.sparse.linalg import cg
3
  from scipy.sparse import csr_matrix
4
- import numpy as np
5
- from skimage.io import imread
6
  from scipy import ndimage
7
- import torch
8
- import PIL.Image as Image
9
- import os
10
- from argparse import ArgumentParser, Namespace
11
- from typing import Dict, Union
12
- from collections import defaultdict
13
- import yaml
14
- import ujson as json
15
  import numpy as np
16
- import torch
17
- import torch.nn.functional as F
18
  from PIL import Image
19
 
20
 
 
1
  from scipy.sparse import diags
2
  from scipy.sparse.linalg import cg
3
  from scipy.sparse import csr_matrix
 
 
4
  from scipy import ndimage
 
 
 
 
 
 
 
 
5
  import numpy as np
 
 
6
  from PIL import Image
7
 
8
 
requirements.txt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ opencv-python==4.6.0.66
2
+ torch==1.9.1+cpu
3
+ torchvision==0.10.1+cpu
4
+ timm==0.5.4
5
+ scipy