Update models/tag2text.py

models/tag2text.py

@@ -25,6 +25,8 @@ import numpy as np
 def read_json(rpath):
     with open(rpath, 'r') as f:
         return json.load(f)
+
+delete_tag_index = [135]
         
 class Tag2Text_Caption(nn.Module):
     def __init__(self,                 
@@ -132,6 +134,7 @@ class Tag2Text_Caption(nn.Module):
             targets = torch.where(torch.sigmoid(logits) > self.threshold , torch.tensor(1.0).to(image.device), torch.zeros(self.num_class).to(image.device))
 
             tag = targets.cpu().numpy()
+            tag[:,delete_tag_index] = 0
             bs = image.size(0)
             tag_input = []
             for b in range(bs):