复现无人机的项目,项目名称为Evidential Detection and Tracking Collaboration

项目名称为Evidential Detection and Tracking Collaboration,主要用于强大的反无人机系统,涉及新问题、基准和算法研究。下面介绍项目的复现步骤:

  1. 安装环境 :使用Anaconda创建并激活名为edtc的虚拟环境,Python版本为3.6,然后执行bash install_pytorch17.sh脚本安装相关依赖。
bash 复制代码
conda create -n edtc python=3.6
conda activate edtc
bash install_pytorch17.sh
  1. 训练YOLO检测器
    • 编辑数据集设置文件/path/to/EDTC/yolov5/data/antiuav.yaml,配置数据集路径等信息。
    • 进入/path/to/EDTC/yolov5目录,执行python train.py开始训练YOLO检测器。
bash 复制代码
cd /path/to/EDTC/yolov5
python train.py
  1. 训练跟踪分支
    • 运行python tracking/create_default_local_file.py --workspace_dir . --data_dir ./data --save_dir .命令设置项目路径,之后可通过编辑lib/train/admin/local.py(训练路径相关)、experiments/uavtrack/baseline.yaml(阶段1训练路径相关)、experiments/uavtrack_eh/baseline.yaml(阶段2训练路径相关)文件修改路径。
    • 阶段1训练:执行python tracking/train.py --script uavtrack --config baseline --save_dir . --mode multiple --nproc_per_node 8命令进行训练。
    • 阶段2训练:执行python tracking/train.py --script uavtrack_eh --config baseline --save_dir /PATH/TO/SAVE/UAVTRACK_EH --mode multiple --nproc_per_node 8 --stage1_model /STAGE1/MODEL/PATH命令进行训练,需指定阶段1模型的路径。
  2. 在AntiUAV600数据集上评估
    • 从[Models and Raw results](Google Driver)下载预训练模型。
    • 编辑lib/test/evaluation/local.py(测试路径相关)和experiments/uavtrack_eh/baseline.yaml(YOLO预训练模型路径,修改其中第133 - 134行)文件,设置相关路径。
    • 设置环境变量:
bash 复制代码
export PYTHONPATH=$PYTHONPATH:/path/to/EDTC
export PYTHONPATH=$PYTHONPATH:/path/to/EDTC/yolov5
复制代码
- 运行评估命令:`python tracking/test.py uavtrack_eh baseline --dataset antiuav --threads 32 --num_gpus 8 --params__model /path/to/UAVTrackEH.pth.tar --params__search_area_scale 4.55`。若要可视化跟踪结果,需编辑`/path/to/EDTC/lib/test/evaluation/environment.py`文件的第27行,将`self.show_result`设置为`True`。
- 执行`python tracking/evaluate_antiuav_performance.py`命令获取评估指标。

在复现过程中,若数据集尚未发布,需等待数据集发布后获取并按要求放置在指定路径。同时,确保各路径设置正确,若涉及多GPU训练,需保证GPU环境正常且符合代码要求。

相关推荐
说私域1 小时前
互联网生态下赢家群体的崛起与“开源AI智能名片链动2+1模式S2B2C商城小程序“的赋能效应
人工智能·小程序·开源
董厂长4 小时前
langchain :记忆组件混淆概念澄清 & 创建Conversational ReAct后显示指定 记忆组件
人工智能·深度学习·langchain·llm
G皮T8 小时前
【人工智能】ChatGPT、DeepSeek-R1、DeepSeek-V3 辨析
人工智能·chatgpt·llm·大语言模型·deepseek·deepseek-v3·deepseek-r1
九年义务漏网鲨鱼8 小时前
【大模型学习 | MINIGPT-4原理】
人工智能·深度学习·学习·语言模型·多模态
元宇宙时间8 小时前
Playfun即将开启大型Web3线上活动,打造沉浸式GameFi体验生态
人工智能·去中心化·区块链
开发者工具分享8 小时前
文本音频违规识别工具排行榜(12选)
人工智能·音视频
产品经理独孤虾8 小时前
人工智能大模型如何助力电商产品经理打造高效的商品工业属性画像
人工智能·机器学习·ai·大模型·产品经理·商品画像·商品工业属性
老任与码9 小时前
Spring AI Alibaba(1)——基本使用
java·人工智能·后端·springaialibaba
蹦蹦跳跳真可爱5899 小时前
Python----OpenCV(图像増强——高通滤波(索贝尔算子、沙尔算子、拉普拉斯算子),图像浮雕与特效处理)
人工智能·python·opencv·计算机视觉
雷羿 LexChien9 小时前
从 Prompt 管理到人格稳定:探索 Cursor AI 编辑器如何赋能 Prompt 工程与人格风格设计(上)
人工智能·python·llm·编辑器·prompt