From cc86da9006c62b0c38c3df1c623f3c43099128d2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E6=B4=8B=E6=B4=8B?= <584244991@qq.com>
Date: Fri, 16 Dec 2022 03:01:40 +0000
Subject: [PATCH 1/4] update
 TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/utils/datasets.py.
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: 张洋洋 <584244991@qq.com>
---
 .../contrib/cv/UltraFast_for_TensorFlow2/utils/datasets.py      | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/utils/datasets.py b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/utils/datasets.py
index da95e1c50..d0266f398 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, num_parallel_calls=64)
+    ds = ds.map(processor.process_json, num_parallel_calls=32)
     return ds
 
     
-- 
Gitee


From 099c51cfae13d3d6564e0e17a22c564a066a61ad Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E6=B4=8B=E6=B4=8B?= <584244991@qq.com>
Date: Fri, 16 Dec 2022 03:03:26 +0000
Subject: [PATCH 2/4] update
 TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/train_ultrafast.py.
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: 张洋洋 <584244991@qq.com>
---
 .../train_ultrafast.py                        | 22 +++++++++----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/train_ultrafast.py b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/train_ultrafast.py
index d0d38ad16..5c44a1342 100644
--- a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/train_ultrafast.py
+++ b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/train_ultrafast.py
@@ -28,8 +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.global_options().op_compiler_cache_mode="enable"
+npu.global_options().op_compiler_cache_dir="/mnt/home/test_user08/UltraFast_NPU/my_kernel_cache"
 npu.open().as_default()
 
 import tensorflow as tf
@@ -45,15 +45,15 @@ from utils.datasets import llamas_dataset, labelme_dataset
 
 parser = argparse.ArgumentParser(description='Train Ultrafast Net')
 parser.add_argument('--epochs', type=int, default=1)
-parser.add_argument('--batch-size', type=int, default=16)
-parser.add_argument('--max-images', type=int, default=None)
-parser.add_argument('--prefetch-size', type=int, default=256)
-parser.add_argument('--resnet-weights', default=None)
-parser.add_argument('--base-path', default='')
-parser.add_argument('--llamas-path', '--data-path', default='./data/llamas/')
-parser.add_argument('--labelme-path', default=[], action='append')
-parser.add_argument('--output-path', default='./')
-parser.add_argument('--model-name', default='ultrafast')
+parser.add_argument('--batch_size', type=int, default=16)
+parser.add_argument('--max_images', type=int, default=None)
+parser.add_argument('--prefetch_size', type=int, default=256)
+parser.add_argument('--resnet_weights', default=None)
+parser.add_argument('--base_path', default='')
+parser.add_argument('--llamas_path', '--data_path', default='./data/llamas/')
+parser.add_argument('--labelme_path', default=[], action='append')
+parser.add_argument('--output_path', default='./')
+parser.add_argument('--model_name', default='ultrafast')
 
 args = parser.parse_args()
 print(args)
-- 
Gitee


From 09c06136b033eb434f418569e157eb28b8bed395 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E6=B4=8B=E6=B4=8B?= <584244991@qq.com>
Date: Fri, 16 Dec 2022 03:11:52 +0000
Subject: [PATCH 3/4] update
 TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_performance_1p.sh.
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: 张洋洋 <584244991@qq.com>
---
 .../cv/UltraFast_for_TensorFlow2/test/train_performance_1p.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_performance_1p.sh b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_performance_1p.sh
index 81418499e..37d00d1e1 100644
--- a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_performance_1p.sh
+++ b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_performance_1p.sh
@@ -114,13 +114,13 @@ batch_size=16
 if [ x"${modelarts_flag}" != x ];
 then
     python3.7 ./train_ultrafast.py \
-        --data-path=${data_path}/data/llamas \
+        --data_path=${data_path}/data/llamas \
         --output_path=${output_path} \
         --epochs=${train_epochs} \
         --batch_size=${batch_size} 1>${print_log} 2>&1
 else
     python3.7 ./train_ultrafast.py \
-        --data-path=${data_path}/data/llamas \
+        --data_path=${data_path}/data/llamas \
         --output_path=${output_path} \
         --epochs=${train_epochs} \
         --batch_size=${batch_size} 1>${print_log} 2>&1
-- 
Gitee


From 39c1905f1037dd2d4e09766f0cfd59c7c50348be Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E6=B4=8B=E6=B4=8B?= <584244991@qq.com>
Date: Fri, 16 Dec 2022 03:12:34 +0000
Subject: [PATCH 4/4] update
 TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_full_1p.sh.
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: 张洋洋 <584244991@qq.com>
---
 .../cv/UltraFast_for_TensorFlow2/test/train_full_1p.sh        | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_full_1p.sh b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_full_1p.sh
index 81418499e..37d00d1e1 100644
--- a/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_full_1p.sh
+++ b/TensorFlow2/contrib/cv/UltraFast_for_TensorFlow2/test/train_full_1p.sh
@@ -114,13 +114,13 @@ batch_size=16
 if [ x"${modelarts_flag}" != x ];
 then
     python3.7 ./train_ultrafast.py \
-        --data-path=${data_path}/data/llamas \
+        --data_path=${data_path}/data/llamas \
         --output_path=${output_path} \
         --epochs=${train_epochs} \
         --batch_size=${batch_size} 1>${print_log} 2>&1
 else
     python3.7 ./train_ultrafast.py \
-        --data-path=${data_path}/data/llamas \
+        --data_path=${data_path}/data/llamas \
         --output_path=${output_path} \
         --epochs=${train_epochs} \
         --batch_size=${batch_size} 1>${print_log} 2>&1
-- 
Gitee