mirror of
https://github.com/Refound-445/nonebot-plugin-nailongremove.git
synced 2025-11-04 21:22:43 +08:00
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
from enum import Enum, auto
|
|
from pathlib import Path
|
|
from typing import List, Optional, Tuple
|
|
|
|
from cookit import StrEnum
|
|
from nonebot import get_plugin_config
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class ModelType(int, Enum):
|
|
CLASSIFICATION = 0
|
|
TARGET_DETECTION = 1
|
|
|
|
|
|
class Model1Type(StrEnum):
|
|
TINY = auto()
|
|
M = auto()
|
|
|
|
@property
|
|
def yolox_size(self) -> Tuple[int, int]:
|
|
return {
|
|
Model1Type.TINY: (416, 416),
|
|
Model1Type.M: (640, 640),
|
|
}[self]
|
|
|
|
|
|
class Config(BaseModel):
|
|
proxy: Optional[str] = None
|
|
|
|
nailong_bypass_superuser: bool = True
|
|
nailong_bypass_admin: bool = True
|
|
nailong_need_admin: bool = False
|
|
nailong_list_scenes: List[str] = Field(default_factory=list)
|
|
nailong_blacklist: bool = True
|
|
nailong_priority: int = 100
|
|
|
|
nailong_recall: bool = True
|
|
nailong_mute_seconds: int = 0
|
|
nailong_tip: str = "本群禁止发送奶龙!"
|
|
nailong_failed_tip: str = "{:Reply($message_id)}呜,不要发奶龙了嘛 🥺 👉👈"
|
|
nailong_check_all_frames: bool = False
|
|
|
|
nailong_model_dir: Path = Field(
|
|
default_factory=lambda: Path.cwd() / "data" / "nailongremove",
|
|
)
|
|
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_type: Model1Type = Model1Type.TINY
|
|
nailong_model1_yolox_size: Optional[Tuple[int, int]] = None
|
|
nailong_model1_score: float = 0.5
|
|
|
|
nailong_github_token: Optional[str] = None
|
|
|
|
|
|
config = get_plugin_config(Config)
|