【文件属性】:
文件名称:autonomous-object-detection:自主导航系统的目标检测
文件大小:823KB
文件格式:ZIP
更新时间:2024-03-31 14:55:08
系统开源
自主导航的目标检测
该存储库提供了对导航数据集执行对象检测的核心支持。 对3D对象检测和域自适应的支持处于试验阶段,并将在以后添加。 该项目为培训,评估,推理,可视化提供支持。
此仓库还包含以下代码:
exp
新:预训练模型现在可用
先决条件
火炬> = 1.1
火炬视觉> = 0.3
tensorboardX(可选,可视化是必需的)
数据集
这项工作为以下数据集提供支持(与自主导航的对象检测有关):
目录结构:
+-- data
| +-- bdd100k
| +-- IDD_Detection
| +-- cityscapes
+-- autonmous-object-detection
.......
入门
下载所需的数据集
在cfg.py设置数据集路径
创建数据列表
开始培训和评估
文献资料
设置配置
默认情况下,所有路径和超参数均从cfg.py加载。 要求用户
【文件预览】:
autonomous-object-detection-master
----inference.ipynb(169KB)
----evaluation_baseline.py(4KB)
----transforms.py(861B)
----imports.py(623B)
----train_baseline.py(3KB)
----engine.py(4KB)
----utils.py(9KB)
----eval_idd_bdd.py(6KB)
----exp()
--------optimal_transport.ipynb(13KB)
--------evaluate_script.py(6KB)
--------exp.ipynb(465KB)
--------evaluation_transport.py(6KB)
--------train_script.py(3KB)
----assets()
--------eval_baseline_idd.png(602KB)
----LICENSE(1KB)
----coco_utils.py(9KB)
----README.md(6KB)
----get_datalists.py(16KB)
----coco_eval.py(12KB)
----datasets()
--------idd.py(3KB)
--------bdd.py(3KB)
--------cityscapes.py(3KB)
--------.ipynb_checkpoints()
----cfg.py(501B)