o Ďi@sXddlZddlZddlZddlZddlmZddZddZddZd d Z d d Z dS) N)ImagecCs&d}|D] }|d|vr|d7}q|S)Nrclass) predictions target_classcount predictionrr_/home/jeff/fluffinator/venv/lib/python3.10/site-packages/roboflow/util/active_learning_utils.pycount_class_occurances s  r cCs&t||ks|rt|||krdSdS)NFT)lenr )rrequired_objects_countrequired_class_countrrrr count_comparisonss rcCs,||d|dkr|krdSdSdS)NheightwidthTFr)r minimum_size_requirementmaximum_size_requirementrrr check_box_size!srcCsDt|}t}|d}|j|dddt|}| dS)NRGBZJPEG)qualityformatascii) ropenioBytesIOconvertsavebase64 b64encodegetvaluedecode) image_pathimagebuffered image_rgbimg_strrrr base64_encode*s   r)cCs\t|}t|}|dkrtd|}ddi}t||d}tj|||d}t|dS)NzCLIP FEATURIZE URL NOT IN ENVzYou need to ad CLIP_FEATURE_URL to your env vars. To learn more about this active learning feature, contact Roboflow sales https://roboflow.com/sales. You can remove the similarity keys from your conditionals to use other active learning functionality.z Content-Typez text/plain)image1image2)dataheaders similarity)r) Exceptionjsondumpsrequestspostfloat)r*r+CLIP_FEATURIZE_URLurlr-r,rrrr clip_encode3sr8) r rr0r2PILrr rrr)r8rrrr s