def TF_rotate(x, angle=0.0, target=None):
assert len(x.shape) == 3
h, w, nc = x.shape
# Rotate using edge fill mode
return rotate(x, angle, mode='edge', order=1)
微信
公众号
扫码关注公众号