Snap_tag

整合包地址~希望帮到您

通过网盘分享的文件：Snaptag.7z 链接: https://pan.baidu.com/s/1PZOvpi1P2vb9hKc_JsYhMw?pwd=snap 提取码: snap

概述

Snaptag 是一个基于图像处理和标签生成的 Python 应用程序。该项目使用 PyTorch、Transformers 和 PyQt5 等库来进行图像批量处理，并通过MiniCPM-v2.6-int模型生成相应的标签。它支持对大量图像的自动标记，并通过简洁的用户界面提供高效的用户体验

Snaptag is a Python application based on image processing and tag generation. The project uses libraries such as PyTorch, Transformers, and PyQt5 to perform batch image processing and generate corresponding tags using the MiniCPM-v2.6-int model. It supports automatic tagging for large volumes of images and provides an efficient user experience through a simple user interface

Updated version（版本更新）

截至到2024/8/27我解决了整合包环境错误问题！

截至到2024/8/26我添加了整合包，大家可以下载百度网盘及夸克网盘的整合版本

截至到2024/8/26我添加了三个功能，“top_k,top_p,temperature”，你可以自定义这些数值的大小来确定返回的文字偏向 （准确性 // 想象力）

As of 2024/8/26, I have added three features: "top_k, top_p, temperature." You can customize the values of these parameters to determine the bias of the generated text (accuracy or creativity).

Parameter parsing（参数解析）

top-k 选择固定数量的前 k 个词汇。 top-p 是根据累积概率选择足够的词汇，直到达到指定的概率阈值。 Temperature 控制了生成文本时的随机性

top-k selects a fixed number of top k words. top-p selects enough words based on cumulative probability until a specified probability threshold is reached. Temperature controls the randomness of generated text

UI

抖音：https://v.douyin.com/ih1RjPJx/ 3@7.com :4pm

使用方法（Installation Steps）

克隆仓库：

git clone https://github.com/SS-snap/Snaptag.git

下载MiniCPM-v2.6-int4至 Snaptag\models\model

(Download MiniCPM-v2.6-int4 to Snaptag\models\model) 请下载模型和相关配置文件至指定目录 Please download the model and related configuration files to the specified directory 模型地址：https://huggingface.co/openbmb/MiniCPM-V-2_6-int4/tree/main

定位并创建环境

cd Snaptag

创建虚拟环境： 建议使用 virtualenv 或 conda 创建一个虚拟环境来隔离依赖项：

virtualenv venv
source venv/bin/activate

或者使用 conda：

conda create -n snaptag_env python=3.10
conda activate snaptag_env

安装依赖（Install dependencies）

pip install -r requirements.txt

找到对应自己的cuda版本进行推理

example pip install torch==2.1.2+cu118 torchvision==0.16.2+cu118 --index-url https://download.pytorch.org/whl/cu118

启动（Start the application）

在终端输入python run.py

找到我