提交 ea557c50 编写于 作者: D dorren

update config

上级 e410d2e1
...@@ -66,7 +66,6 @@ Loss: ...@@ -66,7 +66,6 @@ Loss:
PostProcess: PostProcess:
name: CANLabelDecode name: CANLabelDecode
character_dict_path: ppocr/utils/dict/latex_symbol_dict.txt
Metric: Metric:
name: CANMetric name: CANMetric
...@@ -76,6 +75,7 @@ Train: ...@@ -76,6 +75,7 @@ Train:
dataset: dataset:
name: SimpleDataSet name: SimpleDataSet
data_dir: ./train_data/CROHME/training/images/ data_dir: ./train_data/CROHME/training/images/
label_file_list: ["./train_data/CROHME/training/labels.txt"]
transforms: transforms:
- DecodeImage: - DecodeImage:
channel_first: False channel_first: False
...@@ -86,11 +86,9 @@ Train: ...@@ -86,11 +86,9 @@ Train:
- GrayImageChannelFormat: - GrayImageChannelFormat:
inverse: True inverse: True
- CANLabelEncode: - CANLabelEncode:
character_dict_path: ppocr/utils/dict/latex_symbol_dict.txt
lower: False lower: False
- KeepKeys: - KeepKeys:
keep_keys: ['image', 'label'] keep_keys: ['image', 'label']
label_file_list: ["./train_data/CROHME/training/labels.txt"]
loader: loader:
shuffle: True shuffle: True
batch_size_per_card: 8 batch_size_per_card: 8
...@@ -102,6 +100,7 @@ Eval: ...@@ -102,6 +100,7 @@ Eval:
dataset: dataset:
name: SimpleDataSet name: SimpleDataSet
data_dir: ./train_data/CROHME/evaluation/images/ data_dir: ./train_data/CROHME/evaluation/images/
label_file_list: ["./train_data/CROHME/evaluation/labels.txt"]
transforms: transforms:
- DecodeImage: - DecodeImage:
channel_first: False channel_first: False
...@@ -112,11 +111,9 @@ Eval: ...@@ -112,11 +111,9 @@ Eval:
- GrayImageChannelFormat: - GrayImageChannelFormat:
inverse: True inverse: True
- CANLabelEncode: - CANLabelEncode:
character_dict_path: ppocr/utils/dict/latex_symbol_dict.txt
lower: False lower: False
- KeepKeys: - KeepKeys:
keep_keys: ['image', 'label'] keep_keys: ['image', 'label']
label_file_list: ["./train_data/CROHME/evaluation/labels.txt"]
loader: loader:
shuffle: False shuffle: False
drop_last: False drop_last: False
......
...@@ -11,6 +11,11 @@ ...@@ -11,6 +11,11 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
"""
This code is refer from:
https://github.com/LBH1024/CAN/models/densenet.py
"""
from __future__ import absolute_import from __future__ import absolute_import
from __future__ import division from __future__ import division
......
...@@ -66,7 +66,6 @@ Loss: ...@@ -66,7 +66,6 @@ Loss:
PostProcess: PostProcess:
name: CANLabelDecode name: CANLabelDecode
character_dict_path: ppocr/utils/dict/latex_symbol_dict.txt
Metric: Metric:
name: CANMetric name: CANMetric
...@@ -76,6 +75,7 @@ Train: ...@@ -76,6 +75,7 @@ Train:
dataset: dataset:
name: SimpleDataSet name: SimpleDataSet
data_dir: ./train_data/CROHME_lite/training/images/ data_dir: ./train_data/CROHME_lite/training/images/
label_file_list: ["./train_data/CROHME_lite/training/labels.txt"]
transforms: transforms:
- DecodeImage: - DecodeImage:
channel_first: False channel_first: False
...@@ -86,11 +86,9 @@ Train: ...@@ -86,11 +86,9 @@ Train:
- GrayImageChannelFormat: - GrayImageChannelFormat:
inverse: True inverse: True
- CANLabelEncode: - CANLabelEncode:
character_dict_path: ppocr/utils/dict/latex_symbol_dict.txt
lower: False lower: False
- KeepKeys: - KeepKeys:
keep_keys: ['image', 'label'] keep_keys: ['image', 'label']
label_file_list: ["./train_data/CROHME_lite/training/labels.txt"]
loader: loader:
shuffle: True shuffle: True
batch_size_per_card: 8 batch_size_per_card: 8
...@@ -102,6 +100,7 @@ Eval: ...@@ -102,6 +100,7 @@ Eval:
dataset: dataset:
name: SimpleDataSet name: SimpleDataSet
data_dir: ./train_data/CROHME_lite/evaluation/images/ data_dir: ./train_data/CROHME_lite/evaluation/images/
label_file_list: ["./train_data/CROHME_lite/evaluation/labels.txt"]
transforms: transforms:
- DecodeImage: - DecodeImage:
channel_first: False channel_first: False
...@@ -112,11 +111,9 @@ Eval: ...@@ -112,11 +111,9 @@ Eval:
- GrayImageChannelFormat: - GrayImageChannelFormat:
inverse: True inverse: True
- CANLabelEncode: - CANLabelEncode:
character_dict_path: ppocr/utils/dict/latex_symbol_dict.txt
lower: False lower: False
- KeepKeys: - KeepKeys:
keep_keys: ['image', 'label'] keep_keys: ['image', 'label']
label_file_list: ["./train_data/CROHME_lite/evaluation/labels.txt"]
loader: loader:
shuffle: False shuffle: False
drop_last: False drop_last: False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册