diff --git a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/README.md b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/README.md index fda58942fdc91d2049c6bae9acdbba22bd9b33b3..2d24c7c41d59a53cf20d0cebfcf148fa596b435c 100644 --- a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/README.md +++ b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/README.md @@ -75,6 +75,12 @@ UltraFast是基于点回归的方式进行车道线识别的网络模型 |---------------|-------|-------| | UltraFast Acc | 99.7% | 99.7% | +- 性能结果比对 + +| 性能指标项 | GPU实测 | NPU实测 | +|---------------|-------|-------| +| FPS | 36.17 |61.76 | +

高级参考

diff --git a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/train_ultrafast.py b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/train_ultrafast.py index 582a886367151dd3173f09e096cbe2dc492885f8..d0d38ad1643cb1fc20a42b91d040ed010d515854 100644 --- a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/train_ultrafast.py +++ b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/train_ultrafast.py @@ -28,6 +28,8 @@ import npu_device as npu +npu_device.global_options().op_compiler_cache_mode="enable" +npu_device.global_options().op_compiler_cache_dir="/mnt/home/test_user08/UltraFast_NPU/my_kernel_cache" npu.open().as_default() import tensorflow as tf diff --git a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/utils/datasets.py b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/utils/datasets.py index 2ede39232253aee5b89be74e351692df6626b8b0..da95e1c5077ce4b33c66d29ab0d597d052419227 100644 --- a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/utils/datasets.py +++ b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/utils/datasets.py @@ -189,7 +189,7 @@ def load_json_dataset(json_file_pattern, processor, max_records=None, shuffle_si if premap_func is not None: ds = premap_func(ds) - ds = ds.map(processor.process_json) + ds = ds.map(processor.process_json, num_parallel_calls=64) return ds