search

FrancescoSaverioZuppichini / glasses

High-quality Neural Networks for Computer Vision 😎
https://francescosaveriozuppichini.github.io/glasses-webapp/
MIT License
431 stars 37 forks source link

[Enhancement Inference + Finetuning Functionality] #277

Open Jacobsolawetz opened 2 years ago

Jacobsolawetz commented 2 years ago

Enhancement

Perhaps this functionality already exists, but I was thinking it would be nice to have AutoModel have something like .finetune(dataset_location) and .infer(input_img)

FrancescoSaverioZuppichini commented 2 years ago

Hi @Jacobsolawetz unfortunately the aim of glasses is just to provide well-written models easy to use but the logic behind how to train them is up to you. I could add a binding to pytorch-lighting to make it easier to finetune them thou. What do you think?