上传者: 42131628
|
上传时间: 2021-11-29 16:54:16
|
文件大小: 10.83MB
|
文件类型: -
yolo-midas自主导航
将YOLOv3与MiDaS与单个Resnext101主干结合。 想法是针对两个不同的应用程序使用单个特征提取器,在这种情况下为对象检测和单眼深度估计。
请阅读文章以了解更多详细信息
可以在model/mde_net.py看到模型架构的更改
训练
该模型在“建筑安全装备数据”上进行了训练,可在此处找到 。 如果需要对自定义数据集进行培训,请参考页面中提到的数据准备步骤。
将数据放在data/customdata/custom.data文件夹中
python3.6 train.py --data data/customdata/custom.data --batch 8 --cache --cfg cfg/mde.conf --epochs 50 --img-size 512
请参考配置文件cfg/mde.cfg更改网络配置,冻结不同的分支。 该模型是YO