MCPcopy Index your code

hub / github.com/mindee/doctr / endpoints

Endpoints52 in github.com/mindee/doctr

Routeperform_kie
router.post("/", response_model=list[KIEOut], status_code=status.HTTP_200_OK, summary="Perform KIE")
api/app/routes/kie.py:None
Routeperform_ocr
router.post("/", response_model=list[OCROut], status_code=status.HTTP_200_OK, summary="Perform OCR")
api/app/routes/ocr.py:None
Routetest_artefact_detection
pytest.mark.parametrize( "input_size, num_samples", [ [[512, 512], 3], # Actual set has 2700
tests/pytorch/test_datasets_pt.py:None
Routetest_box_iou
pytest.mark.parametrize( "box1, box2, iou, abs_tol", [ [[[0, 0, 0.5, 0.5]], [[0, 0, 0.5, 0.5]]
tests/common/test_utils_metrics.py:None
Routetest_channel_shuffle
pytest.mark.parametrize( "input_dtype, input_size", [ [torch.float32, (3, 32, 32)], [t
tests/pytorch/test_transforms_pt.py:None
Routetest_classification_architectures
pytest.mark.parametrize( "arch_name, input_shape, output_size", [ ["vgg16_bn_r", (3, 32, 32),
tests/pytorch/test_models_classification_pt.py:None
Routetest_classification_models
pytest.mark.parametrize( "arch_name, input_shape", [ ["mobilenet_v3_small_crop_orientation", (
tests/pytorch/test_models_classification_pt.py:None
Routetest_cocotext_dataset
pytest.mark.parametrize( "input_size, num_samples, recognition, detection", [ [[512, 512], 3,
tests/pytorch/test_datasets_pt.py:None
Routetest_convert_target_to_relative
pytest.mark.parametrize( "target", [ # Boxes {"boxes": np.random.rand(3, 4), "labels":
tests/common/test_datasets_utils.py:None
Routetest_cord
pytest.mark.parametrize( "input_size, num_samples, recognition, detection", [ [[512, 512], 3,
tests/pytorch/test_datasets_pt.py:None
Routetest_detection_models
pytest.mark.parametrize( "arch_name, input_shape, output_size, out_prob", [ ["db_resnet34", (3
tests/pytorch/test_models_detection_pt.py:None
Routetest_download_from_url
patch("doctr.utils.data._urlretrieve")
tests/common/test_utils_data.py:None
Routetest_download_from_url_customizing_cache_dir
patch.dict(os.environ, {"DOCTR_CACHE_DIR": "/test"}, clear=True)
tests/common/test_utils_data.py:None
Routetest_download_from_url_error_creating_directory
patch.dict(os.environ, {"HOME": "/"}, clear=True)
tests/common/test_utils_data.py:None
Routetest_download_from_url_error_creating_directory_with_env_var
patch.dict(os.environ, {"HOME": "/", "DOCTR_CACHE_DIR": "/test"}, clear=True)
tests/common/test_utils_data.py:None
Routetest_encode_decode
pytest.mark.parametrize( "input_str", [ "frtvorêt", "for98€t", "uéîUY",
tests/common/test_datasets_utils.py:None
Routetest_encode_sequences
pytest.mark.parametrize( "sequences, vocab, target_size, sos, eos, pad, dynamic_len, error, out_shape, gts
tests/common/test_datasets_utils.py:None
Routetest_funsd
pytest.mark.parametrize( "input_size, num_samples, recognition, detection", [ [[512, 512], 3,
tests/pytorch/test_datasets_pt.py:None
Routetest_gaussian_blur
pytest.mark.parametrize( "input_dtype, input_shape", [ [torch.float32, (3, 32, 32)], [
tests/pytorch/test_transforms_pt.py:None
Routetest_gaussian_noise
pytest.mark.parametrize( "input_dtype,input_shape", [ [torch.float32, (3, 32, 32)], [t
tests/pytorch/test_transforms_pt.py:None
Routetest_ic03
pytest.mark.parametrize( "input_size, num_samples, recognition, detection", [ [[512, 512], 3,
tests/pytorch/test_datasets_pt.py:None
Routetest_ic13_dataset
pytest.mark.parametrize( "input_size, num_samples, recognition, detection", [ [[512, 512], 5,
tests/pytorch/test_datasets_pt.py:None
Routetest_iiit5k
pytest.mark.parametrize( "input_size, num_samples, recognition, detection", [ [[32, 128], 1, F
tests/pytorch/test_datasets_pt.py:None
Routetest_imgur5k_dataset
pytest.mark.parametrize( "input_size, num_samples, recognition, detection", [ [[512, 512], 3,
tests/pytorch/test_datasets_pt.py:None
Routetest_merge_multi_strings
pytest.mark.parametrize( "seq_list, overlap_ratio, last_overlap_ratio, merged", [ # One charac
tests/common/test_models_recognition_utils.py:None
Routetest_merge_strings
pytest.mark.parametrize( "a, b, overlap_ratio, merged", [ # Last character of first string and
tests/common/test_models_recognition_utils.py:None
Routetest_models_onnx_export
pytest.mark.parametrize( "arch_name, input_shape, output_size", [ ["vgg16_bn_r", (3, 32, 32),
tests/pytorch/test_models_classification_pt.py:None
Routetest_models_onnx_export
pytest.mark.parametrize( "arch_name, input_shape, output_size", [ ["db_resnet34", (3, 512, 512
tests/pytorch/test_models_detection_pt.py:None
Routetest_models_onnx_export
pytest.mark.parametrize( "arch_name, input_shape", [ ["crnn_vgg16_bn", (3, 32, 128)],
tests/pytorch/test_models_recognition_pt.py:None
Routetest_multithread_exec
pytest.mark.parametrize( "input_seq, func, output_seq", [ [[1, 2, 3], lambda x: 2 * x, [2, 4,
tests/common/test_utils_multithreading.py:None
Routetest_multithread_exec_multiprocessing_disable
patch.dict(os.environ, {"DOCTR_MULTIPROCESSING_DISABLE": "TRUE"}, clear=True)
tests/common/test_utils_multithreading.py:None
Routetest_polygon_iou
pytest.mark.parametrize( "rbox1, rbox2, iou, abs_tol", [ [[[[0, 0], [0.5, 0], [0.5, 0.5], [0,
tests/common/test_utils_metrics.py:None
Routetest_preprocessor
pytest.mark.parametrize( "batch_size, output_size, input_tensor, expected_batches, expected_value", [
tests/pytorch/test_models_preprocessor_pt.py:None
Routetest_random_crop
pytest.mark.parametrize( "target", [ np.array([[15, 20, 35, 30]]), # box np.array([[[
tests/pytorch/test_transforms_pt.py:None
Routetest_random_resize
pytest.mark.parametrize( "p,preserve_aspect_ratio,symmetric_pad,target", [ [1, True, False, np
tests/pytorch/test_transforms_pt.py:None
Routetest_random_shadow
pytest.mark.parametrize( "input_dtype,input_shape", [ [torch.float32, (3, 32, 32)], [t
tests/pytorch/test_transforms_pt.py:None
Routetest_randomhorizontalflip
pytest.mark.parametrize( "p,target", [ [1, np.array([[0.1, 0.1, 0.3, 0.4]], dtype=np.float32)]
tests/pytorch/test_transforms_pt.py:None
Routetest_reco_postprocessors
pytest.mark.parametrize( "post_processor, input_shape", [ [CTCPostProcessor, [2, 119, 30]],
tests/pytorch/test_models_recognition_pt.py:None
Routetest_recognition_models
pytest.mark.parametrize( "arch_name, input_shape", [ ["crnn_vgg16_bn", (3, 32, 128)],
tests/pytorch/test_models_recognition_pt.py:None
Routetest_recognition_zoo
pytest.mark.parametrize( "input_shape", [ (128, 128, 3), (32, 1024, 3), # test case s
tests/pytorch/test_models_recognition_pt.py:None
Routetest_resolve_lines
pytest.mark.parametrize( "input_boxes, lines", [ [[[0, 0.5, 0.1, 0.6], [0, 0.3, 0.2, 0.4], [0,
tests/common/test_models_builder.py:None
Routetest_sort_boxes
pytest.mark.parametrize( "input_boxes, sorted_idxs", [ [[[0, 0.5, 0.1, 0.6], [0, 0.3, 0.2, 0.4
tests/common/test_models_builder.py:None
Routetest_split_crops
pytest.mark.parametrize( "crops, max_ratio, target_ratio, target_overlap_ratio, num_crops", [
tests/common/test_models_recognition_predictor.py:None
Routetest_split_crops_cases
pytest.mark.parametrize( "inputs, max_ratio, target_ratio, target_overlap_ratio, expected_remap_required,
tests/common/test_models_recognition_predictor.py:None
Routetest_sroie
pytest.mark.parametrize( "input_size, num_samples, recognition, detection", [ [[512, 512], 3,
tests/pytorch/test_datasets_pt.py:None
Routetest_svhn
pytest.mark.parametrize( "input_size, num_samples, recognition, detection", [ [[32, 128], 3, F
tests/pytorch/test_datasets_pt.py:None
Routetest_svt
pytest.mark.parametrize( "input_size, num_samples, recognition, detection", [ [[512, 512], 3,
tests/pytorch/test_datasets_pt.py:None
Routetest_synthtext
pytest.mark.parametrize( "input_size, num_samples, recognition, detection", [ [[512, 512], 2,
tests/pytorch/test_datasets_pt.py:None
Routetest_translate
pytest.mark.parametrize( "input_str, vocab, output_str", [ ["f orêt", "latin", "foret"],
tests/common/test_datasets_utils.py:None
Routetest_wildreceipt_dataset
pytest.mark.parametrize( "input_size, num_samples, recognition, detection", [ [[512, 512], 2,
tests/pytorch/test_datasets_pt.py:None
Routetext_detection
router.post("/", response_model=list[DetectionOut], status_code=status.HTTP_200_OK, summary="Perform text dete
api/app/routes/detection.py:None
Routetext_recognition
router.post( "/", response_model=list[RecognitionOut], status_code=status.HTTP_200_OK, summary="Perform te
api/app/routes/recognition.py:None