Update main_test_SRMNet.py

main_test_SRMNet.py

@@ -3,6 +3,7 @@ import cv2
 import glob
 import numpy as np
 from collections import OrderedDict
+from skimage import img_as_ubyte
 import os
 import torch
 import requests