o ,h2 @srddlZddlZddlZddlZddlZgddgddddddddf ddZdd Zd d Ze d kr7edSdS) Ni`i enTcpuc Cs|durtdtj||dkrdnd| | ||||d} | rt|} | |} t1| | \} }tjj | j| |ddddgdg|rPd d d d d d dddnddd Wdn1s_wYt |}tj |td|j jddtd|j jddddl}||}dd}|dj|| i}|d|\}}td| jd|jtd|jd|jtjj|| |dddtjj|||dddtd |d!dSdS)"NFzEWARNING: it is recommended to use -d dynamic flag when exporting onnxrT)gpudetector recognizerquantizemodel_storage_directoryuser_network_directorydownload_enabled inputoutput batch_sizeheightwidth)rrdim1dim2)rrr)rr) export_paramsdo_constant_folding opset_version input_names output_names dynamic_axesverbosezModel Inputs:  zP********************************************************************************zModel Outputs: rcSs"|jr |S|S)N) requires_graddetachrnumpy)tensorr#I/var/www/html/scripts/venv/lib/python3.10/site-packages/easyocr/export.pyto_numpyBs z!export_detector..to_numpyz!torch outputs: y_torch_out.shape=z feature_torch_out.shape=zonnx outputs: y_onnx_out.shape=z feature_onnx_out.shape=gMbP?gh㈵>)rtolatolzModel exported to z8 and tested with ONNXRuntime, and the result looks good!)printeasyocrReadertorchrandtono_gradronnxexportloadchecker check_modelgraphrr onnxruntimeInferenceSession get_inputsnamerunshapenptestingassert_allclose)detector_onnx_save_pathin_shape lang_listr r r dynamicdevicer rr ocr_reader dummy_input y_torch_outfeature_torch_out detector_onnxr5 ort_sessionr% ort_inputs y_onnx_outfeature_onnx_outr#r#r$export_detector sf           rLcCst}|jdddtdgdd|jddtd d d |jd d ddd|jdddtgddd|jddtdd|jddtdd|}|j}|dkrOdn||_t|jdkr]t d|S)Nz-lz --lang_list+rz--l en ch_sim ... (language lists for easyocr))nargstypedefaulthelpz-sz--detector_onnx_save_pathzdetector_craft.onnxzbexport detector onnx file path ending in .onnxDo not pass in this flag to avoid exporting detector)rOrPrQz-dz --dynamic store_truez)Dynamic input output shapes for detector)actionrQz-isz --in_shaperz/-is 1 3 608 800 (bsize, channel, height, width)z-mz--model_storage_directoryz'model storage directory for craft model)rOrQz-uz--user_network_directoryzuser model storage directoryNonezQInput shape must have four values (bsize, channel, height, width) eg. 1 3 608 800) argparseArgumentParser add_argumentstrint parse_argsr>lenr? ValueError)parserargsdpathr#r#r$r[VsB   r[cCs*t}t|j|j|j|j|j|jddS)N)r>r?r@r r rA)r[rLr>r?r@r r rA)r_r#r#r$maints ra__main__) rVr/r+r)r!r;rLr[ra__name__r#r#r#r$s* M