mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2026-09-24 23:33:08 +08:00
mv dataset in config
This commit is contained in:
@@ -1,6 +1,5 @@
|
||||
Global:
|
||||
algorithm: CRNN
|
||||
dataset: common
|
||||
use_gpu: true
|
||||
epoch_num: 300
|
||||
log_smooth_window: 20
|
||||
|
||||
+12
@@ -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↩
|
||||
+2
-3
@@ -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, \
|
||||
|
||||
Reference in New Issue
Block a user