search

cviviers / YOLOv5-6D-Pose

6-DoF Pose estimation based on the YOLOv5 framework. Specific focus on instruments in X-ray applications
https://ieeexplore.ieee.org/document/10478293
GNU General Public License v3.0
32 stars 6 forks source link

How to modify my code if I don't have a segmentation Mask Folder #16

Closed tongfengqi closed 2 days ago

tongfengqi commented 3 days ago

I noticed that the segmentation Masks Folder is Optional, if I don't have a Segmentation Mask Folder, how should I modify my code. When I was training, the error message was as follows image

tongfengqi commented 2 days ago

This error should have nothing to do with the mask, but I don't know how to solve it

cviviers commented 2 days ago

Hi @tongfengqi,

The code uses the object initial bounding box width and height in the label files to determine optimal anchor box sizes. Do you have the widths and heights in your label files?

As for as the segmentation goes - yes they are optional, but from personal experience I saw the model generalizes much better if you can train with changing the background.