a C g@sdddlZddlmZmZddZddZGdddeZGd d d eZGd d d eZeegZ dS) N)ImageRGBImageBuffercCs0z ddl}Wnty*ddlm}Yn0|S)Nrcv2)r ImportErrorZcvrrT/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/willow/plugins/opencv.py_cv2s   r cCs ddl}|S)Nrnumpyr rrr_numpysr c@sXeZdZddZeddZejddZejddZ ejd d Z ejd d Z d S)BaseOpenCVImagecCs||_||_dSN)imagesize)selfrrrrr__init__szBaseOpenCVImage.__init__cCs tdSr)r clsrrrcheckszBaseOpenCVImage.checkcCs|jSr)rrrrrget_sizeszBaseOpenCVImage.get_sizecCsdS)Nrrrrrget_frame_count!szBaseOpenCVImage.get_frame_countcCsdSNFrrrrr has_alpha&szBaseOpenCVImage.has_alphacCsdSrrrrrr has_animation+szBaseOpenCVImage.has_animationN) __name__ __module__ __qualname__r classmethodrr operationrrrrrrrrr s    r cs6eZdZefddZeeeddZZ S)OpenCVColorImagecsttdSr)superrr r __class__rrr2s zOpenCVColorImage.checkcCsRt}t}|j|j|jd}||jd|jdd}|||j}|||jS)z_ Converts a Color Image buffer into a numpy array suitable for use with OpenCV )Zdtyperr) r r Z frombufferdataZuint8reshapercvtColorZ COLOR_RGB2BGR)rZ image_bufferr rrrrrfrom_buffer_rgb7s z OpenCVColorImage.from_buffer_rgb) rrrr rrconverter_fromrr* __classcell__rrr$rr"1s r"c@sXeZdZdZdZdZdZejddZ ejddd Z d d Z e e ed d ZdS)OpenCVGrayscaleImagerr&g?)(r.cCsBt}t}||jddd}|dur*gS||d}|SdS)zQ Find interesting features of an image suitable for cropping to. g{Gz?g?N))r r ZgoodFeaturesToTrackrr(tolist)rr rZpointsrrrdetect_featuresLsz$OpenCVGrayscaleImage.detect_features haarcascade_frontalface_alt2.xmlcCsNt}||}||}||j}|||j|j|j|j }dd|DS)z| Run OpenCV face detection on the image. Returns a list of coordinates representing a box around each face. cSs8g|]0}|d|d|d|d|d|dfqS)rrr1r&r).0facerrr ls z5OpenCVGrayscaleImage.detect_faces..) r _find_cascadeZCascadeClassifierZ equalizeHistrZdetectMultiScaleface_haar_scaleface_min_neighborsface_haar_flags face_min_size)rcascade_filenamerZcascadeZequalised_imageZfacesrrr detect_faces\s   z!OpenCVGrayscaleImage.detect_facescCs0tj|s,tjtjtjtd|}|S)z{ Find the requested OpenCV cascade file. If a relative path was provided, check local cascades directory. z data/cascades)ospathisabsjoindirname__file__)rr=rrrr8vs z"OpenCVGrayscaleImage._find_cascadecCs"t}||j|j}|||jS)zF Convert OpenCVColorImage to an OpenCVGrayscaleImage. )r r)rZCOLOR_BGR2GRAYr)rZ colour_imagerrrrr from_colorszOpenCVGrayscaleImage.from_colorN)r4)rrrr;r:r9r<rr!r3r>r8r r+r"rErrrrr-Fs   r-) r?Z willow.imagerrr r r r"r-Zwillow_image_classesrrrrsG