提交 f17b96b9 编写于 作者: W WenmuZhou

Delete useless code

上级 f2ebeae5
...@@ -310,34 +310,3 @@ class TPS(nn.Layer): ...@@ -310,34 +310,3 @@ class TPS(nn.Layer):
[-1, image.shape[2], image.shape[3], 2]) [-1, image.shape[2], image.shape[3], 2])
batch_I_r = F.grid_sample(x=image, grid=batch_P_prime) batch_I_r = F.grid_sample(x=image, grid=batch_P_prime)
return batch_I_r return batch_I_r
if __name__ == '__main__':
import paddle
from ppocr.utils.save_load import load_dygraph_pretrain
from ppocr.utils.logging import get_logger
np.random.seed(0)
img = np.random.random((1, 3, 32, 100)).astype(np.float32)
x = paddle.to_tensor(img)
batch_P_prime = np.load(
'/Users/zhoujun20/Desktop/code/static/PaddleOCR/output/tps1.npy')
batch_P_prime = paddle.to_tensor(batch_P_prime)
model = TPS(in_channels=3, num_fiducial=20, loc_lr=0.1, model_name='small')
load_dygraph_pretrain(
model,
get_logger(),
'/Users/zhoujun20/Desktop/code/static/PaddleOCR/output/tps',
load_static_weights=True)
model.eval()
y = model(x)
print(y.shape)
static = np.load(
'/Users/zhoujun20/Desktop/code/static/PaddleOCR/output/tps.npy')
print(static.dtype, y.numpy().dtype, static.shape, y.numpy().shape)
diff = y.numpy() - static
print(diff.max())
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册