chensjtu/GaussianObject

Rabbit dataset

seulqxq opened this issue · 2 comments

seulqxq commented

Hi,
I want to train using the rabbit dataset, using the mask generated by segment_anything.ipynb, but running
python pred_poses.py -s data/realcap/rabbit --sparse_num 4
gets wrong.
image
Here is the data:
image

TXSevenXT commented

I've got this problem :(
For the moment, no solution :(

TXSevenXT commented

Find a solution in deleting "0" in the code :
original_masks = [np.array(Image.open(mask).resize(image.size))[:, :, 0] / 255.0 for mask, image in zip(masks, original_images)]
Becomes original_masks = [np.array(Image.open(mask).resize(image.size))[:, :,] / 255.0 for mask, image in zip(masks, original_images)]

I will try with my others pictures and tell you if it's ok ^^