a yf@sLddlmZddlmZddlmZddlmZmZGdddejj Z dS))copy)yolo)OBBModel) DEFAULT_CFGRANKcs:eZdZdZeddffdd Zd ddZdd ZZS) OBBTrainerah A class extending the DetectionTrainer class for training based on an Oriented Bounding Box (OBB) model. Example: ```python from ultralytics.models.yolo.obb import OBBTrainer args = dict(model="yolov8n-obb.pt", data="dota8.yaml", epochs=3) trainer = OBBTrainer(overrides=args) trainer.train() ``` Ncs(|dur i}d|d<t|||dS)z4Initialize a OBBTrainer object with given arguments.Nobbtask)super__init__)selfcfgZ overrides _callbacks __class__]/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/models/yolo/obb/train.pyr szOBBTrainer.__init__TcCs0t|d|jd|otdkd}|r,|||S)z>Return OBBModel initialized with specified config and weights.nc)chrverbose)rdatarload)r r weightsrmodelrrr get_models zOBBTrainer.get_modelcCs"d|_tjj|j|jt|jdS)z@Return an instance of OBBValidator for validation of YOLO model.)Zbox_lossZcls_lossZdfl_loss)save_dirargs)Z loss_namesrrZ OBBValidatorZ test_loaderrrr)r rrr get_validator'szOBBTrainer.get_validator)NNT) __name__ __module__ __qualname____doc__rr rr __classcell__rrrrr s  rN) rZultralytics.modelsrZultralytics.nn.tasksrZultralytics.utilsrrdetectZDetectionTrainerrrrrrs