BiaPyX / BiaPy

Open source Python library for building bioimage analysis pipelines
https://BiaPyX.github.io
MIT License
114 stars 28 forks source link

Support Torchvision models #38

Closed danifranco closed 6 months ago

danifranco commented 7 months ago

Study and implement the possibility of using Torchvision models. Special attention to the number of channel and input shape of the networks, maybe we need to control only that.

danifranco commented 6 months ago

Done in 6fc0622!