одно изображение, которое уверенно распознаётся как собака, и теперь хочу выбрать какой-нибудь пиксель, и оптимизировать 3 его канала так, чтобы увеличить уверенность нейросети в том, что это самолёт. Как отдать один пиксель в оптимизатор на pytoch?
# одно изображение
__batch__ = batch[16:17, :, :, :].numpy()
__batch__ = torch.tensor(__batch__, dtype=torch.float32, requires_grad=True, device='cuda')
# Хотим оптимизировать только 1 пиксель
optimizer = optim.Adam([__batch__[0, 0:3, 10, 10]], lr=5e-3) # ValueError: can't optimize a non-leaf Tensor
Оптимизируй число и подставляй его в картинку
Обсуждают сегодня