OpenGVLab
/

internimage_t_1k_224

Image Classification

feature-extraction

Model card Files Files and versions Community

czczup commited on 13 days ago

Commit

5bbdaf8

·

verified ·

1 Parent(s): 7b940a2

Upload models

Files changed (1) hide show

modeling_internimage.py +2 -2

modeling_internimage.py CHANGED Viewed

@@ -853,8 +853,8 @@ class InternImageModel(PreTrainedModel):
             remove_center=config.remove_center,  # for InternImage-H/G
         )
-    def forward(self, tensor):
-        return self.model.forward_features(tensor)
 class InternImageModelForImageClassification(PreTrainedModel):

             remove_center=config.remove_center,  # for InternImage-H/G
         )
+    def forward(self, pixel_values):
+        return self.model.forward_features(pixel_values)
 class InternImageModelForImageClassification(PreTrainedModel):