ianpan
/

total-classifier

Image Classification

total_classifier

feature-extraction

Model card Files Files and versions

ianpan commited on Feb 25

Commit

0b22878

·

verified ·

1 Parent(s): 1eebdfb

Upload model

Files changed (1) hide show

modeling.py +1 -1

modeling.py CHANGED Viewed

@@ -519,7 +519,7 @@ class TotalClassifierModel(PreTrainedModel):
         if 2 in crop_dims:
             # swap plane
-            x = x0.transpose(2, 0)
             smin2, smax2 = self.crop_single_plane(
                 x, device, organ, threshold, buffer, speed_up
             )

         if 2 in crop_dims:
             # swap plane
+            x = x0.swapaxes(2, 0)
             smin2, smax2 = self.crop_single_plane(
                 x, device, organ, threshold, buffer, speed_up
             )