dontlearncpp / YOLO-DCow

8 stars 0 forks source link

Is this a dual-channel form of YOLOv8? Can RGB and depth maps be input at the input terminal at the same time? #7

Open Gunkill opened 11 months ago

Gunkill commented 11 months ago

So can we replace the depth map with an infrared image?

Gunkill commented 11 months ago

and I can't seem to find your network config file in the code

dontlearncpp commented 11 months ago

The config file is 'ultralytics/models/v8/yolov8n.yaml'. The input is changed to 4 channels which included RGB and depth. The file path of depth images was specified in the readme. As you know, the depth channel can be replaced with any type of data. For my opinion, if the data was aligned with RGB images, will help improve the AP.

dontlearncpp commented 4 weeks ago

It is a dual-modal version of YOLOv8, both the depth and RGB channels were sent into the network! Thank you.