mirror of
https://github.com/aljazceru/InvSR.git
synced 2025-12-17 06:14:22 +01:00
29 lines
729 B
Python
29 lines
729 B
Python
#!/usr/bin/env python
|
|
# -*- coding:utf-8 -*-
|
|
# Power by Zongsheng Yue 2021-12-07 21:37:58
|
|
|
|
import cv2
|
|
import numpy as np
|
|
|
|
def modcrop(im, sf):
|
|
h, w = im.shape[:2]
|
|
h -= (h % sf)
|
|
w -= (w % sf)
|
|
return im[:h, :w,]
|
|
|
|
#-----------------------------------------Transform--------------------------------------------
|
|
class Bicubic:
|
|
def __init__(self, scale=None, out_shape=None, matlab_mode=True):
|
|
self.scale = scale
|
|
self.out_shape = out_shape
|
|
|
|
def __call__(self, im):
|
|
out = cv2.resize(
|
|
im,
|
|
dsize=self.out_shape,
|
|
fx=self.scale,
|
|
fy=self.scale,
|
|
interpolation=cv2.INTER_CUBIC,
|
|
)
|
|
return out
|