52Hz commited on
Commit
712b739
1 Parent(s): f7c89d9

Update main_test_SRMNet.py

Browse files
Files changed (1) hide show
  1. main_test_SRMNet.py +1 -0
main_test_SRMNet.py CHANGED
@@ -3,6 +3,7 @@ import cv2
3
  import glob
4
  import numpy as np
5
  from collections import OrderedDict
 
6
  import os
7
  import torch
8
  import requests
 
3
  import glob
4
  import numpy as np
5
  from collections import OrderedDict
6
+ from skimage import img_as_ubyte
7
  import os
8
  import torch
9
  import requests