update config

pull/7940/head
dorren 2022-10-17 19:58:35 +08:00
parent e410d2e1c9
commit ea557c50ee
3 changed files with 9 additions and 10 deletions

View File

@ -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

View File

@ -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

View File

@ -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