Сделал тут по другому нейронку из туториала по торчу, решил выложить сюда готовый код:
!unzip /content/PennFudanPed.zip
1.1K1.1K показов
9595 открытий
Сделал тут по другому нейронку из туториала по торчу, решил выложить сюда готовый код:
!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())