remove call to upsample

mlmed · Jan 3, 2025 · f21df39 · f21df39
1 parent 09bafae
commit f21df39
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/torchxrayvision/baseline_models/chexpert/__init__.py b/torchxrayvision/baseline_models/chexpert/__init__.py
@@ -57,7 +57,9 @@ def __init__(self, weights_zip="", num_models=30):
 
     def forward(self, x):
         x = x.repeat(1, 3, 1, 1)
-        x = self.upsample(x)
+
+        x = utils.fix_resolution(x, 320, self)
+        utils.warn_normalization(x)
 
         # expecting values between [-1024,1024]
         x = x / 512
@@ -79,6 +81,7 @@ def forward(self, x):
 
     def features(self, x):
         x = x.repeat(1, 3, 1, 1)
+
         x = utils.fix_resolution(x, 320, self)
         utils.warn_normalization(x)