a yf @sddlZddlZddlmZddlmZddlmZejj ejj e ddddZ ejj ejj e ddd d Z ejj ejj e ddd d Z ejj ejj e ddd dZdS)N)Explorer)ASSETS) TORCH_1_13zExplorer requires torch>=1.13)reasoncCstdd}||jdd}t|dks.J|jtdd}t|dksNJ|jddgdd }t|dkspJ|}t|dksJ|d }t|dksJd S) zdTest the correctness and response length of similarity calculations and SQL queries in the Explorer. coco8.yaml)data)idxzbus.jpg)Zimgr limitzWHERE labels LIKE '%zebra%'N)rcreate_embeddings_table get_similarlenrZsimilarity_indexZ sql_query)expsimilarZsim_idxsqlrO/var/www/html/django/DPS/env/lib/python3.9/site-packages/tests/test_explorer.pytest_similarity s   rcCs|tddd}|jddt|jddks2J|jdd gd d }t|dksTJ|jdd gd d }t|tj j sxJd S) zRTest detection functionalities and verify embedding table includes bounding boxes.rz yolo11n.ptrmodelTforceZbboxesrrr r N rrrtableheadrZ plot_similar isinstancePILZImagerrrrrtest_dets  r"cCs|tddd}|jddt|jddks2J|jdd gd d }t|dksTJ|jdd gd d }t|tj j sxJd S) z]Test segmentation functionalities and ensure the embedding table includes segmentation masks.zcoco8-seg.yamlzyolo11n-seg.ptrTrmasksrrr rr Nrr!rrrtest_seg+s  r$cCs|tddd}|jddt|jddks2J|jdd gd d }t|dksTJ|jdd gd d }t|tj j sxJd S) zUTest pose estimation functionality and verify the embedding table includes keypoints.zcoco8-pose.yamlzyolo11n-pose.ptrTrZ keypointsrrr rr Nrr!rrr test_pose8s  r%)r ZpytestZ ultralyticsrZultralytics.utilsrZultralytics.utils.torch_utilsrmarkZslowZskipifrr"r$r%rrrrs