Realcat commited on
Commit
77708bb
·
1 Parent(s): a01f750
imcui/hloc/match_dense.py CHANGED
@@ -312,7 +312,7 @@ confs = {
312
  },
313
  "preprocessing": {
314
  "grayscale": False,
315
- "force_resize": False,
316
  "resize_max": 1024,
317
  "width": 320,
318
  "height": 240,
@@ -330,7 +330,7 @@ confs = {
330
  },
331
  "preprocessing": {
332
  "grayscale": False,
333
- "force_resize": False,
334
  "resize_max": 1024,
335
  "width": 320,
336
  "height": 240,
@@ -348,7 +348,7 @@ confs = {
348
  },
349
  "preprocessing": {
350
  "grayscale": False,
351
- "force_resize": False,
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.matcher.upsample_res = self.conf["upsample_res"]
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):