Как нарезать изображения факела в виде обалденных изображений

Question

Я работаю над проблемой, в которой у меня есть координаты для нарезки изображения, такие как

X кордината, координата Y, высота, ширина области обрезки

Так что, если у меня есть изображение факела, полученное с использованием

img = Variable(img.cuda())

, как мы можем нарезать это изображение, чтобы получить определенную область изображения [y: y + высота, x: x + ширина] .Спасибо

Answer 1

Я получил решение, используя эту запись

img[:, :, y:y+height, x:x+width]

Таким образом, на выходе будет изображение факела с измененным размером.Спасибо

Answer 2

Если я правильно понимаю ваш вопрос, то вы можете сделать это так же, как в numpy.

Вот краткий пример:

import torch
t = torch.rand(5, 5)
# original matrix
print(t)
h = 2
w = 2
x = 1
y = 1
# cropped out matrix
print(t[x:x+h, y:y+w])

Вывод:

tensor([[ 0.5402,  0.4106,  0.9904,  0.9556,  0.2217],
        [ 0.4533,  0.6300,  0.5352,  0.2710,  0.4307],
        [ 0.6389,  0.5660,  0.1582,  0.5701,  0.1614],
        [ 0.1717,  0.4071,  0.4960,  0.2127,  0.5587],
        [ 0.9529,  0.2865,  0.6667,  0.7401,  0.3372]])
tensor([[ 0.6300,  0.5352],
        [ 0.5660,  0.1582]])

Как видите, матрица 2x2 обрезана из t.