├── .gitignore ├── LICENSE ├── README-zh_CN.md ├── README.md ├── assets ├── Mesh-candidate Bestfit.png ├── comp.png ├── docsynth300k.png ├── example │ ├── academic.jpg │ ├── exam_paper.jpg │ ├── financial.jpg │ ├── fuzzy_scan.jpg │ ├── poster.jpg │ ├── ppt.jpg │ └── textbook.jpg ├── radar.png ├── script.sh ├── showcase.png └── visualization.png ├── demo.py ├── doclayout_yolo ├── __init__.py ├── assets │ ├── bus.jpg │ └── zidane.jpg ├── cfg │ ├── __init__.py │ ├── datasets │ │ ├── d4la.yaml │ │ ├── doclaynet.yaml │ │ └── docsynth300k.yaml │ ├── default.yaml │ ├── models │ │ ├── README.md │ │ ├── rt-detr │ │ │ ├── rtdetr-l.yaml │ │ │ ├── rtdetr-resnet101.yaml │ │ │ ├── rtdetr-resnet50.yaml │ │ │ └── rtdetr-x.yaml │ │ ├── v10 │ │ │ ├── yolov10b.yaml │ │ │ ├── yolov10l.yaml │ │ │ ├── yolov10m-doclayout.yaml │ │ │ ├── yolov10m.yaml │ │ │ ├── yolov10n.yaml │ │ │ ├── yolov10s.yaml │ │ │ └── yolov10x.yaml │ │ ├── v3 │ │ │ ├── yolov3-spp.yaml │ │ │ ├── yolov3-tiny.yaml │ │ │ └── yolov3.yaml │ │ ├── v5 │ │ │ ├── yolov5-p6.yaml │ │ │ └── yolov5.yaml │ │ ├── v6 │ │ │ └── yolov6.yaml │ │ ├── v8 │ │ │ ├── yolov8-cls-resnet101.yaml │ │ │ ├── yolov8-cls-resnet50.yaml │ │ │ ├── yolov8-cls.yaml │ │ │ ├── yolov8-ghost-p2.yaml │ │ │ ├── yolov8-ghost-p6.yaml │ │ │ ├── yolov8-ghost.yaml │ │ │ ├── yolov8-obb.yaml │ │ │ ├── yolov8-p2.yaml │ │ │ ├── yolov8-p6.yaml │ │ │ ├── yolov8-pose-p6.yaml │ │ │ ├── yolov8-pose.yaml │ │ │ ├── yolov8-rtdetr.yaml │ │ │ ├── yolov8-seg-p6.yaml │ │ │ ├── yolov8-seg.yaml │ │ │ ├── yolov8-world.yaml │ │ │ ├── yolov8-worldv2.yaml │ │ │ └── yolov8.yaml │ │ └── v9 │ │ │ ├── yolov9c.yaml │ │ │ └── yolov9e.yaml │ └── trackers │ │ ├── botsort.yaml │ │ └── bytetrack.yaml ├── data │ ├── __init__.py │ ├── annotator.py │ ├── augment.py │ ├── base.py │ ├── build.py │ ├── converter.py │ ├── dataset.py │ ├── explorer │ │ ├── __init__.py │ │ ├── explorer.py │ │ ├── gui │ │ │ ├── __init__.py │ │ │ └── dash.py │ │ └── utils.py │ ├── loaders.py │ ├── split_dota.py │ └── utils.py ├── engine │ ├── __init__.py │ ├── exporter.py │ ├── model.py │ ├── predictor.py │ ├── results.py │ ├── trainer.py │ ├── tuner.py │ └── validator.py ├── hub │ ├── __init__.py │ ├── auth.py │ ├── session.py │ └── utils.py ├── models │ ├── __init__.py │ ├── fastsam │ │ ├── __init__.py │ │ ├── model.py │ │ ├── predict.py │ │ ├── prompt.py │ │ ├── utils.py │ │ └── val.py │ ├── nas │ │ ├── __init__.py │ │ ├── model.py │ │ ├── predict.py │ │ └── val.py │ ├── rtdetr │ │ ├── __init__.py │ │ ├── model.py │ │ ├── predict.py │ │ ├── train.py │ │ └── val.py │ ├── sam │ │ ├── __init__.py │ │ ├── amg.py │ │ ├── build.py │ │ ├── model.py │ │ ├── modules │ │ │ ├── __init__.py │ │ │ ├── decoders.py │ │ │ ├── encoders.py │ │ │ ├── sam.py │ │ │ ├── tiny_encoder.py │ │ │ └── transformer.py │ │ └── predict.py │ ├── utils │ │ ├── __init__.py │ │ ├── loss.py │ │ └── ops.py │ ├── yolo │ │ ├── __init__.py │ │ ├── classify │ │ │ ├── __init__.py │ │ │ ├── predict.py │ │ │ ├── train.py │ │ │ └── val.py │ │ ├── detect │ │ │ ├── __init__.py │ │ │ ├── predict.py │ │ │ ├── train.py │ │ │ ├── val.py │ │ │ └── val_backup.py │ │ ├── model.py │ │ ├── obb │ │ │ ├── __init__.py │ │ │ ├── predict.py │ │ │ ├── train.py │ │ │ └── val.py │ │ ├── pose │ │ │ ├── __init__.py │ │ │ ├── predict.py │ │ │ ├── train.py │ │ │ └── val.py │ │ └── segment │ │ │ ├── __init__.py │ │ │ ├── predict.py │ │ │ ├── train.py │ │ │ └── val.py │ └── yolov10 │ │ ├── __init__.py │ │ ├── model.py │ │ ├── predict.py │ │ ├── train.py │ │ └── val.py ├── nn │ ├── __init__.py │ ├── autobackend.py │ ├── modules │ │ ├── __init__.py │ │ ├── block.py │ │ ├── conv.py │ │ ├── g2l_crm.py │ │ ├── head.py │ │ ├── modeling │ │ │ ├── __init__.py │ │ │ ├── aspp.py │ │ │ ├── backbone │ │ │ │ ├── __init__.py │ │ │ │ ├── drn.py │ │ │ │ ├── mobilenet.py │ │ │ │ ├── resnet.py │ │ │ │ └── xception.py │ │ │ ├── decoder.py │ │ │ ├── deeplab.py │ │ │ └── sync_batchnorm │ │ │ │ ├── __init__.py │ │ │ │ ├── batchnorm.py │ │ │ │ ├── comm.py │ │ │ │ ├── replicate.py │ │ │ │ └── unittest.py │ │ ├── transformer.py │ │ └── utils.py │ └── tasks.py ├── solutions │ ├── __init__.py │ ├── ai_gym.py │ ├── distance_calculation.py │ ├── heatmap.py │ ├── object_counter.py │ └── speed_estimation.py ├── trackers │ ├── README.md │ ├── __init__.py │ ├── basetrack.py │ ├── bot_sort.py │ ├── byte_tracker.py │ ├── track.py │ └── utils │ │ ├── __init__.py │ │ ├── gmc.py │ │ ├── kalman_filter.py │ │ └── matching.py └── utils │ ├── __init__.py │ ├── autobatch.py │ ├── benchmarks.py │ ├── callbacks │ ├── __init__.py │ ├── base.py │ ├── clearml.py │ ├── comet.py │ ├── dvc.py │ ├── hub.py │ ├── mlflow.py │ ├── neptune.py │ ├── raytune.py │ ├── tensorboard.py │ └── wb.py │ ├── checks.py │ ├── dist.py │ ├── downloads.py │ ├── errors.py │ ├── files.py │ ├── instance.py │ ├── loss.py │ ├── metrics.py │ ├── ops.py │ ├── patches.py │ ├── plotting.py │ ├── tal.py │ ├── torch_utils.py │ ├── triton.py │ └── tuner.py ├── format_docsynth300k.py ├── mesh-candidate_bestfit ├── README.md ├── augmentation.py ├── bestfit_generator.py ├── combine_layouts.py ├── map_dict.py ├── rendering.py ├── utils │ ├── base.py │ └── process.py └── visualize.ipynb ├── pyproject.toml ├── train.py └── val.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/LICENSE -------------------------------------------------------------------------------- /README-zh_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/README-zh_CN.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/README.md -------------------------------------------------------------------------------- /assets/Mesh-candidate Bestfit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/assets/Mesh-candidate Bestfit.png -------------------------------------------------------------------------------- /assets/comp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/assets/comp.png -------------------------------------------------------------------------------- /assets/docsynth300k.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/assets/docsynth300k.png -------------------------------------------------------------------------------- /assets/example/academic.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/assets/example/academic.jpg -------------------------------------------------------------------------------- /assets/example/exam_paper.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/assets/example/exam_paper.jpg -------------------------------------------------------------------------------- /assets/example/financial.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/assets/example/financial.jpg -------------------------------------------------------------------------------- /assets/example/fuzzy_scan.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/assets/example/fuzzy_scan.jpg -------------------------------------------------------------------------------- /assets/example/poster.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/assets/example/poster.jpg -------------------------------------------------------------------------------- /assets/example/ppt.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/assets/example/ppt.jpg -------------------------------------------------------------------------------- /assets/example/textbook.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/assets/example/textbook.jpg -------------------------------------------------------------------------------- /assets/radar.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/assets/radar.png -------------------------------------------------------------------------------- /assets/script.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/assets/script.sh -------------------------------------------------------------------------------- /assets/showcase.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/assets/showcase.png -------------------------------------------------------------------------------- /assets/visualization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/assets/visualization.png -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/demo.py -------------------------------------------------------------------------------- /doclayout_yolo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/assets/bus.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/assets/bus.jpg -------------------------------------------------------------------------------- /doclayout_yolo/assets/zidane.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/assets/zidane.jpg -------------------------------------------------------------------------------- /doclayout_yolo/cfg/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/cfg/datasets/d4la.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/datasets/d4la.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/datasets/doclaynet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/datasets/doclaynet.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/datasets/docsynth300k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/datasets/docsynth300k.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/default.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/README.md -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/rt-detr/rtdetr-l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/rt-detr/rtdetr-l.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/rt-detr/rtdetr-resnet101.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/rt-detr/rtdetr-resnet101.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/rt-detr/rtdetr-resnet50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/rt-detr/rtdetr-resnet50.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/rt-detr/rtdetr-x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/rt-detr/rtdetr-x.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v10/yolov10b.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v10/yolov10b.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v10/yolov10l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v10/yolov10l.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v10/yolov10m-doclayout.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v10/yolov10m-doclayout.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v10/yolov10m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v10/yolov10m.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v10/yolov10n.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v10/yolov10n.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v10/yolov10s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v10/yolov10s.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v10/yolov10x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v10/yolov10x.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v3/yolov3-spp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v3/yolov3-spp.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v3/yolov3-tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v3/yolov3-tiny.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v3/yolov3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v3/yolov3.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v5/yolov5-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v5/yolov5-p6.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v5/yolov5.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v5/yolov5.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v6/yolov6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v6/yolov6.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8-cls-resnet101.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8-cls-resnet101.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8-cls-resnet50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8-cls-resnet50.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8-cls.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8-cls.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8-ghost-p2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8-ghost-p2.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8-ghost-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8-ghost-p6.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8-ghost.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8-ghost.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8-obb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8-obb.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8-p2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8-p2.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8-p6.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8-pose-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8-pose-p6.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8-pose.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8-pose.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8-rtdetr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8-rtdetr.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8-seg-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8-seg-p6.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8-seg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8-seg.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8-world.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8-world.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8-worldv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8-worldv2.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v8/yolov8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v8/yolov8.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v9/yolov9c.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v9/yolov9c.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/models/v9/yolov9e.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/models/v9/yolov9e.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/trackers/botsort.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/trackers/botsort.yaml -------------------------------------------------------------------------------- /doclayout_yolo/cfg/trackers/bytetrack.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/cfg/trackers/bytetrack.yaml -------------------------------------------------------------------------------- /doclayout_yolo/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/data/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/data/annotator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/data/annotator.py -------------------------------------------------------------------------------- /doclayout_yolo/data/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/data/augment.py -------------------------------------------------------------------------------- /doclayout_yolo/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/data/base.py -------------------------------------------------------------------------------- /doclayout_yolo/data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/data/build.py -------------------------------------------------------------------------------- /doclayout_yolo/data/converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/data/converter.py -------------------------------------------------------------------------------- /doclayout_yolo/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/data/dataset.py -------------------------------------------------------------------------------- /doclayout_yolo/data/explorer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/data/explorer/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/data/explorer/explorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/data/explorer/explorer.py -------------------------------------------------------------------------------- /doclayout_yolo/data/explorer/gui/__init__.py: -------------------------------------------------------------------------------- 1 | # Ultralytics YOLO 🚀, AGPL-3.0 license 2 | -------------------------------------------------------------------------------- /doclayout_yolo/data/explorer/gui/dash.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/data/explorer/gui/dash.py -------------------------------------------------------------------------------- /doclayout_yolo/data/explorer/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/data/explorer/utils.py -------------------------------------------------------------------------------- /doclayout_yolo/data/loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/data/loaders.py -------------------------------------------------------------------------------- /doclayout_yolo/data/split_dota.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/data/split_dota.py -------------------------------------------------------------------------------- /doclayout_yolo/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/data/utils.py -------------------------------------------------------------------------------- /doclayout_yolo/engine/__init__.py: -------------------------------------------------------------------------------- 1 | # Ultralytics YOLO 🚀, AGPL-3.0 license 2 | -------------------------------------------------------------------------------- /doclayout_yolo/engine/exporter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/engine/exporter.py -------------------------------------------------------------------------------- /doclayout_yolo/engine/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/engine/model.py -------------------------------------------------------------------------------- /doclayout_yolo/engine/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/engine/predictor.py -------------------------------------------------------------------------------- /doclayout_yolo/engine/results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/engine/results.py -------------------------------------------------------------------------------- /doclayout_yolo/engine/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/engine/trainer.py -------------------------------------------------------------------------------- /doclayout_yolo/engine/tuner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/engine/tuner.py -------------------------------------------------------------------------------- /doclayout_yolo/engine/validator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/engine/validator.py -------------------------------------------------------------------------------- /doclayout_yolo/hub/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/hub/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/hub/auth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/hub/auth.py -------------------------------------------------------------------------------- /doclayout_yolo/hub/session.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/hub/session.py -------------------------------------------------------------------------------- /doclayout_yolo/hub/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/hub/utils.py -------------------------------------------------------------------------------- /doclayout_yolo/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/models/fastsam/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/fastsam/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/models/fastsam/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/fastsam/model.py -------------------------------------------------------------------------------- /doclayout_yolo/models/fastsam/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/fastsam/predict.py -------------------------------------------------------------------------------- /doclayout_yolo/models/fastsam/prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/fastsam/prompt.py -------------------------------------------------------------------------------- /doclayout_yolo/models/fastsam/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/fastsam/utils.py -------------------------------------------------------------------------------- /doclayout_yolo/models/fastsam/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/fastsam/val.py -------------------------------------------------------------------------------- /doclayout_yolo/models/nas/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/nas/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/models/nas/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/nas/model.py -------------------------------------------------------------------------------- /doclayout_yolo/models/nas/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/nas/predict.py -------------------------------------------------------------------------------- /doclayout_yolo/models/nas/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/nas/val.py -------------------------------------------------------------------------------- /doclayout_yolo/models/rtdetr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/rtdetr/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/models/rtdetr/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/rtdetr/model.py -------------------------------------------------------------------------------- /doclayout_yolo/models/rtdetr/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/rtdetr/predict.py -------------------------------------------------------------------------------- /doclayout_yolo/models/rtdetr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/rtdetr/train.py -------------------------------------------------------------------------------- /doclayout_yolo/models/rtdetr/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/rtdetr/val.py -------------------------------------------------------------------------------- /doclayout_yolo/models/sam/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/sam/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/models/sam/amg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/sam/amg.py -------------------------------------------------------------------------------- /doclayout_yolo/models/sam/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/sam/build.py -------------------------------------------------------------------------------- /doclayout_yolo/models/sam/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/sam/model.py -------------------------------------------------------------------------------- /doclayout_yolo/models/sam/modules/__init__.py: -------------------------------------------------------------------------------- 1 | # Ultralytics YOLO 🚀, AGPL-3.0 license 2 | -------------------------------------------------------------------------------- /doclayout_yolo/models/sam/modules/decoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/sam/modules/decoders.py -------------------------------------------------------------------------------- /doclayout_yolo/models/sam/modules/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/sam/modules/encoders.py -------------------------------------------------------------------------------- /doclayout_yolo/models/sam/modules/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/sam/modules/sam.py -------------------------------------------------------------------------------- /doclayout_yolo/models/sam/modules/tiny_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/sam/modules/tiny_encoder.py -------------------------------------------------------------------------------- /doclayout_yolo/models/sam/modules/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/sam/modules/transformer.py -------------------------------------------------------------------------------- /doclayout_yolo/models/sam/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/sam/predict.py -------------------------------------------------------------------------------- /doclayout_yolo/models/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Ultralytics YOLO 🚀, AGPL-3.0 license 2 | -------------------------------------------------------------------------------- /doclayout_yolo/models/utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/utils/loss.py -------------------------------------------------------------------------------- /doclayout_yolo/models/utils/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/utils/ops.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/classify/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/classify/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/classify/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/classify/predict.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/classify/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/classify/train.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/classify/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/classify/val.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/detect/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/detect/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/detect/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/detect/predict.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/detect/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/detect/train.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/detect/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/detect/val.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/detect/val_backup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/detect/val_backup.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/model.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/obb/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/obb/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/obb/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/obb/predict.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/obb/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/obb/train.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/obb/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/obb/val.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/pose/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/pose/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/pose/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/pose/predict.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/pose/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/pose/train.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/pose/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/pose/val.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/segment/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/segment/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/segment/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/segment/predict.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/segment/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/segment/train.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolo/segment/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolo/segment/val.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolov10/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolov10/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolov10/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolov10/model.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolov10/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolov10/predict.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolov10/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolov10/train.py -------------------------------------------------------------------------------- /doclayout_yolo/models/yolov10/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/models/yolov10/val.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/autobackend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/autobackend.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/block.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/conv.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/g2l_crm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/g2l_crm.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/head.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/modeling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/modeling/aspp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/modeling/aspp.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/modeling/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/modeling/backbone/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/modeling/backbone/drn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/modeling/backbone/drn.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/modeling/backbone/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/modeling/backbone/mobilenet.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/modeling/backbone/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/modeling/backbone/resnet.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/modeling/backbone/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/modeling/backbone/xception.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/modeling/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/modeling/decoder.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/modeling/deeplab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/modeling/deeplab.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/modeling/sync_batchnorm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/modeling/sync_batchnorm/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/modeling/sync_batchnorm/batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/modeling/sync_batchnorm/batchnorm.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/modeling/sync_batchnorm/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/modeling/sync_batchnorm/comm.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/modeling/sync_batchnorm/replicate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/modeling/sync_batchnorm/replicate.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/modeling/sync_batchnorm/unittest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/modeling/sync_batchnorm/unittest.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/transformer.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/modules/utils.py -------------------------------------------------------------------------------- /doclayout_yolo/nn/tasks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/nn/tasks.py -------------------------------------------------------------------------------- /doclayout_yolo/solutions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/solutions/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/solutions/ai_gym.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/solutions/ai_gym.py -------------------------------------------------------------------------------- /doclayout_yolo/solutions/distance_calculation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/solutions/distance_calculation.py -------------------------------------------------------------------------------- /doclayout_yolo/solutions/heatmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/solutions/heatmap.py -------------------------------------------------------------------------------- /doclayout_yolo/solutions/object_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/solutions/object_counter.py -------------------------------------------------------------------------------- /doclayout_yolo/solutions/speed_estimation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/solutions/speed_estimation.py -------------------------------------------------------------------------------- /doclayout_yolo/trackers/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/trackers/README.md -------------------------------------------------------------------------------- /doclayout_yolo/trackers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/trackers/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/trackers/basetrack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/trackers/basetrack.py -------------------------------------------------------------------------------- /doclayout_yolo/trackers/bot_sort.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/trackers/bot_sort.py -------------------------------------------------------------------------------- /doclayout_yolo/trackers/byte_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/trackers/byte_tracker.py -------------------------------------------------------------------------------- /doclayout_yolo/trackers/track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/trackers/track.py -------------------------------------------------------------------------------- /doclayout_yolo/trackers/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Ultralytics YOLO 🚀, AGPL-3.0 license 2 | -------------------------------------------------------------------------------- /doclayout_yolo/trackers/utils/gmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/trackers/utils/gmc.py -------------------------------------------------------------------------------- /doclayout_yolo/trackers/utils/kalman_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/trackers/utils/kalman_filter.py -------------------------------------------------------------------------------- /doclayout_yolo/trackers/utils/matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/trackers/utils/matching.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/autobatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/autobatch.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/benchmarks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/benchmarks.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/callbacks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/callbacks/__init__.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/callbacks/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/callbacks/base.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/callbacks/clearml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/callbacks/clearml.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/callbacks/comet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/callbacks/comet.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/callbacks/dvc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/callbacks/dvc.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/callbacks/hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/callbacks/hub.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/callbacks/mlflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/callbacks/mlflow.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/callbacks/neptune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/callbacks/neptune.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/callbacks/raytune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/callbacks/raytune.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/callbacks/tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/callbacks/tensorboard.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/callbacks/wb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/callbacks/wb.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/checks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/checks.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/dist.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/downloads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/downloads.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/errors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/errors.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/files.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/instance.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/loss.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/metrics.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/ops.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/patches.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/patches.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/plotting.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/tal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/tal.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/torch_utils.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/triton.py -------------------------------------------------------------------------------- /doclayout_yolo/utils/tuner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/doclayout_yolo/utils/tuner.py -------------------------------------------------------------------------------- /format_docsynth300k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/format_docsynth300k.py -------------------------------------------------------------------------------- /mesh-candidate_bestfit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/mesh-candidate_bestfit/README.md -------------------------------------------------------------------------------- /mesh-candidate_bestfit/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/mesh-candidate_bestfit/augmentation.py -------------------------------------------------------------------------------- /mesh-candidate_bestfit/bestfit_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/mesh-candidate_bestfit/bestfit_generator.py -------------------------------------------------------------------------------- /mesh-candidate_bestfit/combine_layouts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/mesh-candidate_bestfit/combine_layouts.py -------------------------------------------------------------------------------- /mesh-candidate_bestfit/map_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/mesh-candidate_bestfit/map_dict.py -------------------------------------------------------------------------------- /mesh-candidate_bestfit/rendering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/mesh-candidate_bestfit/rendering.py -------------------------------------------------------------------------------- /mesh-candidate_bestfit/utils/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/mesh-candidate_bestfit/utils/base.py -------------------------------------------------------------------------------- /mesh-candidate_bestfit/utils/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/mesh-candidate_bestfit/utils/process.py -------------------------------------------------------------------------------- /mesh-candidate_bestfit/visualize.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/mesh-candidate_bestfit/visualize.ipynb -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/pyproject.toml -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/train.py -------------------------------------------------------------------------------- /val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opendatalab/DocLayout-YOLO/HEAD/val.py --------------------------------------------------------------------------------