search

lkeab / gaussian-grouping

[ECCV'2024] Gaussian Grouping for open-world Anything reconstruction, segmentation and editing.
https://arxiv.org/abs/2312.00732
Apache License 2.0
607 stars 43 forks source link

How to change the size of the feature vector (currently 16) #41

Open RonnyVelastegui2018 opened 3 months ago

RonnyVelastegui2018 commented 3 months ago

I am trying to modify the size of the feature vector. By default, it is 16. I am trying to use a feature vector of 3 for example.

I modified the file "gaussian-grouping/scene/gaussian_model.py", by changing line 52: self.num_objects = 3

But, when running the code I get this error:

RuntimeError: Given groups=1, weight of size [150, 3, 1, 1], expected input[1, 16, 479, 639] to have 3 channels, but got 16 channels instead

What other part of the code should I modify in order to use a feature vector of 3?

ymq2017 commented 2 months ago

Hi, thanks for watching our work!

For changing dim of the feature vector, you can refer to this issue.