提交 476c51a7 编写于 作者: T tink2123

mv dataset in config

上级 56c6c3ae
Global: Global:
algorithm: CRNN algorithm: CRNN
dataset: common
use_gpu: true use_gpu: true
epoch_num: 300 epoch_num: 300
log_smooth_window: 20 log_smooth_window: 20
......
#. /paddle/set_env.sh↩
export CUDA_VISIBLE_DEVICES="0,1,2,3"
export PYTHONPATH=$PYTHONPATH:.↩
export FLAGS_fraction_of_gpu_memory_to_use=1.0↩
python_bin_dir="/opt/_internal/cpython-3.7.0/bin/"
alias python=$python_bin_dir"python3.7"
alias pip=$python_bin_dir"pip3.7"
alias ipython=$python_bin_dir"ipython3"
export LD_LIBRARY_PATH=/opt/_internal/cpython-3.7.0/lib:$LD_LIBRARY_PATH
export PYTHONPATH=$PYTHONPATH:.↩
ldconfig↩
...@@ -79,9 +79,8 @@ def main(): ...@@ -79,9 +79,8 @@ def main():
'fetch_varname_list':eval_fetch_varname_list} 'fetch_varname_list':eval_fetch_varname_list}
metrics = eval_det_run(exe, config, eval_info_dict, "test") metrics = eval_det_run(exe, config, eval_info_dict, "test")
else: else:
dataset = config['Global']['dataset'] reader_type = config['Global']['reader_yml']
assert dataset in ['lmdb', 'common'] if "chinese" in reader_type:
if dataset == 'common':
eval_reader = reader_main(config=config, mode="eval") eval_reader = reader_main(config=config, mode="eval")
eval_info_dict = {'program': eval_program, \ eval_info_dict = {'program': eval_program, \
'reader': eval_reader, \ 'reader': eval_reader, \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册