5 个仓库
Training processes that associate image data with corresponding text descriptions to improve prompt adherence.
Distinct from Text Model Training: Focuses on image-text pair association for generative models rather than general text-only model training.
Explore 5 awesome GitHub repositories matching artificial intelligence & ml · Caption-Based Training. Refine with filters or upvote what's useful.
This project is a toolkit for fine-tuning and managing text-to-image diffusion models. It focuses on low-rank adaptation to create small, portable weight files that customize model styles and behaviors without modifying the entire base model. The project provides specialized utilities for model distillation using singular value decomposition to extract adapters from fully trained models, as well as tools for blending and merging multiple adapters through weight interpolation. It includes capabilities for subject inversion and pivotal tuning to increase the visual fidelity of specific identiti
Implements training capabilities that link images with text descriptions to improve generation accuracy.
BLIP is a vision-language model framework that combines contrastive, matching, and language modeling objectives to align images with text. Built on a multimodal encoder-decoder architecture, it supports distributed data-parallel training with cosine learning rate scheduling and sliding-window metric tracking for training stability. The framework provides capabilities for image captioning, visual question answering, and cross-modal retrieval, scoring semantic alignment between images and text through learned embeddings. It includes toolkits for fine-tuning pre-trained models on custom datasets
Trains vision-language models to generate descriptive captions for images using paired image-caption datasets.
Neuraltalk is an automated image captioning system that generates natural language descriptions for images. It utilizes a deep learning model that integrates a pretrained convolutional neural network for visual feature extraction with a recurrent neural network decoder to produce text sequences. The project provides a full workflow for training and evaluating captioning models, including weight optimization via backpropagation and gradient descent. It includes tools for measuring caption accuracy by comparing generated text against reference descriptions. The system covers data preprocessing
Optimizes model parameters to predict sentence descriptions by associating image features with ground-truth text.
Kolors 是一个生成式模型实现,用于根据自然语言描述和视觉参考合成逼真的图像。它利用潜在扩散模型框架来生成高保真图像,并在压缩的潜在空间内运行以提高生成效率和质量。 该系统作为一个多语言图像生成器,能够解释多种语言的文本提示词,从而产生语义准确的视觉输出。它包含一个自定义模型训练流水线,使用低秩自适应(LoRA)技术,通过少量图像集教会模型特定的主题或艺术风格。 该项目涵盖了广泛的图像合成和编辑功能,包括文生图和图生图转换。它提供了通过深度图或姿态图进行空间布局控制的工具、用于美学一致性的视觉身份注入,以及基于掩码的修复(Inpainting)以重构或修改特定图像区域。 该实现包括图像质量评估工具,根据人类偏好指标对生成图像的美学和语义质量进行评分。
Provides the ability to interpret text prompts in multiple languages to produce semantically accurate visual outputs.
This project is a research framework and toolkit designed for training large-scale vision transformers and multimodal language models. It provides a comprehensive suite for vision-language pretraining, enabling the development of models that map images and text into shared latent spaces. The framework is distinguished by its capabilities in high-fidelity image generation and multimodal research, utilizing normalizing flows and variational autoencoders to produce images from text prompts or class labels. It supports the development of both generative and contrastive models, allowing for a wide
Maps images and text into a shared space using captioning-based pretraining and self-supervised losses.