Error 'numpy.ndarray' object has no attribute 'unsqueeze' when passing bounding boxes into segment_cellular_image

[ellenemerson]

Bounding boxes are converted to ndarray when resized in sam_inference.py (line 234), then unsqueeze is called later as if they are still tensors (line 251). Tested locally by removing the np.array(...) in line 234, and it seems to work - can create a PR if that is helpful.

cellSAM/cellSAM/sam_inference.py

Lines 230 to 251 in a034a51

	if boxes_per_heatmap is None:
	boxes_per_heatmap = self.generate_bounding_boxes(images, device=device)
	else:
	boxes_per_heatmap = (
	np.array(boxes_per_heatmap) * 1024 / max(images[0].shape)
	)
	# B, N, 4
	if not fast:
	boxes_per_heatmap = boxes_per_heatmap[0]
	low_masks = []
	low_masks_thresholded = []
	scores = []
	for input_bbox in boxes_per_heatmap:
	# if fast , passes N, 4
	# else, passes 4
	while len(input_bbox.shape) < 2:
	input_bbox = input_bbox.unsqueeze(0)