Could you tell me how to test my own images for 3D reconstruction? Do I need to modify the part related to masks in the code?

Traceback (most recent call last):
File "/root/autodl-tmp/sam-3d-objects/demo.py", line 18, in
output = inference(image, mask, seed=42)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/root/autodl-tmp/sam-3d-objects/notebook/inference.py", line 108, in __call__
image = self.merge_mask_to_rgba(image, mask)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/root/autodl-tmp/sam-3d-objects/notebook/inference.py", line 98, in merge_mask_to_rgba
rgba_image = np.concatenate([image[..., :3], mask], axis=-1)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: all the input array dimensions except for the concatenation axis must match exactly, but along dimension 0, the array at index 0 has size 1149 and the array at index 1 has size 4480