Итак, мой вопрос довольно прост:
- У меня есть несколько ограничивающих рамок, образованных:
[TopLeft_X, TopLeft_Y,
BottomRight_X, BottomRight_Y] - А также у меня есть тензорная маска размером
[900,
1600] Итак, вот первые 3 ограничивающих прямоугольника, которые у меня есть:
tensor([[1041, 465, 1102, 508],
[1037, 468, 1108, 519],
[1041, 465, 1107, 515]], device='cuda:0', dtype=torch.int32)
По сути, я хочу получить количество пикселей (оценивается как 1), что каждый ограничивающий прямоугольник пересекается с маской.
В настоящее время я просматриваю все ограничивающие прямоугольники и получаю соответствующие срезы как:
map_mask [y_min: y_max, x_min: x_max] -> Вывод в 2D
Как я могу сделать это одновременно без for l oop в PyTorch?