Update modeling_fgclip.py
Browse files- modeling_fgclip.py +0 -2
modeling_fgclip.py
CHANGED
|
@@ -149,8 +149,6 @@ class FGCLIPModel(CLIPModel):
|
|
| 149 |
feature_map = feature_map.view(bs, h, w, -1).permute(0, 3, 1, 2)
|
| 150 |
x_rois = roi_align(feature_map.type(torch.float32),box_info, (1, 1), 1.0, -1, True)[..., 0, 0]
|
| 151 |
|
| 152 |
-
x_rois = x_rois / x_rois.norm(p=2, dim=-1, keepdim=True)
|
| 153 |
-
|
| 154 |
return x_rois
|
| 155 |
|
| 156 |
def get_text_features(
|
|
|
|
| 149 |
feature_map = feature_map.view(bs, h, w, -1).permute(0, 3, 1, 2)
|
| 150 |
x_rois = roi_align(feature_map.type(torch.float32),box_info, (1, 1), 1.0, -1, True)[..., 0, 0]
|
| 151 |
|
|
|
|
|
|
|
| 152 |
return x_rois
|
| 153 |
|
| 154 |
def get_text_features(
|