From f2ced8b8179dbcd40bb7f5b63536d9e96771c3d9 Mon Sep 17 00:00:00 2001 From: Jaedong Date: Fri, 8 Jan 2021 16:19:45 +0900 Subject: [PATCH] [fix] bug --- lib/tools.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/tools.py b/lib/tools.py index 071c295..00fdc6d 100755 --- a/lib/tools.py +++ b/lib/tools.py @@ -141,9 +141,9 @@ def __init__(self, checkpoint, use_cuda, half_precision, im_fe_ratio = 16, postp self.model = ImMatchNet(use_cuda=use_cuda, multi_gpu=multi_gpu, half_precision=half_precision, checkpoint=checkpoint) self.model.eval() - self.feature_extractor = ExtractFeatureMap(im_fe_ratio=self.im_fe_ratio, device=postprocess_device) self.postprocess_device = postprocess_device self.im_fe_ratio = im_fe_ratio + self.feature_extractor = ExtractFeatureMap(im_fe_ratio=self.im_fe_ratio, device=postprocess_device) def __call__(self, batch, num_pts=2000, iter_step=1000, central_align = True, *args, **kwargs): '''