TexTeller/texteller/models/det_model/__pycache__/inference.cpython-310.pyc

o
{<7B>g<EFBFBD><00>@s<>dd
lZdd
l
Z
dd
lZdd
lZdd
lZddlmZ
ddlmZ
ddl	m
Z

ddlmZ
hd<07>
ZGdd	<09>d	e
<0A>Zd
d<0B>Zdd
<0A>Zdedeefdd<11>Zd
S)<12>N)
<01>tqdm)
<01>List<73>
)
<01>Compose)
<01>Bbox>Z	CenterNet<65>RCNNZSSDZYOLOZFairMOTZGFLZ	RetinaNetZFaceZDeepSORTZS2ANetZTOODZPicoDetZStrongBaselineZTTFNet<65>HRNetZJDEZPPYOLOEZSOLOv2ZFCOSZDETRZYOLOXZSTGCNc@s(eZ
dZd
Zdd<03>Zdd<05>Zdd<07>ZdS)	<09>
PredictConfigztset config of preprocess, postprocess and visualize
    Args:
        infer_config (str): path of infer_cfg.yml
    cs
t|
<01>
<01>
}t
<01>|<02>
}Wd<00>
n1sw



Y
|<00>|<03>

|d
|_|d|_|d|_|d|_|d|_|<03>	dd<07>|_
|<03>	dd	<09>|_|<03>	d
d<00>|_|<03>	dd<00>|_
|<03>	dd<00>|_gd
<0A>
<01><00>f
dd<0F>t|j<07>
D<00>
|_|jdkrz|<03>	dd	<09>rztd<12>

|<00><12>
dS)N<>archZ
Preprocess<EFBFBD>min_subgraph_size<7A>
label_list<EFBFBD>use_dynamic_shape<70>draw_threshold<6C><00>?<3F>maskF<6B>trackerZNMS<4D>
fpn_stride))r
<00><>r
)rr
r
)r
r
r)rrr
)r
rrc
s"i|]
\}
}|<02>|
t<00><00>
<00>qS<00>)
<01>len)<03>.0<EFBFBD>
i<>label<65>
Z
color_poolr<00>C/Users/Leehy/Code/TexTeller/texteller/models/det_model/inference.py<70>
<dictcomp>-s"z*PredictConfig.__init__.<locals>.<dictcomp>rZexport_onnxzJThe RCNN export model is used for ONNX and it only supports batch_size = 1)<13>open<65>yamlZ	safe_load<61>check_modelr
<00>preprocess_infosrrr
<00>getrrr<00>nmsr<00>	enumerate<74>colors<72>print<6E>print_config)<04>self<6C>infer_config<69>
f<>yml_confrrr<00>__init__s*

<1C>


















<04>zPredictConfig.__init__cCs0tD]}||
d
vr

dSqt
d<03>|
d
t<00><02>
<01>
)zZ
        Raises:
            ValueError: loaded model not in supported model type
        r
TzUnsupported arch: {}, expect {})<03>SUPPORT_MODELS<4C>
ValueError<EFBFBD>format)r&r)Z
support_modelrrrr5s

<02>
<02>
<06>zPredictConfig.check_modelc
CsRtd
<01>

tdd|j
f<00>

tdd<00>

|jD]}
tdd|
df<00>

qtd	<09>

dS)
Nz,-----------  Model Configuration -----------z%s: %sz
Model Archz%s: zTransform Orderz--%s: %sztransform op<6F>typez,--------------------------------------------)r$r
r)r&Zop_inforrrr%@s






zPredictConfig.print_configN)<07>__name__<5F>
__module__<EFBFBD>__qualname__<5F>__doc__r*rr%rrrrr	s

r	c
Cs<>|
D]H}|\}}}}}}	||jkrJ|j
t|<04>
}
|j|
}t<04>|t|<06>
t|<07>
ft|<08>
t|	<09>
f|d
<01>
t<04>|d<02>|
|<05>t|<06>
t|d<00>
ftjd|d
<01>
q|S)N<>z
{}: {:.2f}<7D>r)	rr<00>intr#<00>cv2<76>	rectangle<6C>putTextr-<00>FONT_HERSHEY_SIMPLEX)<0C>image<67>outputsr'<00>output<75>cls_id<69>score<72>xmin<69>ymin<69>xmax<61>ymaxr<00>colorrrr<00>	draw_bboxIs






*


<04><02>rDc's<>t|
j
<01>
}g}tj<03>|d
<01>}tj|dd<03>
d}d}d}	t|<03>
D<00>
]<5D>}
t<07>|
<EFBFBD>
}|dur9t	d|
<EFBFBD>d<07><03>

|<05>
|
<EFBFBD>

