Spaces:
Running
Running
Realcat
commited on
Commit
·
77708bb
1
Parent(s):
a01f750
fix: roma
Browse files- imcui/hloc/match_dense.py +3 -3
- imcui/hloc/matchers/roma.py +1 -1
imcui/hloc/match_dense.py
CHANGED
@@ -312,7 +312,7 @@ confs = {
|
|
312 |
},
|
313 |
"preprocessing": {
|
314 |
"grayscale": False,
|
315 |
-
"force_resize":
|
316 |
"resize_max": 1024,
|
317 |
"width": 320,
|
318 |
"height": 240,
|
@@ -330,7 +330,7 @@ confs = {
|
|
330 |
},
|
331 |
"preprocessing": {
|
332 |
"grayscale": False,
|
333 |
-
"force_resize":
|
334 |
"resize_max": 1024,
|
335 |
"width": 320,
|
336 |
"height": 240,
|
@@ -348,7 +348,7 @@ confs = {
|
|
348 |
},
|
349 |
"preprocessing": {
|
350 |
"grayscale": False,
|
351 |
-
"force_resize":
|
352 |
"resize_max": 1024,
|
353 |
"width": 320,
|
354 |
"height": 240,
|
|
|
312 |
},
|
313 |
"preprocessing": {
|
314 |
"grayscale": False,
|
315 |
+
"force_resize": True,
|
316 |
"resize_max": 1024,
|
317 |
"width": 320,
|
318 |
"height": 240,
|
|
|
330 |
},
|
331 |
"preprocessing": {
|
332 |
"grayscale": False,
|
333 |
+
"force_resize": True,
|
334 |
"resize_max": 1024,
|
335 |
"width": 320,
|
336 |
"height": 240,
|
|
|
348 |
},
|
349 |
"preprocessing": {
|
350 |
"grayscale": False,
|
351 |
+
"force_resize": True,
|
352 |
"resize_max": 1024,
|
353 |
"width": 320,
|
354 |
"height": 240,
|
imcui/hloc/matchers/roma.py
CHANGED
@@ -57,7 +57,7 @@ class Roma(BaseModel):
|
|
57 |
device=device,
|
58 |
amp_dtype=amp_dtype,
|
59 |
)
|
60 |
-
self.
|
61 |
logger.info("Load Roma model done.")
|
62 |
|
63 |
def _forward(self, data):
|
|
|
57 |
device=device,
|
58 |
amp_dtype=amp_dtype,
|
59 |
)
|
60 |
+
self.net.upsample_res = self.conf["upsample_res"]
|
61 |
logger.info("Load Roma model done.")
|
62 |
|
63 |
def _forward(self, data):
|