В ./sam лежат ноутбуки экспериментов с моделями SAM
Colab для inpainting с помощью классической SD 1.5: https://colab.research.google.com/drive/1WBN67Lre6zxQDZtntoPzEjeEC-ruaVxY?usp=sharing
Colab с конвертацией этой модели в onnx: https://colab.research.google.com/drive/1Wmi6Uc6TcgDGeJLv4J-UfcjtSPoZOofr?usp=sharing
Colab с inpainting для Consistence Models (эта модель не имеет релевантных весов, и работает плохо на большинстве изображений): https://colab.research.google.com/drive/1rh40K2AIeaLcR64XCZO-eAA8c2gBKlN_?usp=sharing
Encoder и Decoder модели в model_repository, пример клиента в client.py
Веса на диске в ./mobile_sam
Интерфейс позволяет работать с изображениями, отправлять данные на сервер и принимать их.
Пример(prompt - picture):
Исходное изображение
Изменённое изображение
python app.py
- MobileSAM - https://github.com/ChaoningZhang/MobileSAM