Spaces:
Build error
Build error
edits
Browse files
app.py
CHANGED
@@ -65,6 +65,7 @@ def match(query_feat, pos_feat, LoweRatioTh=0.9):
|
|
65 |
dist = torch.cdist(query_feat, pos_feat)
|
66 |
best1 = torch.argmin(dist, dim=1)
|
67 |
best2 = torch.argmin(dist, dim=0)
|
|
|
68 |
arange = torch.arange(best2.size(0))
|
69 |
reciprocal = best1[best2]==arange
|
70 |
# check Lowe ratio test
|
@@ -245,7 +246,7 @@ iface = gr.Interface(
|
|
245 |
gr.inputs.Slider(minimum=0, maximum=6, step=1, default=2, label="Scale"),
|
246 |
gr.inputs.Slider(minimum=1, maximum=255, step=25, default=100, label="Binarization Threshold"),
|
247 |
gr.inputs.Textbox(lines=1, default="", label="SF IDs to show (comma separated numbers from 0-255; typing 'rX' will return X random SFs", optional=True),
|
248 |
-
gr.inputs.Checkbox(default=True, label="Show only matching", optional=False),
|
249 |
],
|
250 |
outputs=[
|
251 |
gr.outputs.Image(type="plot", label="First Image SFs"),
|
|
|
65 |
dist = torch.cdist(query_feat, pos_feat)
|
66 |
best1 = torch.argmin(dist, dim=1)
|
67 |
best2 = torch.argmin(dist, dim=0)
|
68 |
+
print('best2.size',best2.size)
|
69 |
arange = torch.arange(best2.size(0))
|
70 |
reciprocal = best1[best2]==arange
|
71 |
# check Lowe ratio test
|
|
|
246 |
gr.inputs.Slider(minimum=0, maximum=6, step=1, default=2, label="Scale"),
|
247 |
gr.inputs.Slider(minimum=1, maximum=255, step=25, default=100, label="Binarization Threshold"),
|
248 |
gr.inputs.Textbox(lines=1, default="", label="SF IDs to show (comma separated numbers from 0-255; typing 'rX' will return X random SFs", optional=True),
|
249 |
+
gr.inputs.Checkbox(default=True, label="Show only matching SFs", optional=False),
|
250 |
],
|
251 |
outputs=[
|
252 |
gr.outputs.Image(type="plot", label="First Image SFs"),
|