提交 5492c10a 编写于 作者: L LDOUBLEV

fix comments

上级 c7131fec
...@@ -4,12 +4,11 @@ Global: ...@@ -4,12 +4,11 @@ Global:
epoch_num: 500 epoch_num: 500
log_smooth_window: 20 log_smooth_window: 20
print_batch_step: 10 print_batch_step: 10
save_model_dir: ./output/ch_PP-OCR_V3/ save_model_dir: ./output/ch_PP-OCR_v3_det/
save_epoch_step: 100 save_epoch_step: 100
eval_batch_step: eval_batch_step:
- 0 - 0
- 400 - 400
load_static_weights: false
cal_metric_during_train: false cal_metric_during_train: false
pretrained_model: null pretrained_model: null
checkpoints: null checkpoints: null
...@@ -137,6 +136,7 @@ Train: ...@@ -137,6 +136,7 @@ Train:
img_mode: BGR img_mode: BGR
channel_first: false channel_first: false
- DetLabelEncode: null - DetLabelEncode: null
- CopyPaste:
- IaaAugment: - IaaAugment:
augmenter_args: augmenter_args:
- type: Fliplr - type: Fliplr
......
...@@ -4,12 +4,11 @@ Global: ...@@ -4,12 +4,11 @@ Global:
epoch_num: 500 epoch_num: 500
log_smooth_window: 20 log_smooth_window: 20
print_batch_step: 10 print_batch_step: 10
save_model_dir: ./output/ch_PP-OCR_V3/ save_model_dir: ./output/ch_PP-OCR_V3_det/
save_epoch_step: 100 save_epoch_step: 100
eval_batch_step: eval_batch_step:
- 0 - 0
- 400 - 400
load_static_weights: false
cal_metric_during_train: false cal_metric_during_train: false
pretrained_model: null pretrained_model: null
checkpoints: null checkpoints: null
......
...@@ -66,7 +66,7 @@ class Head(nn.Layer): ...@@ -66,7 +66,7 @@ class Head(nn.Layer):
self.conv3 = nn.Conv2DTranspose( self.conv3 = nn.Conv2DTranspose(
in_channels=in_channels // 4, in_channels=in_channels // 4,
out_channels=1, out_channels=1,
kernel_size=kernel_size[1], kernel_size=kernel_size[2],
stride=2, stride=2,
weight_attr=ParamAttr( weight_attr=ParamAttr(
initializer=paddle.nn.initializer.KaimingUniform()), initializer=paddle.nn.initializer.KaimingUniform()),
......
...@@ -20,7 +20,7 @@ import paddle ...@@ -20,7 +20,7 @@ import paddle
from paddle import nn from paddle import nn
import paddle.nn.functional as F import paddle.nn.functional as F
from paddle import ParamAttr from paddle import ParamAttr
import os
__dir__ = os.path.dirname(os.path.abspath(__file__)) __dir__ = os.path.dirname(os.path.abspath(__file__))
sys.path.append(__dir__) sys.path.append(__dir__)
sys.path.insert(0, os.path.abspath(os.path.join(__dir__, '../../..'))) sys.path.insert(0, os.path.abspath(os.path.join(__dir__, '../../..')))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册