timm

timm — это библиотека, содержащая модели компьютерного зрения уровня SOTA, слои, утилиты, оптимизаторы, планировщики, загрузчики данных, аугментации, а также скрипты обучения и оценки.

Он поставляется с более чем 700 предобученными моделями и разработан так, чтобы быть гибким и удобным в использовании.

Ознакомьтесь с руководством для быстрого старта, чтобы начать работу с библиотекой timm. Вы узнаете, как загружать, находить и использовать предобученные модели, входящие в состав библиотеки.

Update on GitHub