Сделал тут по другому нейронку из туториала по торчу, решил выложить сюда готовый код:!unzip /content/PennFudanPed.zip
prediction
Image.fromarray(img.mul(255).permute(1, 2, 0).byte().numpy())
Image.fromarray(prediction[0]['masks'][0, 0].mul(255).byte().cpu().numpy())