python练习 双线性插值对图像进行伸缩变换

import numpy as np
import cv2
##双线性插值
def resize(src,dstsize):#输入src 和size
	if src.ndim==3:
		dstsize.append(3)
	dst=np.array(np.zeros(dstsize),src.dtype)
	factory=float(np.size(src,0))/dstsize[0] 
	factorx=float(np.size(src,1))/dstsize[1]
	print 'factory',factory,'factorx',factorx
	srcheight=np.size(src,0)
	srcwidth=np.size(src,1)
	print 'srcwidth',srcwidth,'srcheight',srcheight
	for i in range(dstsize[0]):
		for j in range(dstsize[1]):
			y=float(i)*factory
			x=float(j)*factorx
			if y+1>srcheight:  #越界判断
				y-=1
			if x+1>srcwidth:
				x-=1 
			cy=np.ceil(y)
			fy=cy-1
			cx=np.ceil(x)
			fx=cx-1
			w1=(cx-x)*(cy-y)
			w2=(x-fx)*(cy-y)
			w3=(cx-x)*(y-fy)
			w4=(x-fx)*(y-fy)	
			if (x-np.floor(x)>1e-6 or y-np.floor(y)>1e-6): 
				t=src[fy,fx]*w1+src[fy,cx]*w2+src[cy,fx]*w3+src[cy,cx]*w4
				t=np.ubyte