Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Detection Issues with Cylindrical Object in Custom Dataset #129

Open
Septomide opened this issue Dec 2, 2024 · 0 comments
Open

Detection Issues with Cylindrical Object in Custom Dataset #129

Septomide opened this issue Dec 2, 2024 · 0 comments

Comments

@Septomide
Copy link

Hi,

I am working with a custom dataset containing multiple objects. Most objects are detected well, but I am experiencing issues with the detection of a cylindrical object. It seems to perform poorly, likely due to symmetry-related challenges.

roi, roi_mask_trunc, roi_region, .. look the same like they do for other objects. The sym_info is set for this object, and is of shape (942, 3, 3). The sym_info set in the models_info.json:

{
  # rotation around z, cylinder is upright
  "symmetries_continuous": [
    {
      "axis": [0, 0, 1],
      "offset": [0, 0, 0]
    }
  ],
  # rotation 180 around x and y 
  "symmetries_discrete": [ 
    [
      1, 0, 0, 0,
      0, -1, 0, 0,
      0, 0, -1, 0,
      0, 0, 0, 1
    ],
    [
      -1, 0, 0, 0,
      0, 1, 0, 0,
      0, 0, -1, 0,
      0, 0, 0, 1
    ]
  ]
}

Image samples given from the dataloader of the cylinder object:
image_roi_2
image_2_roi_mask_visib
image_2_roi_mask_full
image_2_region

Results after training (center, rotation, and scale of the cylinder are completely off):
full
c1
c2

Did I miss anything obvious? Any guidance on improving detection for symmetrical objects would be greatly appreciated.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant