jUddlmZddlZddlZddlmZddlmZdgZ ddgd d d iZ e e e e effe d <Gd deZdS))AnyN)requires_package)_BasePredictorArtefactDetectoryolov8_artefact)r )bar_codeqr_codelogophotozRhttps://doctr-static.mindee.com/models?id=v0.8.1/yolo_artefact-f9d66f14.onnx&src=0) input_shapelabelsurl default_cfgsc0eZdZdZ ddedededzd eedzd eeeefdzd ed ed e ddffd Z de j de j fdZ dee j deee j deeeee ffdZd e ddfdZxZS)ra A class to detect artefacts in images >>> from doctr.io import DocumentFile >>> from doctr.contrib.artefacts import ArtefactDetector >>> doc = DocumentFile.from_images(["path/to/image.jpg"]) >>> detector = ArtefactDetector() >>> results = detector(doc) Args: arch: the architecture to use batch_size: the batch size to use model_path: the path to the model to use labels: the labels to use input_shape: the input shape to use mask_labels: the mask labels to use conf_threshold: the confidence threshold to use iou_threshold: the intersection over union threshold to use **kwargs: additional arguments to be passed to `download_from_url` rN?arch batch_size model_pathrrconf_threshold iou_thresholdkwargsreturnc tjd|t|d|d||pt|d|_|pt|d|_||_||_dS)Nr)rrrrr)super__init__rrrrr) selfrrrrrrrr __class__s Z/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/doctr/contrib/artefacts.pyr zArtefactDetector.__init__0sz oJL4Fu4MZdoohnooo< T 28 < &K,t*<]*K,*imgctjtj||jd|jdfdtjdz S)Nrr)rrrgo@)np transposecv2resizerarray)r!r%s r# preprocesszArtefactDetector.preprocessAsI|CJsT-=a-@$BRSTBU,VWWYbccfhfnotfufuuur$output input_imagesc  g}t||D]}t|d|dD]\}}|jdd\}}||jdz ||jdz } } |D]} gtjtj| D]} | dd} tj| }||jkrtj| }| d| d| d| df\}}}}t||dz z | z}t||dz z | z}t||dz z| z}t||dz z| z} |j |t|||||gddD}dD}tj|||j|j}fd |D| Ȑ||_|S) Nrrrr )label confidenceboxcg|] }|d S)r3r.0ress r# z0ArtefactDetector.postprocess..`sBBBCSZBBBr$cg|] }|d S)r2rr5s r#r8z0ArtefactDetector.postprocess..asJJJCc,/JJJr$c g|] }| Srr)r6isample_resultss r#r8z0ArtefactDetector.postprocess..cs%N%N%NAnQ&7%N%N%Nr$)zipshaperr'r(squeezeamaxrargmaxintappendrfloatr)dnnNMSBoxesr_results)r!r-r.resultsbatchoutr% org_height org_width width_scale height_scaler7rowclasses_scores max_scoreclass_idxywhxminyminxmaxymaxboxesscores keep_indicesr<s @r# postprocesszArtefactDetector.postprocessDsP.. 3 3Ea%(33 3 3S(+ "1" % I,58H8K,KZZ^ZjklZmMm\ 33C%'N!|BJsOO<<),QRR$&GN$;$; $(;;;')y'@'@H),QQQQ)GJAq!Q#&AE ['@#A#AD#&AE \'A#B#BD#&AE ['@#A#AD#&AE \'A#B#BD*11)-X)>.3I.>.>(,dD$'?33CB>BBBEJJ>JJJF#&7#3#3E64CVX\Xj#k#kL%N%N%N%N%N%N%NNNN>222253 3>  r$c tddddlm}ddlm}|jr|jrt|j|jD]\}}|d| ||D]z}|d\}}} } |d } | ||| d |d d d | |||f| |z | |z dd d{|j di|dSdSdS)z Display the results Args: **kwargs: additional keyword arguments to be passed to `plt.show` matplotlibz'`.show()` requires matplotlib installedrN) Rectangle) rb)figsizer3r1 r2z.2fred)colorFr)fill edgecolor linewidthr)rmatplotlib.pyplotpyplotmatplotlib.patchesrarG_inputsr=figureimshowtextgca add_patchshow) r!rpltrar%r7objrWrXrYrZr1s r#rszArtefactDetector.showjsu 'PQQQ''''''000000 = #T\ # dm<< # #S 8 ,,, 3C-0Z*D$dLEHHT4E)K)KC 4E)K)K)KSXHYYYGGII''! 4,t TD[u`eqrsss""6"""" # # # # # #r$)rrNNNrr)__name__ __module__ __qualname____doc__strrBlisttuplerDrr r'ndarrayr,dictr^rs __classcell__)r"s@r#rrs|.&!%#'37 #"++++$J + S D + 3S=)D0 ++++ ++++++"vbjvRZvvvv$$rz"2$$tBJGWBX$]abfgkloqtltgubv]w$$$$L#S#T########r$)typingrr)numpyr'doctr.file_utilsrbaser__all__rr~rz__annotations__rrr$r#rs  ------  &:::c+ d3S#X&'g#g#g#g#g#~g#g#g#g#g#r$