diff --git a/auto_training/utils/export_specs.py b/auto_training/utils/export_specs.py index 1676c06601c..f836036c87b 100644 --- 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 index 14022465e07..4e1df3c73bb 100644 --- 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