a yf]@sddlZddlZddlmZddlZddlZddlmZddl m Z m Z ddl m Z mZmZddlmZmZmZmZmZmZmZmZddlmZmZmZddlmZmZdd l m!Z!Gd d d e j"Z#Gd d d Z$ddZ%d ddZ&d!ddZ'd"ddZ(ddZ)d#ddZ*dS)$N)Path)Image) dataloader distributed)GroundingDataset YOLODatasetYOLOMultiModalDataset)LOADERSLoadImagesAndVideosLoadPilAndNumpyLoadScreenshots LoadStreams LoadTensor SourceTypes autocast_list) IMG_FORMATS PIN_MEMORY VID_FORMATS)RANKcolorstr) check_filecs8eZdZdZfddZddZddZdd ZZS) InfiniteDataLoaderzV Dataloader that reuses workers. Uses same syntax as vanilla DataLoader. cs6tj|i|t|dt|jt|_dS)zFDataloader that infinitely recycles workers, inherits from DataLoader. batch_samplerN)super__init__object __setattr___RepeatSamplerr__iter__iterator)selfargskwargs __class__R/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/data/build.pyr#szInfiniteDataLoader.__init__cCs t|jjS)z2Returns the length of the batch sampler's sampler.)lenrsamplerr r%r%r&__len__)szInfiniteDataLoader.__len__ccs"tt|D]}t|jVq dS)z,Creates a sampler that repeats indefinitely.N)ranger'nextr)r _r%r%r&r-szInfiniteDataLoader.__iter__cCs||_dS)zt Reset iterator. This is useful when we want to modify settings of dataset while training. N)Z _get_iteratorrr)r%r%r&reset2szInfiniteDataLoader.reset) __name__ __module__ __qualname____doc__rr*rr. __classcell__r%r%r#r&rs  rc@s eZdZdZddZddZdS)rzl Sampler that repeats forever. Args: sampler (Dataset.sampler): The sampler to repeat. cCs ||_dS)z@Initializes an object that repeats a given sampler indefinitely.N)r()r r(r%r%r&rCsz_RepeatSampler.__init__ccst|jEdHqdS)z4Iterates over the 'sampler' and yields its contents.N)iterr(r)r%r%r&rGsz_RepeatSampler.__iter__N)r/r0r1r2rrr%r%r%r&r;srcCs&td}tj|t|dS)z\Set dataloader worker seed https://pytorch.org/docs/stable/notes/randomness.html#dataloader.lN)torchZ initial_seednprandomseed)Z worker_idZ worker_seedr%r%r& seed_workerMs  r9trainF c Csv|rtnt}|||j||dk||jp&||jp.d|jp6dt||dkrHdndt|d|j|j ||dkrn|j nddS) Build YOLO Dataset.r:NF?: ?)img_pathimgsz batch_sizeaugmenthyprectcache single_clsstridepadprefixtaskclassesdatafraction) rrrBrFrGrHintrrLrMrO) cfgrAbatchrNmoderFrIZ multi_modaldatasetr%r%r&build_yolo_datasetTs$  rUcCsjt|||j||dk||jp||jp$d|jp,dt||dkr>dndt|d|j|j|dkrb|j nddS) r<r:NFr=r>r?r@)rA json_filerBrCrDrErFrGrHrIrJrKrLrMrO) rrBrFrGrHrPrrLrMrO)rQrArVrRrSrFrIr%r%r&build_groundingjs" rWTc Cst|t|}tj}ttt|d|}|dkr|rXt|||d} n*|rft|} n|rtt|} nt|||d} t| d| | S)a& Loads an inference source for object detection and applies necessary transformations. Args: source (str, Path, Tensor, PIL.Image, np.ndarray): The input source for inference. batch (int, optional): Batch size for dataloaders. Default is 1. vid_stride (int, optional): The frame interval for video sources. Default is 1. buffer (bool, optional): Determined whether stream frames will be buffered. Default is False. Returns: dataset (Dataset): A dataset object for the specified input source. ) vid_stridebuffer)rRr{ source_type) rzr}rrr r r r setattr) rsrRr{r|streamrtrurvrwr}rTr%r%r&load_inference_sources     r)r:Fr;F)r:Fr;)TrX)NrYrYF)+r_r7pathlibrnumpyr6r5ZPILrZtorch.utils.datarrZultralytics.data.datasetrrrZultralytics.data.loadersr r r r r rrrZultralytics.data.utilsrrrZultralytics.utilsrrZultralytics.utils.checksrZ DataLoaderrrr9rUrWrgrzrr%r%r%r&s&  (