o Ďi# @sddlmZddlZddlZddlZddlZddlZddlZddlZ ddl Z ddl m Z ddl Zd;ddZdd?ddZdgdfd@ddZdgdfdAd"d#ZdBd%d&ZdBd'd(Zd)d*Zd+d,Zd-d.ZdCdDd3d4ZdEd7d8Zd9d:ZdS)F) annotationsN)Path file_pathstrdest_dircCs:t|}||WddS1swYdS)zxUnzips compressed .zip file. Example inputs: file_path: 'data/01_alb_id.zip' dest_dir: 'data/' N)zipfileZipFile extractall)rrzfr K/home/jeff/fluffinator/venv/lib/python3.10/site-packages/sahi/utils/file.pyunzips "r indent int | NonecCs\t|jjdddt|ddd}tj||dt|dWddS1s'wYdS) z Saves json formatted data (given as "data") as save_path Example inputs: data: {"image_id": 5} save_path: "dirname/coco.json" indent: Train json files with indent=None, val json files with indent=4 Tparentsexist_okwutf-8encoding),:) separatorsclsrN)rparentmkdiropenjsondump NumpyEncoder)data save_pathroutfiler r r save_jsons "r$cseZdZfddZZS)r csHt|tjr t|St|tjrt|St|tjr|St |SN) isinstancenpintegerintfloatingfloatndarraytolistsuperdefault)selfobj __class__r r r//s    zNumpyEncoder.default)__name__ __module__ __qualname__r/ __classcell__r r r2r r .sr r load_pathrcCs<t||d}t|}Wd|S1swY|S)zLoads json formatted data (given as "data") from load_path Encoding type can be specified with 'encoding' argument. Example inputs: load_path: "dirname/coco.json" rN)rrload)r8r json_filer!r r r load_json:s   r;z.json directorycontainslistverboser)return list[str]cs|rtndd}g}t|D]tfdd|Dr*ttj|}||qt|}t |j }|d|d|d|S)aWalk given directory and return a list of file path with desired extension. Args: directory: str "data/coco/" contains: list A list of strings to check if the target file contains them, example: ["coco.png", ".jpg", "jpeg"] verbose: int 0: no print 1: print number of files Returns: filepath_list : list List of file paths c_dSr%r akr r r \zlist_files..c3|] }|vVqdSr%lower.0 strtocheckfiler r bzlist_files.. There are z listed files in folder: /) printoslistdiranyrpathjoinappendlenrname)r=r>r@ verboseprint filepath_listfilepathnumber_of_files folder_namer rOr list_filesGs  rcTbooltuple[list, list]c s|rtndd}g}g}t|D]-\}}}|D]%tfdd|Dr=tj|} || | |d} || qqt|} |tj d} |d| d| d||fS) a1Walk given directory recursively and return a list of file path with desired extension. Arguments ------- directory : str "data/coco/" contains : list A list of strings to check if the target file contains them, example: ["coco.png", ".jpg", "jpeg"] verbose : bool If true, prints some results Returns ------- relative_filepath_list : list List of file paths relative to given directory abs_filepath_list : list List of absolute file paths c_rCr%r rDr r r rGrHz(list_files_recursively..c3rIr%rJrLrOr r rQrRz)list_files_recursively..rSz listed files in folder .) rUrVwalkrXrYrZr[splitr\sep) r=r>r@r^abs_filepath_listrelative_filepath_listr_f abs_filepathrelative_filepathrarbr rOr list_files_recursivelyns   rrrYcCs"t|}tj|\}}||fS)zZTakes a file path, returns (base_filename_with_extension, base_filename_without_extension))ntpathbasenamerVrYsplitext)rYbase_filename_with_extensionbase_filename_without_extensionrnr r r get_base_filenames rxcCstj|\}}|S)zGet the file extension from a given file path. Args: path (str): The file path. Returns: str: The file extension. )rVrYru)rYrnfile_extensionr r r get_file_extensions rzcCs:t|d}t|}Wd|S1swY|S)z Loads pickle formatted data (given as "data") from load_path Example inputs: load_path: "dirname/coco.pickle" rbN)rpickler9)r8r:r!r r r load_pickles   r}cCsPt|jjdddt|d}t||WddS1s!wYdS)z Saves pickle formatted data (given as "data") as save_path Example inputs: data: {"image_id": 5} save_path: "dirname/coco.pickle" TrwbN)rrrrr|r)r!r"r#r r r save_pickles "rcCs"td||gd}t||}|S)a4Imports a predefined detection class by class name. Args: model_type: str "yolov5", "detectron2", "mmdet", "huggingface" etc model_name: str Name of the detection model class (example: "MmdetDetectionModel") Returns: class_: class with given path z sahi.models.)fromlist) __import__getattr) model_type class_namemoduleclass_r r r import_model_classs r str | Pathrrjcs~tr |sstStd}fdd|D}dd|D}|r5t|dnd}|S)a2Increment path, i.e. runs/exp --> runs/exp{sep}0, runs/exp{sep}1 etc. Args: path: str The base path to increment. exist_ok: bool If True, return the path as is if it already exists. If False, increment the path. sep: str The separator to use between the base path and the increment number. Returns: str: The incremented path. Example: >>> increment_path("runs/exp", sep="_") 'runs/exp_0' >>> increment_path("runs/exp_0", sep="_") 'runs/exp_1' *cs&g|]}tddj|qS)z%sz(\d+))researchstem)rMdrYrjr r s&z"increment_path..cSs g|] }|rt|dqS)r)r)groups)rMmr r r rs r<)rexistsrglobmax)rYrrjdirsmatchesindicesnr rr increment_pathsrfrom_urlto_pathcCs6t|jjdddtj|stj||dSdS)zDownloads a file from the given URL and saves it to the specified path. Args: from_url (str): The URL of the file to download. to_path (str): The path where the downloaded file should be saved. Returns: None TrN) rrrrVrYrurllibrequest urlretrieve)rrr r r download_from_urls rcCsddl}d|jvS)zCheck if the current environment is a Google Colab instance. Returns: bool: True if the environment is a Google Colab instance, False otherwise. rNz google.colab)sysmodules)rr r r is_colab s r)rrrrr%)rr)r)r8rrr)r=rr>r?r@r)rArB)r=rr>r?r@rdrAre)rYr)Tr)rYrrrdrjrrAr)rrrr) __future__rrrrsrVr|rurllib.requestrrpathlibrnumpyr'r r$ JSONEncoderr r;rcrrrxrzr}rrrrrr r r r s6      ' ,