Spaces:
Running
on
Zero
Running
on
Zero
add normalization
Browse files
app.py
CHANGED
@@ -60,6 +60,13 @@ def siglip_detector(image, texts):
|
|
60 |
return probs
|
61 |
|
62 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
63 |
def infer_siglip(image, candidate_labels):
|
64 |
candidate_labels = [label.lstrip(" ") for label in candidate_labels.split(",")]
|
65 |
siglip_out = siglip_detector(image, candidate_labels)
|
|
|
60 |
return probs
|
61 |
|
62 |
|
63 |
+
def normalize_tensor(tensor):
|
64 |
+
# no other normalization works well for visual purposes
|
65 |
+
sum_tensor = torch.sum(tensor)
|
66 |
+
normalized_tensor = tensor / sum_tensor
|
67 |
+
|
68 |
+
return normalized_tensor
|
69 |
+
|
70 |
def infer_siglip(image, candidate_labels):
|
71 |
candidate_labels = [label.lstrip(" ") for label in candidate_labels.split(",")]
|
72 |
siglip_out = siglip_detector(image, candidate_labels)
|