Repository navigation
In 331_fine_tune_SAM_mito.ipynb, the class SAMDataset(Dataset) needs to be adjusted to have 3 bands to go into the processor #81
Description
The demo no longer works.
Change this line in the class SAMDataset(Dataset)
# prepare image and prompt for the model inputs = self.processor(image, input_boxes=[[prompt]], return_tensors="pt")
to
# prepare image and prompt for the model inputs = self.processor(np.stack((image,)*3, axis=-1), input_boxes=[[prompt]], return_tensors="pt")
Activity
ZoreAnuj commentedon Mar 12, 2024
Thanks man!
VaishnaviMudaliar commentedon Mar 15, 2024
I spent a week trying to figure this out. Thanks for the solution!
RoyAn2386 commentedon Mar 17, 2024
Thanks man~~
ajaidevanath commentedon May 19, 2024
Still having issues
ajaidevanath commentedon May 19, 2024
ERROR:"setting an array element with a sequence. The requested array has an inhomogeneous shape after 1 dimensions. The detected shape was (2,) + inhomogeneous part"