ianpan
/

ct-crop

Object Detection

feature-extraction

computed_tomography

Model card Files Files and versions Community

ianpan commited on Feb 5

Commit

51ec09c

·

verified ·

1 Parent(s): 39592a1

Upload model

Files changed (1) hide show

modeling.py +1 -1

modeling.py CHANGED Viewed

@@ -328,7 +328,7 @@ class CTCropModel(PreTrainedModel):
         x = self.normalize(x)
         # avg pooling
         features = F.adaptive_avg_pool2d(self.backbone(x), 1).flatten(1)
-        coords = self.linear(features).sigmoid()
         if add_buffer is not None:
             coords = self.add_buffer_to_coords(coords, add_buffer)

         x = self.normalize(x)
         # avg pooling
         features = F.adaptive_avg_pool2d(self.backbone(x), 1).flatten(1)
+        coords = self.linear(self.dropout(features)).sigmoid()
         if add_buffer is not None:
             coords = self.add_buffer_to_coords(coords, add_buffer)