a yf@sxddlmZddlmZddlmZddlmZmZm Z m Z m Z m Z ddl mZmZGdddeZGdd d eZd S) )Path)Model)yolo)ClassificationModelDetectionModelOBBModel PoseModelSegmentationModel WorldModel)ROOT yaml_loadcs.eZdZdZd fdd ZeddZZS) YOLOz1YOLO (You Only Look Once) object detection model. yolov8n.ptNFcsRt|}d|jvr<|jdvrz.ptz.ymlz.yaml)verbosemodeltaskrN) rstemsuffix YOLOWorldtype __class____dict__super__init__)selfrrrpathZ new_instancerY/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/models/yolo/model.pyrs    z YOLO.__init__c Cs~ttjjtjjtjjdttjjtjj tjj dt tj j tj jtj jdttjjtjjtjjdttjjtjjtjjddS)z=Map head to model, trainer, validator, and predictor classes.)rtrainer validator predictor)classifydetectsegmentposeobb)rrr#ZClassificationTrainerZClassificationValidatorZClassificationPredictorrr$ZDetectionTrainerDetectionValidatorDetectionPredictorr r%ZSegmentationTrainerZSegmentationValidatorZSegmentationPredictorrr&Z PoseTrainerZ PoseValidatorZ PosePredictorrr'Z OBBTrainerZ OBBValidatorZ OBBPredictorrrrrtask_maps4z YOLO.task_map)rNF)__name__ __module__ __qualname____doc__rpropertyr+ __classcell__rrrrr s r cs<eZdZdZd ddfdd Zedd Zd d ZZS) rz"YOLO-World object detection model.yolov8s-world.ptFN)returncs8tj|d|dt|jds4ttdd|j_dS)a Initialize YOLOv8-World model with a pre-trained model file. Loads a YOLOv8-World model for object detection. If no custom class names are provided, it assigns default COCO class names. Args: model (str | Path): Path to the pre-trained model file. Supports *.pt and *.yaml formats. verbose (bool): If True, prints additional information during initialization. r$rnameszcfg/datasets/coco8.yamlN)rrhasattrrr r getr4)rrrrrrrAs  zYOLOWorld.__init__cCsdttjjtjjtjjdiS)z4Map head to model, validator, and predictor classes.r$)rr!r"r )r rr$r(r)ZworldZ WorldTrainerr*rrrr+RszYOLOWorld.task_mapcCs>|j|d}||vr"||||j_|jr:||jj_dS)zt Set classes. Args: classes (List(str)): A list of categories i.e. ["person"].  N)r set_classesremover4r")rclasses backgroundrrrr8^s  zYOLOWorld.set_classes)r2F) r,r-r.r/rr0r+r8r1rrrrr>s  rN)pathlibrZultralytics.engine.modelrZultralytics.modelsrZultralytics.nn.tasksrrrrr r Zultralytics.utilsr r r rrrrrs    3