a yf@sddlZddlmZddlZddlZddlmZddlm Z ddlm Z ddl m Z ddlmZddlmZd d Zd d Zd dZdddZddZdS)N)List) LetterBox)LOGGER)SETTINGS)check_requirements) xyxy2xywh) plot_imagescs*ddlm}mGfddd|}|S)z4Extracts and returns the schema of a database table.r) LanceModelVectorcsreZdZUeed<eeed<eeed<eeeed<eeeeed<eeeeed<ed<dS) z get_table_schema..Schemaim_filelabelsclsbboxesmasks keypointsZvectorN)__name__ __module__ __qualname__str__annotations__rintfloatr vector_sizer[/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/data/explorer/utils.pySchemas   r)lancedb.pydanticr r )rr rrrrget_table_schemas rcCs ddlm}Gddd|}|S)zLReturns a LanceModel schema for a database table with specified vector size.r)r c@s2eZdZUeed<eed<eed<eeed<dS)z$get_sim_index_schema..Schemaidxr countZ sim_im_filesN)rrrrrrrrrrrr%s r)rr )r rrrrget_sim_index_schema!s r!cs|d|d<tt|d|dddd}dd|D|d<dd|D|d<fd d|dD|d <d |vr|d nggg|d <d |vr|d nggg|d <|S) zLSanitizes input batch for inference, ensuring correct format and dimensions.r rcSs|dS)Nr)xrrr1z sanitize_batch..)keycSsg|] \}}|qSrr).0box_rrr 2r%z"sanitize_batch..cSsg|] \}}|qSrr)r'r)r rrrr*3r%csg|]}d|qS)namesr)r'i dataset_inforrr*4r%r rr)flattenrtolistsortedzip)batchr.Z box_cls_pairrr-rsanitize_batch.s"  r4Tc CsLddl}t||jr |jddn|}ggg}gg}|dg}|d|ur\|dgng}|dd|kr||dng}|dd|kr|dng}|d g} d } gggggf\} } } }}t|D]\}}t|}t |tj }|j dd \}}t | || |}| t| d d |dd dd|r`t||krt||dkrtj||tjd}|dddd gf|9<|ddddgf|9<| |t||krt||dkrtj||tjdd}| t| d d |dt||kr`||dur`tj||tjd}|dddddd f|9<| || tttj||tjd|qtj| dd} |rtj|ddntjdtjd}|rtj|ddntjdtjd}| rttj| ddntjdtjd}tj| dd} tjdd| Ddd} t| | | |||t|d d d S)z Plot images from the similar set. Args: similar_set (list): Pyarrow or pandas object containing the similar data points plot_labels (bool): Whether to plot labels or not rNlist)Zorientr rrrr iF)center)imager"Zdtype)Zaxis)r3cSsg|]}tj|tjdqS)r9)nparrayZint32)r'crrrr*jr%z%plot_query_result..)rrkptsZ max_subplotssaveZthreaded)pandas isinstanceZ DataFrameto_dictZ to_pydictget enumeratecv2ZimreadZcvtColorZ COLOR_BGR2RGBshapeminappendrZ transposelenr<r=Zfloat32Zuint8ZonesstackZzerosZ concatenaterr)Z similar_setZ plot_labelsrAZ empty_masksZ empty_boxesZimagesrrr?r Z plot_sizeZimgsZ batch_idxZ plot_boxesZ plot_masksZ plot_kptsr,ZimfZimhwrr(maskZkptZboxesrrrplot_query_result:sR     "   " *$$(rPcCstdddlm}tds>tdtd}td|i|tdd}dd d d |d g}|jj j d |d }|j dj j S)z:Plots images with optional labels from a similar data set.z openai>=1.6.1r)OpenAIopenai_api_keyzFOpenAI API key not found in settings. Please enter your API key below.zOpenAI API key: )Zapi_keysystema You are a helpful data scientist proficient in SQL. You need to output exactly one SQL query based on the following schema and a user request. You only need to output the format with fixed selection statement that selects everything from "'table'", like `SELECT * from 'table'` Schema: im_file: string not null labels: list not null child 0, item: string cls: list not null child 0, item: int64 bboxes: list> not null child 0, item: list child 0, item: double masks: list>> not null child 0, item: list> child 0, item: list child 0, item: int64 keypoints: list>> not null child 0, item: list> child 0, item: list child 0, item: double vector: fixed_size_list[256] not null child 0, item: float Some details about the schema: - the "labels" column contains the string values like 'person' and 'dog' for the respective objects in each image - the "cls" column contains the integer values on these classes that map them the labels Example of a correct query: request - Get all data points that contain 2 or more people and at least one dog correct query- SELECT * FROM 'table' WHERE ARRAY_LENGTH(cls) >= 2 AND ARRAY_LENGTH(FILTER(labels, x -> x = 'person')) >= 2 AND ARRAY_LENGTH(FILTER(labels, x -> x = 'dog')) >= 1; )Zrolecontentuserz gpt-3.5-turbo)modelmessages)ropenairQrloggerwarninggetpassupdateZchatZ completionscreatechoicesmessagerT)queryrQrRrXrWresponserrrprompt_sql_queryqs   & *rb)T)r[typingrrFnumpyr<Zultralytics.data.augmentrZultralytics.utilsrrYrZultralytics.utils.checksrZultralytics.utils.opsrZultralytics.utils.plottingrrr!r4rPrbrrrrs         7