picselliahq / picsellia-training-engine

MIT License
2 stars 0 forks source link

feat: train UNet with segmentation dataset #33

Closed hajeressef closed 11 months ago

hajeressef commented 11 months ago

Support training UNet using segmentation datasets (with polygons rather than masks)

The PR checks if it's a mask-based or polygon-based training experiment using the attached dataset versions' names which we will add as constraints to the models.

On Picsellia, we will have two models for UNet: UNet-segmentation and UNet-masks. They will share the same docker image, but the latter has the image and mask prefixes added as parameters, for mask and image matching purposes.