fbpx
  • Imaginaire: библиотека на PyTorch с генеративными моделями

    Imaginaire — это библиотека на PyTorch с генеративными моделями для задач генерации изображений и видеозаписей. Библиотеку разрабатывают исследователи из NVIDIA. Imaginaire содержит оптимизированные имплементации state-of-the-art моделей, основанных на GAN.

    Подробнее про библиотеку

    Библиотека включает в себя набор supervised и unsupervised моделей для image-to-image генерации. Кроме того, в библиотеке содержатся аналогичные модели для video-to-video перевода. Исследователи из NVIDIA включили по несколько state-of-the-art архитектур для каждой задачи. Например, pix2pix и SPADE — для image-to-image перевода; UNIT, MUNIT и FUNIT — для unsupervised перевода из одного изображения в другое; vid2vid, fs-vid2vid и wc-vid2vid — для перевода из одного видео в другое.

    Согласно разработчикам, библиотека упрощает процесс обучения генеративных моделей. Для каждой модели из библиотеки доступен тьюториал по использованию. Кроме того, модели из библиотеки отдельно имеют предобученные веса в открытом доступе.

    Imaginaire выложили под лицензией NVIDIA Software License. Библиотека доступна бесплатно для некоммерческого использования.