提交 75674710 编写于 作者: D dangqingqing

Remove opencv-python in setup

上级 0e358ccc
import numpy as np import numpy as np
try: try:
import cv2 import cv2
except ImportError: except:
cv2 = None print(
"import cv2 error, please install opencv-python: pip install opencv-python"
from cv2 import resize )
__all__ = [ __all__ = [
"load_image", "resize_short", "to_chw", "center_crop", "random_crop", "load_image", "resize_short", "to_chw", "center_crop", "random_crop",
...@@ -76,7 +76,7 @@ def resize_short(im, size): ...@@ -76,7 +76,7 @@ def resize_short(im, size):
h_new = size * h / w h_new = size * h / w
else: else:
w_new = size * w / h w_new = size * w / h
im = resize(im, (h_new, w_new), interpolation=cv2.INTER_CUBIC) im = cv2.resize(im, (h_new, w_new), interpolation=cv2.INTER_CUBIC)
return im return im
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册