q||
<EFBFBD>
<01>dd	<09>|<02><0B>D<00>
}<0C>f
d
d<0B>|D<00>
<01>t<0C><0C>}
|j
d<00>d<0C>}t<0C><0C>}||
}|sgd}n||7}|	d
7}	t	dtj<03>|
<EFBFBD>
<01>d|d<10>d<11><05>

t	d<12>

|
jdvr<>t	t<10>|d<00>
<01>

n<t<10>|d<00>
}|D]2}|ddkr<>|d
|
jkr<>t	t|d<00>
<01>d|d
<00>d|d<00>d|d<00>d|d<00>d|d<00><00><0B>

q<>d
}t<10>|d<00>
D]j}|\}}}}}}||
jk<04>
rA|
jt|<15>
}|tt|d<05><02>
t|<1A>
<01>tt|d<05><02>
t|<19>
<01>f}t|<1C>
dk<02>
rq<>tj<03>tj<03>|
<EFBFBD>
<01>
d<00>d|<1B>d|d<1B>d|d<1B>d|d<1B>d|d<1B>d<1C>}tj<03>||<1D>}t<07>||<1C>
|d
7}q<>|<0B><19>}t<10>|d<00>
D]&}|\}}}}}}||
jk<04>
rrt<07>|t|<17>
t|<18>
ft|<19>
t|<1A>
fdd<14>
<00>
qMt|t<10>|d<00>
|
<01>} |}!tj|!dd<03>
tj<03>|!d<1E>}"tj<03>|!d<1F>}#tj|"dd<03>
tj|#dd<03>
tj<03>|#tj<03>|
<EFBFBD>
<01>}$tj<03>|"tj<03>|
<EFBFBD>
<01>}%t<07>|$|<1F>
t<07>|%| <20>
q|	dk<04>
r<>||	n
d}&t	d |	<09>d!|d<10>d<11><05>

t	d"|&d<10>d<11><03>

t	d#<23>

t	|<05>

dS)$NZ	subimagesT)
<01>exist_okFr
zWarning: Could not read image z
. Skipping...c
S<00>g|]}
|
j<00>qSr<00>
<01>name<6D>r<00>varrrr<00>
<listcomp>j<00>z!predict_image.<locals>.<listcomp>c
<00>i|]	}
|
<01>|
d<00>qS<00>
)
Nr<00>r<00>
k<>
<01>inputsrrrk<00>z!predict_image.<locals>.<dictcomp><3E><02>output_names<65>
input_feedrzONNXRuntime predict time for z: z.4fz secondszONNXRuntime predict: )
r<00><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
 r3<00><00>r4<00>
_z.2fz.jpg)rrrZdraw_boxZ
mask_whitezTotal inference time for z	 images: zAverage time per image: z
ErrorImgs:)rr<00>os<6F>path<74>join<69>makedirsrr6<00>imreadr$<00>append<6E>
get_inputs<EFBFBD>time<6D>run<75>basenamer
<00>np<6E>arrayrr5r<00>maxr<00>splitext<78>imwrite<74>copyr7rD)'Zimgsave_dirr'<00>	predictor<6F>img_list<73>
transformsZ
errImgListZsubimg_save_dirZfirst_image_skippedZ
total_timeZ
num_images<EFBFBD>img_path<74>img<6D>inputs_name<6D>
start_timer;<00>end_timeZinference_time<6D>bboxes<65>bboxZsubimg_counterr<r=r>r?r@rArBr<00>subimgZsubimg_filenameZsubimg_pathZ
img_with_maskZ
img_with_bboxZ
output_dirZdraw_box_dirZmask_white_dirZoutput_file_maskZoutput_file_bboxZavg_time_per_imagerrQr<00>
predict_imageUs<>

















 







<04>
<04>
<04>
<08><02>



0


D


<02>


*
<04>











rwro<00>returnc
s<>t|j
<01>
}||<00>
<01>d
d<02>|
<01><02>D<00>
}<04>f
dd<04>|D<00>
<01>|
jd<00>d<05>d}g}|D]B}|jt|d<00>
}|d}	tt|dd<06><02>
}
tt|d	d<06><02>
}t|d
<00>
}t|d<00>
}
|	|jkrj|<06>t	|
||
|||
||	<09><06>

q(|S)Nc
SrFrrGrIrrrrK<00>rLzpredict.<locals>.<listcomp>c
rMrNrrOrQrrr<00>rSzpredict.<locals>.<dictcomp>rTr
rr3rYrZr4)
rrrbrdrr5rhrrar)rorlr'rnrqr;<00>resr<<00>cls_namer>r?r@rArBrrQr<00>predict<63>s"














 
<02>r{)r\rcr<00>numpyrfr6r<00>typingrZ
preprocessrrr+<00>objectr	rDrw<00>strr{rrrr<00><module>
s






3\