ZiyuG commited on
Commit
1b44373
·
verified ·
1 Parent(s): ebe9e30

Update evaluate.py

Browse files
Files changed (1) hide show
  1. evaluate.py +1 -1
evaluate.py CHANGED
@@ -1,5 +1,5 @@
1
  import cv2, math
2
- import json, os
3
  import numpy as np
4
  from sklearn.preprocessing import Normalizer
5
  from align import align_filter
 
1
  import cv2, math
2
+ import json, os, torch
3
  import numpy as np
4
  from sklearn.preprocessing import Normalizer
5
  from align import align_filter