change object-config.txt

logivations · Jun 11, 2024 · 5910f2d · 5910f2d
1 parent 4e36077
commit 5910f2d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/auto_training/utils/export_specs.py b/auto_training/utils/export_specs.py
@@ -116,9 +116,9 @@ def export_for_lv(args):
     cfg.author = args.author
     export_folder = os.path.join(cfg.work_dir, "export")
     recreate_dir(export_folder)
-    model_name = f"pvt_detector_{cfg.project_name}_{time.strftime('%y%m%d')}"
+    model_name = "pvt_detector"
     write_detector_yaml(cfg=cfg, write_dir=export_folder, name=model_name)
-    write_deepstream_config(cfg=cfg, write_dir=export_folder, name=model_name)
+    #write_deepstream_config(cfg=cfg, write_dir=export_folder, name=model_name)
     write_info_file(cfg=cfg, write_dir=export_folder)
     copy_training_specs(cfg=cfg, write_dir=export_folder)
     print(f"Training info exported successfully to: {export_folder}")

diff --git a/run_make_nvinfer.sh b/run_make_nvinfer.sh
@@ -3,7 +3,6 @@
 MODEL_DIR=$1
 MODEL_DIR=${MODEL_DIR%%/}
 shift
-current_date=$(date +"%d_%m_%Y")
 
 CLASSES=( "$@" )
 CLASSES=$(IFS=';' ; echo "${CLASSES[*]}")
@@ -17,8 +16,8 @@ offsets=123.675;116.28;103.53
 model-color-format=0
 scaling-filter=3 # 0=Nearest, 1=Bilinear 2=VIC-5 Tap interpolation 3=VIC-10 Tap interpolation
 
-onnx-file=pvt_detector_$current_date.onnx
-model-engine-file=pvt_detector_$current_date.onnx_b1_gpu0_fp16.engine
+onnx-file=pvt_detector.onnx
+model-engine-file=pvt_detector.onnx_b1_gpu0_fp16.engine
 
 # model config
 infer-dims=3;384;512