提交 ea557c50 编写于 作者: D dorren

update config

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