Implement multi-model detection in HVideoTool: updated the detection system to support multiple YOLO models simultaneously, enhancing detection capabilities. Reflected changes in the UI with a new model selection menu and updated documentation in README and CLAUDE.md to guide users on model management and configuration.
This commit is contained in:
@@ -12,7 +12,7 @@ import sys
|
||||
|
||||
from . import settings_store
|
||||
from .app import run
|
||||
from .config import AppConfig, normalize_config
|
||||
from .config import AppConfig
|
||||
|
||||
|
||||
def main() -> int:
|
||||
@@ -21,15 +21,14 @@ def main() -> int:
|
||||
description="Инспектор детекции уже наложенной цензуры на картинках.",
|
||||
)
|
||||
parser.add_argument("target", nargs="?", help="путь к проекту для открытия (папка или project.json)")
|
||||
parser.add_argument("--model", dest="model_path", default=None, help="путь к весам (YOLO)")
|
||||
parser.add_argument("--model", dest="model", default=None, help="путь к весам YOLO (.pt) — использовать только эту модель")
|
||||
args = parser.parse_args()
|
||||
|
||||
config = AppConfig()
|
||||
settings_store.apply(config) # persisted defaults first
|
||||
normalize_config(config) # drop any legacy classic/inpaint values
|
||||
|
||||
if args.model_path:
|
||||
config.model_path = args.model_path
|
||||
if args.model:
|
||||
config.detector_models = [args.model]
|
||||
|
||||
return run(config, target=args.target)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user