Morne W comments

Repositories
Issues
Comments

Results 3 comments of


                                            Morne W

albumentations

I can confirm, version 0.5.2 works for me. Thanks @marrylin2019 !

How can I predict segmentation results after training?

This is the predict.py code that works for me, thanks @didnttaken ``` import torch from torchvision.transforms import ToTensor from PIL import Image import archs import numpy as np # Load...

How do I train on different image dimenstions e.g 192 x 192 instead of 96 x 96 pixels?

Example files: ![1](https://github.com/4uiiurz1/pytorch-nested-unet/assets/12203806/a47115f2-10c9-4b2f-9e47-325245641aef) ![1](https://github.com/4uiiurz1/pytorch-nested-unet/assets/12203806/1605ad4b-46c9-4395-acc8-a5bc9eddae77)