forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

去掉注释

This commit is contained in:
WenmuZhou 2020-10-22 18:22:56 +08:00
parent 7c96520de7
commit 08b3f98c4f
1 changed files with 10 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
ppocr/modeling/architectures/model.py
View File

@ -76,7 +76,6 @@ class Model(nn.Layer):
config["Head"]['in_channels'] = in_channels config["Head"]['in_channels'] = in_channels
self.head = build_head(config["Head"]) self.head = build_head(config["Head"])
# @paddle.jit.to_static
def forward(self, x): def forward(self, x):
if self.use_transform: if self.use_transform:
x = self.transform(x) x = self.transform(x)
@ -93,30 +92,29 @@ def check_static():
from ppocr.utils.logging import get_logger from ppocr.utils.logging import get_logger
from tools import program from tools import program
config = program.load_config('configs/rec/rec_mv3_none_none_ctc_lmdb.yml') config = program.load_config('configs/det/det_r50_vd_db.yml')
logger = get_logger() logger = get_logger()
np.random.seed(0) np.random.seed(0)
data = np.random.rand(2, 3, 64, 320).astype(np.float32) data = np.random.rand(1, 3, 640, 640).astype(np.float32)
paddle.disable_static() paddle.disable_static()
x = paddle.to_tensor(data)
config['Architecture']['in_channels'] = 3 config['Architecture']['in_channels'] = 3
config['Architecture']["Head"]['out_channels'] = 6624 config['Architecture']["Head"]['out_channels'] = 37
model = Model(config['Architecture']) model = Model(config['Architecture'])
model.eval() model.eval()
load_dygraph_pretrain( load_dygraph_pretrain(
model, model, logger, 'det_r50_vd_db/best_accuracy', load_static_weights=True)
logger,
'/Users/zhoujun20/Desktop/code/PaddleOCR/cnn_ctc/cnn_ctc',
load_static_weights=True)
x = paddle.to_tensor(data)
y = model(x) y = model(x)
for y1 in y: for y1 in y:
print(y1.shape) print(y1.shape)
static_out = np.load( static_out = np.load('static_out.npy')
'/Users/zhoujun20/Desktop/code/PaddleOCR/output/conv.npy') diff = y.numpy() - static_out
diff = y.reshape((-1, 6624)).numpy() - static_out
print(y.shape, static_out.shape, diff.mean()) print(y.shape, static_out.shape, diff.mean())