mv dataset in config

This commit is contained in:
tink2123 2020-05-10 09:18:27 +00:00
parent 56c6c3ae0e
commit 476c51a73c
3 changed files with 14 additions and 4 deletions

View File

@ -1,6 +1,5 @@
Global:
algorithm: CRNN
dataset: common
use_gpu: true
epoch_num: 300
log_smooth_window: 20

12
set_env.sh Normal file
View File

@ -0,0 +1,12 @@
#. /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↩

View File

@ -79,9 +79,8 @@ def main():
'fetch_varname_list':eval_fetch_varname_list}
metrics = eval_det_run(exe, config, eval_info_dict, "test")
else:
dataset = config['Global']['dataset']
assert dataset in ['lmdb', 'common']
if dataset == 'common':
reader_type = config['Global']['reader_yml']
if "chinese" in reader_type:
eval_reader = reader_main(config=config, mode="eval")
eval_info_dict = {'program': eval_program, \
'reader': eval_reader, \