(self, data, append=False, prefix='A photo depicts')
| 18 | |
| 19 | class CLIPCapDataset(torch.utils.data.Dataset): |
| 20 | def __init__(self, data, append=False, prefix='A photo depicts'): |
| 21 | self.data = data |
| 22 | self.prefix = '' |
| 23 | if append: |
| 24 | self.prefix = prefix |
| 25 | if self.prefix[-1] != ' ': |
| 26 | self.prefix += ' ' |
| 27 | |
| 28 | def __getitem__(self, idx): |
| 29 | c_data = self.data[idx] |
nothing calls this directly
no outgoing calls
no test coverage detected