(self, img)
| 256 | self.eigvec = torch.Tensor(eigvec) |
| 257 | |
| 258 | def __call__(self, img): |
| 259 | if self.alphastd == 0: |
| 260 | return img |
| 261 | |
| 262 | alpha = img.new().resize_(3).normal_(0, self.alphastd) |
| 263 | rgb = self.eigvec.type_as(img).clone() \ |
| 264 | .mul(alpha.view(1, 3).expand(3, 3)) \ |
| 265 | .mul(self.eigval.view(1, 3).expand(3, 3)) \ |
| 266 | .sum(1).squeeze() |
| 267 | |
| 268 | return img.add(rgb.view(3, 1, 1).expand_as(img)) |
| 269 | |
| 270 | |
| 271 | class CutoutDefault(object): |