(self, data, prefix='A photo depicts')
| 65 | |
| 66 | class CLIPCapDataset(torch.utils.data.Dataset): |
| 67 | def __init__(self, data, prefix='A photo depicts'): |
| 68 | self.data = data |
| 69 | self.prefix = prefix |
| 70 | if self.prefix[-1] != ' ': |
| 71 | self.prefix += ' ' |
| 72 | |
| 73 | def __getitem__(self, idx): |
| 74 | c_data = self.data[idx] |
nothing calls this directly
no outgoing calls
no test coverage detected