Announcing TinyMachines.jl

Cheers,

Happy to announce my first Julia package, TinyMachines.jl.

Module contains compact machine learning models written in Flux.jl for semantic segmentation of images. All of them provide access to internal feature outputs at several layers, which have been useful for me in distilling knowledge from Metalhead.jl models to the models in this package: U-Net (standard and compact flavor), Mobile-Unet, and ESPNet.

Feedbacks and improvement tips are welcome. Hope it helps.

5 Likes