Update modeling_fgclip.py
Browse files- modeling_fgclip.py +0 -1
modeling_fgclip.py
CHANGED
|
@@ -149,7 +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 |
|
|
|
|
| 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 |
|
| 153 |
return x_rois
|
| 154 |
|