This commit is contained in:
student_2333 2024-10-31 19:49:44 +08:00
parent 82434bf3dd
commit 356640b2d2
No known key found for this signature in database
GPG Key ID: 665F083BEC56F2A6
3 changed files with 32 additions and 28 deletions

View File

@ -130,7 +130,7 @@ plugins = [
在 nonebot2 项目的 `.env` 文件中添加下表中的必填配置
| 配置项 | 必填 | 默认值 | 说明 |
| :-------------------------: | :--: | :-----------------------------------------------: | :---------------------------------------------------------------------------------------------------------------------------------------------------------------: |
| :---------------------------: | :--: | :-----------------------------------------------: | :---------------------------------------------------------------------------------------------------------------------------------------------------------------: |
| **响应配置** | | | |
| `NAILONG_BYPASS_SUPERUSER` | 否 | `True` | 是否不检查超级用户发送的图片 |
| `NAILONG_BYPASS_ADMIN` | 否 | `True` | 是否不检查群组管理员发送的图片 |
@ -148,9 +148,11 @@ plugins = [
| `NAILONG_MODEL` | 否 | `0` | 选择需要加载的模型,可用模型见下 |
| `NAILONG_AUTO_UPDATE_MODEL` | 否 | `True` | 是否自动更新模型 |
| `NAILONG_CONCURRENCY` | 否 | `1` | 当图片为动图时,针对该图片并发识别图片帧的最大并发数 |
| `NAILONG_ONNX_TRY_TO_USE_GPU` | 否 | `True` | 加载 onnx 模型时是否尝试使用 GPU如果失败则会显示一串警告但是对插件并无影响如果不想看见警告关闭此配置项即可 |
| **模型 1 特定配置** | | | |
| `NAILONG_MODEL1_YOLOX_SIZE` | 否 | `[416,416]` | 针对模型 1自定义模型输入可能会有尺寸更改 |
| `NAILONG_MODEL1_TYPE` | 否 | `tiny` | 模型 1 使用的模型类型,可用 `tiny` / `m` |
| `NAILONG_MODEL1_SCORE` | 否 | `0.5` | 模型 1 置信度阈值,范围 `0` ~ `1` |
| **杂项配置** | | | |
| `NAILONG_GITHUB_TOKEN` | 否 | `None` | GitHub Access Token遇到模型下载或更新问题时可尝试填写 |
@ -184,10 +186,12 @@ plugins = [
## 📝 更新日志
### 2.1.5
### 2.2.0
- 重命名配置项 `NAILONG_YOLOX_SIZE` -> `NAILONG_MODEL1_YOLOX_SIZE`
- 模型 1 现可以自动获取最新版本,也可以通过配置 `NAILONG_MODEL1_TYPE` 选择要使用的类型
- 模型 1 现可以自动获取最新版本,也可以通过配置选择要使用的模型类型
- 模型 1 现可通过配置项控制识别置信度阈值
- 加载 onnx 模型时会默认尝试使用 GPU如果失败则会显示一串警告如果不想看见警告参考上面关闭对应配置
### 2.1.4

View File

@ -32,11 +32,11 @@ class Config(BaseModel):
nailong_model: ModelType = ModelType.CLASSIFICATION
nailong_auto_update_model: bool = True
nailong_concurrency: int = 1
nailong_onnx_try_to_use_gpu: bool = True
nailong_model1_yolox_size: Tuple[int, int] = (416, 416)
nailong_model1_type: str = "tiny"
nailong_model1_score: float = 0.5
nailong_model1_try_to_use_gpu: bool = True
nailong_github_token: Optional[str] = None

View File

@ -43,7 +43,7 @@ session = onnxruntime.InferenceSession(
"CUDAExecutionProvider",
"CPUExecutionProvider",
]
if config.nailong_model1_try_to_use_gpu
if config.nailong_onnx_try_to_use_gpu
else ["CPUExecutionProvider"]
),
)