Spaces:
Running
on
Zero
Running
on
Zero
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):
|