update config
parent
e410d2e1c9
commit
ea557c50ee
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue