update args in visualize_featuremap.py

HaoyiZhu · HaoyiZhu · commit 1fcc14d190ff · 2025-05-26T23:20:40.000+08:00
diff --git a/scripts/visualize_featuremap.py b/scripts/visualize_featuremap.py
@@ -46,13 +46,17 @@ def main():
     ]
     images = np.stack(
         [
-            (resize(image, (448, 448), anti_aliasing=True) * 255).astype(np.uint8)
+            (
+                resize(image, (args.img_size, args.img_size), anti_aliasing=True) * 255
+            ).astype(np.uint8)
             for image in images
         ]
     )
     # images = np.stack(images) # / 255.
     images = torch.from_numpy(images).float().permute(0, 3, 1, 2).to(device) / 255.0
-    images = torch.nn.functional.interpolate(images, size=(448, 448), mode="bilinear")
+    images = torch.nn.functional.interpolate(
+        images, size=(args.img_size, args.img_size), mode="bilinear"
+    )
     # n c h w
     feature_map = model(images, feature_map=True, cat_cls=False)
     feature_map = torch.nn.functional.interpolate(

Original file line number	Diff line number	Diff line change
`@@ -46,13 +46,17 @@ def main():`
`46`	`46`	`]`
`47`	`47`	`images = np.stack(`
`48`	`48`	`[`
`49`		`- (resize(image, (448, 448), anti_aliasing=True) * 255).astype(np.uint8)`
	`49`	`+ (`
	`50`	`+ resize(image, (args.img_size, args.img_size), anti_aliasing=True) * 255`
	`51`	`+ ).astype(np.uint8)`
`50`	`52`	`for image in images`
`51`	`53`	`]`
`52`	`54`	`)`
`53`	`55`	`# images = np.stack(images) # / 255.`
`54`	`56`	`images = torch.from_numpy(images).float().permute(0, 3, 1, 2).to(device) / 255.0`
`55`		`- images = torch.nn.functional.interpolate(images, size=(448, 448), mode="bilinear")`
	`57`	`+ images = torch.nn.functional.interpolate(`
	`58`	`+ images, size=(args.img_size, args.img_size), mode="bilinear"`
	`59`	`+ )`
`56`	`60`	`# n c h w`
`57`	`61`	`feature_map = model(images, feature_map=True, cat_cls=False)`
`58`	`62`	`feature_map = torch.nn.functional.interpolate(`