fix fopa.bug

bcmi · Nov 20, 2023 · 36ff260 · 36ff260
1 parent 2d9f3e5
commit 36ff260
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/libcom/fopa_heat_map/fopa_heat_map.py b/libcom/fopa_heat_map/fopa_heat_map.py
@@ -64,7 +64,7 @@ def __init__(self, device=0, model_type='fopa', **kwargs):
         self.device = check_gpu_device(device)
         fopa_weight = os.path.join(cur_dir, 'pretrained_models', 'FOPA.pth')
         download_pretrained_model(fopa_weight)
-        sopa_weight = os.path.join(cur_dir, 'pretrained_models', 'SOPA.pth.tar')
+        sopa_weight = os.path.join(cur_dir, 'pretrained_models', 'SOPA.pth')
         download_pretrained_model(sopa_weight)
         self.build_pretrained_model(sopa_weight, fopa_weight)
         self.build_data_transformer()

diff --git a/setup.py b/setup.py
@@ -107,7 +107,7 @@ def get_ext_modules():
     cur_dir = os.path.dirname(os.path.abspath(__file__))
     setup(
         name='libcom',
-        version='0.0.1.post7',
+        version='0.0.1.post8',
         description='Image Composition Toolbox',
         long_description=readme(),
         long_description_content_type='text/markdown',