dnt 0.2.4__py3-none-any.whl → 0.3.1.8__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- dnt/__init__.py +3 -2
- dnt/analysis/__init__.py +3 -2
- dnt/analysis/count.py +54 -37
- dnt/analysis/interaction2.py +518 -0
- dnt/analysis/stop.py +22 -17
- dnt/analysis/stop2.py +289 -0
- dnt/analysis/stop3.py +758 -0
- dnt/detect/signal/detector.py +326 -0
- dnt/detect/timestamp.py +105 -0
- dnt/detect/yolov8/detector.py +179 -36
- dnt/detect/yolov8/segmentor.py +60 -2
- dnt/engine/__init__.py +8 -0
- dnt/engine/bbox_interp.py +83 -0
- dnt/engine/bbox_iou.py +20 -0
- dnt/engine/cluster.py +31 -0
- dnt/engine/iob.py +66 -0
- dnt/filter/filter.py +333 -2
- dnt/label/labeler.py +4 -4
- dnt/label/labeler2.py +631 -0
- dnt/shared/__init__.py +2 -1
- dnt/shared/data/coco.names +0 -0
- dnt/shared/data/openimages.names +0 -0
- dnt/shared/data/voc.names +0 -0
- dnt/shared/download.py +12 -0
- dnt/shared/synhcro.py +150 -0
- dnt/shared/util.py +17 -4
- dnt/third_party/fast-reid/__init__.py +1 -0
- dnt/third_party/fast-reid/configs/Base-AGW.yml +19 -0
- dnt/third_party/fast-reid/configs/Base-MGN.yml +12 -0
- dnt/third_party/fast-reid/configs/Base-SBS.yml +63 -0
- dnt/third_party/fast-reid/configs/Base-bagtricks.yml +76 -0
- dnt/third_party/fast-reid/configs/DukeMTMC/AGW_R101-ibn.yml +12 -0
- dnt/third_party/fast-reid/configs/DukeMTMC/AGW_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/DukeMTMC/AGW_R50.yml +7 -0
- dnt/third_party/fast-reid/configs/DukeMTMC/AGW_S50.yml +11 -0
- dnt/third_party/fast-reid/configs/DukeMTMC/bagtricks_R101-ibn.yml +12 -0
- dnt/third_party/fast-reid/configs/DukeMTMC/bagtricks_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/DukeMTMC/bagtricks_R50.yml +7 -0
- dnt/third_party/fast-reid/configs/DukeMTMC/bagtricks_S50.yml +11 -0
- dnt/third_party/fast-reid/configs/DukeMTMC/mgn_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/DukeMTMC/sbs_R101-ibn.yml +12 -0
- dnt/third_party/fast-reid/configs/DukeMTMC/sbs_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/DukeMTMC/sbs_R50.yml +7 -0
- dnt/third_party/fast-reid/configs/DukeMTMC/sbs_S50.yml +11 -0
- dnt/third_party/fast-reid/configs/MOT17/AGW_R101-ibn.yml +12 -0
- dnt/third_party/fast-reid/configs/MOT17/AGW_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/MOT17/AGW_R50.yml +7 -0
- dnt/third_party/fast-reid/configs/MOT17/AGW_S50.yml +11 -0
- dnt/third_party/fast-reid/configs/MOT17/bagtricks_R101-ibn.yml +12 -0
- dnt/third_party/fast-reid/configs/MOT17/bagtricks_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/MOT17/bagtricks_R50.yml +7 -0
- dnt/third_party/fast-reid/configs/MOT17/bagtricks_S50.yml +11 -0
- dnt/third_party/fast-reid/configs/MOT17/mgn_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/MOT17/sbs_R101-ibn.yml +12 -0
- dnt/third_party/fast-reid/configs/MOT17/sbs_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/MOT17/sbs_R50.yml +7 -0
- dnt/third_party/fast-reid/configs/MOT17/sbs_S50.yml +11 -0
- dnt/third_party/fast-reid/configs/MOT20/AGW_R101-ibn.yml +12 -0
- dnt/third_party/fast-reid/configs/MOT20/AGW_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/MOT20/AGW_R50.yml +7 -0
- dnt/third_party/fast-reid/configs/MOT20/AGW_S50.yml +11 -0
- dnt/third_party/fast-reid/configs/MOT20/bagtricks_R101-ibn.yml +12 -0
- dnt/third_party/fast-reid/configs/MOT20/bagtricks_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/MOT20/bagtricks_R50.yml +7 -0
- dnt/third_party/fast-reid/configs/MOT20/bagtricks_S50.yml +11 -0
- dnt/third_party/fast-reid/configs/MOT20/mgn_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/MOT20/sbs_R101-ibn.yml +12 -0
- dnt/third_party/fast-reid/configs/MOT20/sbs_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/MOT20/sbs_R50.yml +7 -0
- dnt/third_party/fast-reid/configs/MOT20/sbs_S50.yml +11 -0
- dnt/third_party/fast-reid/configs/MSMT17/AGW_R101-ibn.yml +12 -0
- dnt/third_party/fast-reid/configs/MSMT17/AGW_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/MSMT17/AGW_R50.yml +7 -0
- dnt/third_party/fast-reid/configs/MSMT17/AGW_S50.yml +11 -0
- dnt/third_party/fast-reid/configs/MSMT17/bagtricks_R101-ibn.yml +13 -0
- dnt/third_party/fast-reid/configs/MSMT17/bagtricks_R50-ibn.yml +12 -0
- dnt/third_party/fast-reid/configs/MSMT17/bagtricks_R50.yml +7 -0
- dnt/third_party/fast-reid/configs/MSMT17/bagtricks_S50.yml +12 -0
- dnt/third_party/fast-reid/configs/MSMT17/mgn_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/MSMT17/sbs_R101-ibn.yml +12 -0
- dnt/third_party/fast-reid/configs/MSMT17/sbs_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/MSMT17/sbs_R50.yml +7 -0
- dnt/third_party/fast-reid/configs/MSMT17/sbs_S50.yml +11 -0
- dnt/third_party/fast-reid/configs/Market1501/AGW_R101-ibn.yml +12 -0
- dnt/third_party/fast-reid/configs/Market1501/AGW_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/Market1501/AGW_R50.yml +7 -0
- dnt/third_party/fast-reid/configs/Market1501/AGW_S50.yml +11 -0
- dnt/third_party/fast-reid/configs/Market1501/bagtricks_R101-ibn.yml +12 -0
- dnt/third_party/fast-reid/configs/Market1501/bagtricks_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/Market1501/bagtricks_R50.yml +7 -0
- dnt/third_party/fast-reid/configs/Market1501/bagtricks_S50.yml +11 -0
- dnt/third_party/fast-reid/configs/Market1501/bagtricks_vit.yml +88 -0
- dnt/third_party/fast-reid/configs/Market1501/mgn_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/Market1501/sbs_R101-ibn.yml +12 -0
- dnt/third_party/fast-reid/configs/Market1501/sbs_R50-ibn.yml +11 -0
- dnt/third_party/fast-reid/configs/Market1501/sbs_R50.yml +7 -0
- dnt/third_party/fast-reid/configs/Market1501/sbs_S50.yml +11 -0
- dnt/third_party/fast-reid/configs/VERIWild/bagtricks_R50-ibn.yml +35 -0
- dnt/third_party/fast-reid/configs/VeRi/sbs_R50-ibn.yml +35 -0
- dnt/third_party/fast-reid/configs/VehicleID/bagtricks_R50-ibn.yml +36 -0
- dnt/third_party/fast-reid/configs/__init__.py +0 -0
- dnt/third_party/fast-reid/fast_reid_interfece.py +175 -0
- dnt/third_party/fast-reid/fastreid/__init__.py +6 -0
- dnt/third_party/fast-reid/fastreid/config/__init__.py +15 -0
- dnt/third_party/fast-reid/fastreid/config/config.py +319 -0
- dnt/third_party/fast-reid/fastreid/config/defaults.py +329 -0
- dnt/third_party/fast-reid/fastreid/data/__init__.py +17 -0
- dnt/third_party/fast-reid/fastreid/data/build.py +194 -0
- dnt/third_party/fast-reid/fastreid/data/common.py +58 -0
- dnt/third_party/fast-reid/fastreid/data/data_utils.py +202 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/AirportALERT.py +50 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/__init__.py +43 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/bases.py +183 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/caviara.py +44 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/cuhk03.py +274 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/cuhk_sysu.py +58 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/dukemtmcreid.py +70 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/grid.py +44 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/iLIDS.py +45 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/lpw.py +49 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/market1501.py +89 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/msmt17.py +114 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/pes3d.py +44 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/pku.py +44 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/prai.py +43 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/prid.py +41 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/saivt.py +47 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/sensereid.py +47 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/shinpuhkan.py +48 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/sysu_mm.py +47 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/thermalworld.py +43 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/vehicleid.py +126 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/veri.py +69 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/veriwild.py +140 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/viper.py +45 -0
- dnt/third_party/fast-reid/fastreid/data/datasets/wildtracker.py +59 -0
- dnt/third_party/fast-reid/fastreid/data/samplers/__init__.py +18 -0
- dnt/third_party/fast-reid/fastreid/data/samplers/data_sampler.py +85 -0
- dnt/third_party/fast-reid/fastreid/data/samplers/imbalance_sampler.py +67 -0
- dnt/third_party/fast-reid/fastreid/data/samplers/triplet_sampler.py +260 -0
- dnt/third_party/fast-reid/fastreid/data/transforms/__init__.py +11 -0
- dnt/third_party/fast-reid/fastreid/data/transforms/autoaugment.py +806 -0
- dnt/third_party/fast-reid/fastreid/data/transforms/build.py +100 -0
- dnt/third_party/fast-reid/fastreid/data/transforms/functional.py +180 -0
- dnt/third_party/fast-reid/fastreid/data/transforms/transforms.py +161 -0
- dnt/third_party/fast-reid/fastreid/engine/__init__.py +15 -0
- dnt/third_party/fast-reid/fastreid/engine/defaults.py +490 -0
- dnt/third_party/fast-reid/fastreid/engine/hooks.py +534 -0
- dnt/third_party/fast-reid/fastreid/engine/launch.py +103 -0
- dnt/third_party/fast-reid/fastreid/engine/train_loop.py +357 -0
- dnt/third_party/fast-reid/fastreid/evaluation/__init__.py +6 -0
- dnt/third_party/fast-reid/fastreid/evaluation/clas_evaluator.py +81 -0
- dnt/third_party/fast-reid/fastreid/evaluation/evaluator.py +176 -0
- dnt/third_party/fast-reid/fastreid/evaluation/query_expansion.py +46 -0
- dnt/third_party/fast-reid/fastreid/evaluation/rank.py +200 -0
- dnt/third_party/fast-reid/fastreid/evaluation/rank_cylib/__init__.py +20 -0
- dnt/third_party/fast-reid/fastreid/evaluation/rank_cylib/setup.py +32 -0
- dnt/third_party/fast-reid/fastreid/evaluation/rank_cylib/test_cython.py +106 -0
- dnt/third_party/fast-reid/fastreid/evaluation/reid_evaluation.py +143 -0
- dnt/third_party/fast-reid/fastreid/evaluation/rerank.py +73 -0
- dnt/third_party/fast-reid/fastreid/evaluation/roc.py +90 -0
- dnt/third_party/fast-reid/fastreid/evaluation/testing.py +88 -0
- dnt/third_party/fast-reid/fastreid/layers/__init__.py +19 -0
- dnt/third_party/fast-reid/fastreid/layers/activation.py +59 -0
- dnt/third_party/fast-reid/fastreid/layers/any_softmax.py +80 -0
- dnt/third_party/fast-reid/fastreid/layers/batch_norm.py +205 -0
- dnt/third_party/fast-reid/fastreid/layers/context_block.py +113 -0
- dnt/third_party/fast-reid/fastreid/layers/drop.py +161 -0
- dnt/third_party/fast-reid/fastreid/layers/frn.py +199 -0
- dnt/third_party/fast-reid/fastreid/layers/gather_layer.py +30 -0
- dnt/third_party/fast-reid/fastreid/layers/helpers.py +31 -0
- dnt/third_party/fast-reid/fastreid/layers/non_local.py +54 -0
- dnt/third_party/fast-reid/fastreid/layers/pooling.py +124 -0
- dnt/third_party/fast-reid/fastreid/layers/se_layer.py +25 -0
- dnt/third_party/fast-reid/fastreid/layers/splat.py +109 -0
- dnt/third_party/fast-reid/fastreid/layers/weight_init.py +122 -0
- dnt/third_party/fast-reid/fastreid/modeling/__init__.py +23 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/__init__.py +18 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/build.py +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/mobilenet.py +195 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/mobilenetv3.py +283 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/osnet.py +525 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/__init__.py +4 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/config.py +396 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/effnet/EN-B0_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/effnet/EN-B1_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/effnet/EN-B2_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/effnet/EN-B3_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/effnet/EN-B4_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/effnet/EN-B5_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/effnet.py +281 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnet.py +596 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-1.6GF_dds_8gpu.yaml +26 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-12GF_dds_8gpu.yaml +26 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-16GF_dds_8gpu.yaml +26 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-200MF_dds_8gpu.yaml +26 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-3.2GF_dds_8gpu.yaml +26 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-32GF_dds_8gpu.yaml +26 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-4.0GF_dds_8gpu.yaml +26 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-400MF_dds_8gpu.yaml +26 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-6.4GF_dds_8gpu.yaml +26 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-600MF_dds_8gpu.yaml +26 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-8.0GF_dds_8gpu.yaml +26 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-800MF_dds_8gpu.yaml +26 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-1.6GF_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-12GF_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-16GF_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-200MF_dds_8gpu.yaml +26 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-3.2GF_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-32GF_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-4.0GF_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-400MF_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-6.4GF_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-600MF_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-8.0GF_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-800MF_dds_8gpu.yaml +27 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/repvgg.py +309 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/resnest.py +365 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/resnet.py +364 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/resnext.py +335 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/shufflenet.py +203 -0
- dnt/third_party/fast-reid/fastreid/modeling/backbones/vision_transformer.py +399 -0
- dnt/third_party/fast-reid/fastreid/modeling/heads/__init__.py +11 -0
- dnt/third_party/fast-reid/fastreid/modeling/heads/build.py +25 -0
- dnt/third_party/fast-reid/fastreid/modeling/heads/clas_head.py +36 -0
- dnt/third_party/fast-reid/fastreid/modeling/heads/embedding_head.py +151 -0
- dnt/third_party/fast-reid/fastreid/modeling/losses/__init__.py +12 -0
- dnt/third_party/fast-reid/fastreid/modeling/losses/circle_loss.py +71 -0
- dnt/third_party/fast-reid/fastreid/modeling/losses/cross_entroy_loss.py +54 -0
- dnt/third_party/fast-reid/fastreid/modeling/losses/focal_loss.py +92 -0
- dnt/third_party/fast-reid/fastreid/modeling/losses/triplet_loss.py +113 -0
- dnt/third_party/fast-reid/fastreid/modeling/losses/utils.py +48 -0
- dnt/third_party/fast-reid/fastreid/modeling/meta_arch/__init__.py +14 -0
- dnt/third_party/fast-reid/fastreid/modeling/meta_arch/baseline.py +188 -0
- dnt/third_party/fast-reid/fastreid/modeling/meta_arch/build.py +26 -0
- dnt/third_party/fast-reid/fastreid/modeling/meta_arch/distiller.py +140 -0
- dnt/third_party/fast-reid/fastreid/modeling/meta_arch/mgn.py +394 -0
- dnt/third_party/fast-reid/fastreid/modeling/meta_arch/moco.py +126 -0
- dnt/third_party/fast-reid/fastreid/solver/__init__.py +8 -0
- dnt/third_party/fast-reid/fastreid/solver/build.py +348 -0
- dnt/third_party/fast-reid/fastreid/solver/lr_scheduler.py +66 -0
- dnt/third_party/fast-reid/fastreid/solver/optim/__init__.py +10 -0
- dnt/third_party/fast-reid/fastreid/solver/optim/lamb.py +123 -0
- dnt/third_party/fast-reid/fastreid/solver/optim/radam.py +149 -0
- dnt/third_party/fast-reid/fastreid/solver/optim/swa.py +246 -0
- dnt/third_party/fast-reid/fastreid/utils/__init__.py +6 -0
- dnt/third_party/fast-reid/fastreid/utils/checkpoint.py +503 -0
- dnt/third_party/fast-reid/fastreid/utils/collect_env.py +158 -0
- dnt/third_party/fast-reid/fastreid/utils/comm.py +255 -0
- dnt/third_party/fast-reid/fastreid/utils/compute_dist.py +200 -0
- dnt/third_party/fast-reid/fastreid/utils/env.py +119 -0
- dnt/third_party/fast-reid/fastreid/utils/events.py +461 -0
- dnt/third_party/fast-reid/fastreid/utils/faiss_utils.py +127 -0
- dnt/third_party/fast-reid/fastreid/utils/file_io.py +520 -0
- dnt/third_party/fast-reid/fastreid/utils/history_buffer.py +71 -0
- dnt/third_party/fast-reid/fastreid/utils/logger.py +211 -0
- dnt/third_party/fast-reid/fastreid/utils/params.py +103 -0
- dnt/third_party/fast-reid/fastreid/utils/precision_bn.py +94 -0
- dnt/third_party/fast-reid/fastreid/utils/registry.py +66 -0
- dnt/third_party/fast-reid/fastreid/utils/summary.py +120 -0
- dnt/third_party/fast-reid/fastreid/utils/timer.py +68 -0
- dnt/third_party/fast-reid/fastreid/utils/visualizer.py +278 -0
- dnt/track/__init__.py +2 -0
- dnt/track/botsort/__init__.py +4 -0
- dnt/track/botsort/bot_tracker/__init__.py +3 -0
- dnt/track/botsort/bot_tracker/basetrack.py +60 -0
- dnt/track/botsort/bot_tracker/bot_sort.py +473 -0
- dnt/track/botsort/bot_tracker/gmc.py +316 -0
- dnt/track/botsort/bot_tracker/kalman_filter.py +269 -0
- dnt/track/botsort/bot_tracker/matching.py +194 -0
- dnt/track/botsort/bot_tracker/mc_bot_sort.py +505 -0
- dnt/track/{dsort/utils → botsort/bot_tracker/tracking_utils}/evaluation.py +14 -4
- dnt/track/{dsort/utils → botsort/bot_tracker/tracking_utils}/io.py +19 -36
- dnt/track/botsort/bot_tracker/tracking_utils/timer.py +37 -0
- dnt/track/botsort/inference.py +96 -0
- dnt/track/config.py +120 -0
- dnt/track/dsort/configs/bagtricks_R50.yml +7 -0
- dnt/track/dsort/configs/deep_sort.yaml +0 -0
- dnt/track/dsort/configs/fastreid.yaml +1 -1
- dnt/track/dsort/deep_sort/deep/checkpoint/ckpt.t7 +0 -0
- dnt/track/dsort/deep_sort/deep/feature_extractor.py +87 -8
- dnt/track/dsort/deep_sort/deep_sort.py +31 -20
- dnt/track/dsort/deep_sort/sort/detection.py +2 -1
- dnt/track/dsort/deep_sort/sort/iou_matching.py +0 -2
- dnt/track/dsort/deep_sort/sort/linear_assignment.py +0 -3
- dnt/track/dsort/deep_sort/sort/nn_matching.py +5 -5
- dnt/track/dsort/deep_sort/sort/preprocessing.py +1 -2
- dnt/track/dsort/deep_sort/sort/track.py +2 -1
- dnt/track/dsort/deep_sort/sort/tracker.py +1 -1
- dnt/track/dsort/dsort.py +43 -33
- dnt/track/re_class.py +117 -0
- dnt/track/sort/sort.py +9 -6
- dnt/track/tracker.py +213 -32
- dnt-0.3.1.8.dist-info/METADATA +117 -0
- dnt-0.3.1.8.dist-info/RECORD +315 -0
- {dnt-0.2.4.dist-info → dnt-0.3.1.8.dist-info}/WHEEL +1 -1
- dnt/analysis/yield.py +0 -9
- dnt/track/dsort/deep_sort/deep/evaluate.py +0 -15
- dnt/track/dsort/deep_sort/deep/original_model.py +0 -106
- dnt/track/dsort/deep_sort/deep/test.py +0 -77
- dnt/track/dsort/deep_sort/deep/train.py +0 -189
- dnt/track/dsort/utils/asserts.py +0 -13
- dnt/track/dsort/utils/draw.py +0 -36
- dnt/track/dsort/utils/json_logger.py +0 -383
- dnt/track/dsort/utils/log.py +0 -17
- dnt/track/dsort/utils/parser.py +0 -35
- dnt/track/dsort/utils/tools.py +0 -39
- dnt-0.2.4.dist-info/METADATA +0 -35
- dnt-0.2.4.dist-info/RECORD +0 -64
- /dnt/{track/dsort/utils → third_party/fast-reid/checkpoint}/__init__.py +0 -0
- {dnt-0.2.4.dist-info → dnt-0.3.1.8.dist-info/licenses}/LICENSE +0 -0
- {dnt-0.2.4.dist-info → dnt-0.3.1.8.dist-info}/top_level.txt +0 -0
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-3.2GF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
DEPTH: 25
|
|
6
|
+
W0: 88
|
|
7
|
+
WA: 26.31
|
|
8
|
+
WM: 2.25
|
|
9
|
+
GROUP_W: 48
|
|
10
|
+
OPTIM:
|
|
11
|
+
LR_POLICY: cos
|
|
12
|
+
BASE_LR: 0.4
|
|
13
|
+
MAX_EPOCH: 100
|
|
14
|
+
MOMENTUM: 0.9
|
|
15
|
+
WEIGHT_DECAY: 5e-5
|
|
16
|
+
WARMUP_ITERS: 5
|
|
17
|
+
TRAIN:
|
|
18
|
+
DATASET: imagenet
|
|
19
|
+
IM_SIZE: 224
|
|
20
|
+
BATCH_SIZE: 512
|
|
21
|
+
TEST:
|
|
22
|
+
DATASET: imagenet
|
|
23
|
+
IM_SIZE: 256
|
|
24
|
+
BATCH_SIZE: 400
|
|
25
|
+
NUM_GPUS: 8
|
|
26
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-32GF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
DEPTH: 23
|
|
6
|
+
W0: 320
|
|
7
|
+
WA: 69.86
|
|
8
|
+
WM: 2.0
|
|
9
|
+
GROUP_W: 168
|
|
10
|
+
OPTIM:
|
|
11
|
+
LR_POLICY: cos
|
|
12
|
+
BASE_LR: 0.2
|
|
13
|
+
MAX_EPOCH: 100
|
|
14
|
+
MOMENTUM: 0.9
|
|
15
|
+
WEIGHT_DECAY: 5e-5
|
|
16
|
+
WARMUP_ITERS: 5
|
|
17
|
+
TRAIN:
|
|
18
|
+
DATASET: imagenet
|
|
19
|
+
IM_SIZE: 224
|
|
20
|
+
BATCH_SIZE: 256
|
|
21
|
+
TEST:
|
|
22
|
+
DATASET: imagenet
|
|
23
|
+
IM_SIZE: 256
|
|
24
|
+
BATCH_SIZE: 200
|
|
25
|
+
NUM_GPUS: 8
|
|
26
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-4.0GF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
DEPTH: 23
|
|
6
|
+
W0: 96
|
|
7
|
+
WA: 38.65
|
|
8
|
+
WM: 2.43
|
|
9
|
+
GROUP_W: 40
|
|
10
|
+
OPTIM:
|
|
11
|
+
LR_POLICY: cos
|
|
12
|
+
BASE_LR: 0.4
|
|
13
|
+
MAX_EPOCH: 100
|
|
14
|
+
MOMENTUM: 0.9
|
|
15
|
+
WEIGHT_DECAY: 5e-5
|
|
16
|
+
WARMUP_ITERS: 5
|
|
17
|
+
TRAIN:
|
|
18
|
+
DATASET: imagenet
|
|
19
|
+
IM_SIZE: 224
|
|
20
|
+
BATCH_SIZE: 512
|
|
21
|
+
TEST:
|
|
22
|
+
DATASET: imagenet
|
|
23
|
+
IM_SIZE: 256
|
|
24
|
+
BATCH_SIZE: 400
|
|
25
|
+
NUM_GPUS: 8
|
|
26
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-400MF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
DEPTH: 22
|
|
6
|
+
W0: 24
|
|
7
|
+
WA: 24.48
|
|
8
|
+
WM: 2.54
|
|
9
|
+
GROUP_W: 16
|
|
10
|
+
OPTIM:
|
|
11
|
+
LR_POLICY: cos
|
|
12
|
+
BASE_LR: 0.8
|
|
13
|
+
MAX_EPOCH: 100
|
|
14
|
+
MOMENTUM: 0.9
|
|
15
|
+
WEIGHT_DECAY: 5e-5
|
|
16
|
+
WARMUP_ITERS: 5
|
|
17
|
+
TRAIN:
|
|
18
|
+
DATASET: imagenet
|
|
19
|
+
IM_SIZE: 224
|
|
20
|
+
BATCH_SIZE: 1024
|
|
21
|
+
TEST:
|
|
22
|
+
DATASET: imagenet
|
|
23
|
+
IM_SIZE: 256
|
|
24
|
+
BATCH_SIZE: 800
|
|
25
|
+
NUM_GPUS: 8
|
|
26
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-6.4GF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
DEPTH: 17
|
|
6
|
+
W0: 184
|
|
7
|
+
WA: 60.83
|
|
8
|
+
WM: 2.07
|
|
9
|
+
GROUP_W: 56
|
|
10
|
+
OPTIM:
|
|
11
|
+
LR_POLICY: cos
|
|
12
|
+
BASE_LR: 0.4
|
|
13
|
+
MAX_EPOCH: 100
|
|
14
|
+
MOMENTUM: 0.9
|
|
15
|
+
WEIGHT_DECAY: 5e-5
|
|
16
|
+
WARMUP_ITERS: 5
|
|
17
|
+
TRAIN:
|
|
18
|
+
DATASET: imagenet
|
|
19
|
+
IM_SIZE: 224
|
|
20
|
+
BATCH_SIZE: 512
|
|
21
|
+
TEST:
|
|
22
|
+
DATASET: imagenet
|
|
23
|
+
IM_SIZE: 256
|
|
24
|
+
BATCH_SIZE: 400
|
|
25
|
+
NUM_GPUS: 8
|
|
26
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-600MF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
DEPTH: 16
|
|
6
|
+
W0: 48
|
|
7
|
+
WA: 36.97
|
|
8
|
+
WM: 2.24
|
|
9
|
+
GROUP_W: 24
|
|
10
|
+
OPTIM:
|
|
11
|
+
LR_POLICY: cos
|
|
12
|
+
BASE_LR: 0.8
|
|
13
|
+
MAX_EPOCH: 100
|
|
14
|
+
MOMENTUM: 0.9
|
|
15
|
+
WEIGHT_DECAY: 5e-5
|
|
16
|
+
WARMUP_ITERS: 5
|
|
17
|
+
TRAIN:
|
|
18
|
+
DATASET: imagenet
|
|
19
|
+
IM_SIZE: 224
|
|
20
|
+
BATCH_SIZE: 1024
|
|
21
|
+
TEST:
|
|
22
|
+
DATASET: imagenet
|
|
23
|
+
IM_SIZE: 256
|
|
24
|
+
BATCH_SIZE: 800
|
|
25
|
+
NUM_GPUS: 8
|
|
26
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-8.0GF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
DEPTH: 23
|
|
6
|
+
W0: 80
|
|
7
|
+
WA: 49.56
|
|
8
|
+
WM: 2.88
|
|
9
|
+
GROUP_W: 120
|
|
10
|
+
OPTIM:
|
|
11
|
+
LR_POLICY: cos
|
|
12
|
+
BASE_LR: 0.4
|
|
13
|
+
MAX_EPOCH: 100
|
|
14
|
+
MOMENTUM: 0.9
|
|
15
|
+
WEIGHT_DECAY: 5e-5
|
|
16
|
+
WARMUP_ITERS: 5
|
|
17
|
+
TRAIN:
|
|
18
|
+
DATASET: imagenet
|
|
19
|
+
IM_SIZE: 224
|
|
20
|
+
BATCH_SIZE: 512
|
|
21
|
+
TEST:
|
|
22
|
+
DATASET: imagenet
|
|
23
|
+
IM_SIZE: 256
|
|
24
|
+
BATCH_SIZE: 400
|
|
25
|
+
NUM_GPUS: 8
|
|
26
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-800MF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
DEPTH: 16
|
|
6
|
+
W0: 56
|
|
7
|
+
WA: 35.73
|
|
8
|
+
WM: 2.28
|
|
9
|
+
GROUP_W: 16
|
|
10
|
+
OPTIM:
|
|
11
|
+
LR_POLICY: cos
|
|
12
|
+
BASE_LR: 0.8
|
|
13
|
+
MAX_EPOCH: 100
|
|
14
|
+
MOMENTUM: 0.9
|
|
15
|
+
WEIGHT_DECAY: 5e-5
|
|
16
|
+
WARMUP_ITERS: 5
|
|
17
|
+
TRAIN:
|
|
18
|
+
DATASET: imagenet
|
|
19
|
+
IM_SIZE: 224
|
|
20
|
+
BATCH_SIZE: 1024
|
|
21
|
+
TEST:
|
|
22
|
+
DATASET: imagenet
|
|
23
|
+
IM_SIZE: 256
|
|
24
|
+
BATCH_SIZE: 800
|
|
25
|
+
NUM_GPUS: 8
|
|
26
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-1.6GF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
SE_ON: True
|
|
6
|
+
DEPTH: 27
|
|
7
|
+
W0: 48
|
|
8
|
+
WA: 20.71
|
|
9
|
+
WM: 2.65
|
|
10
|
+
GROUP_W: 24
|
|
11
|
+
OPTIM:
|
|
12
|
+
LR_POLICY: cos
|
|
13
|
+
BASE_LR: 0.8
|
|
14
|
+
MAX_EPOCH: 100
|
|
15
|
+
MOMENTUM: 0.9
|
|
16
|
+
WEIGHT_DECAY: 5e-5
|
|
17
|
+
WARMUP_ITERS: 5
|
|
18
|
+
TRAIN:
|
|
19
|
+
DATASET: imagenet
|
|
20
|
+
IM_SIZE: 224
|
|
21
|
+
BATCH_SIZE: 1024
|
|
22
|
+
TEST:
|
|
23
|
+
DATASET: imagenet
|
|
24
|
+
IM_SIZE: 256
|
|
25
|
+
BATCH_SIZE: 800
|
|
26
|
+
NUM_GPUS: 8
|
|
27
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-12GF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
SE_ON: True
|
|
6
|
+
DEPTH: 19
|
|
7
|
+
W0: 168
|
|
8
|
+
WA: 73.36
|
|
9
|
+
WM: 2.37
|
|
10
|
+
GROUP_W: 112
|
|
11
|
+
OPTIM:
|
|
12
|
+
LR_POLICY: cos
|
|
13
|
+
BASE_LR: 0.4
|
|
14
|
+
MAX_EPOCH: 100
|
|
15
|
+
MOMENTUM: 0.9
|
|
16
|
+
WEIGHT_DECAY: 5e-5
|
|
17
|
+
WARMUP_ITERS: 5
|
|
18
|
+
TRAIN:
|
|
19
|
+
DATASET: imagenet
|
|
20
|
+
IM_SIZE: 224
|
|
21
|
+
BATCH_SIZE: 512
|
|
22
|
+
TEST:
|
|
23
|
+
DATASET: imagenet
|
|
24
|
+
IM_SIZE: 256
|
|
25
|
+
BATCH_SIZE: 400
|
|
26
|
+
NUM_GPUS: 8
|
|
27
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-16GF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
SE_ON: True
|
|
6
|
+
DEPTH: 18
|
|
7
|
+
W0: 200
|
|
8
|
+
WA: 106.23
|
|
9
|
+
WM: 2.48
|
|
10
|
+
GROUP_W: 112
|
|
11
|
+
OPTIM:
|
|
12
|
+
LR_POLICY: cos
|
|
13
|
+
BASE_LR: 0.2
|
|
14
|
+
MAX_EPOCH: 100
|
|
15
|
+
MOMENTUM: 0.9
|
|
16
|
+
WEIGHT_DECAY: 5e-5
|
|
17
|
+
WARMUP_ITERS: 5
|
|
18
|
+
TRAIN:
|
|
19
|
+
DATASET: imagenet
|
|
20
|
+
IM_SIZE: 224
|
|
21
|
+
BATCH_SIZE: 256
|
|
22
|
+
TEST:
|
|
23
|
+
DATASET: imagenet
|
|
24
|
+
IM_SIZE: 256
|
|
25
|
+
BATCH_SIZE: 200
|
|
26
|
+
NUM_GPUS: 8
|
|
27
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-200MF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
SE_ON: True
|
|
6
|
+
DEPTH: 13
|
|
7
|
+
W0: 24
|
|
8
|
+
WA: 36.44
|
|
9
|
+
WM: 2.49
|
|
10
|
+
GROUP_W: 8
|
|
11
|
+
OPTIM:
|
|
12
|
+
LR_POLICY: cos
|
|
13
|
+
BASE_LR: 0.8
|
|
14
|
+
MAX_EPOCH: 100
|
|
15
|
+
MOMENTUM: 0.9
|
|
16
|
+
WEIGHT_DECAY: 5e-5
|
|
17
|
+
TRAIN:
|
|
18
|
+
DATASET: imagenet
|
|
19
|
+
IM_SIZE: 224
|
|
20
|
+
BATCH_SIZE: 1024
|
|
21
|
+
TEST:
|
|
22
|
+
DATASET: imagenet
|
|
23
|
+
IM_SIZE: 256
|
|
24
|
+
BATCH_SIZE: 800
|
|
25
|
+
NUM_GPUS: 8
|
|
26
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-3.2GF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
SE_ON: True
|
|
6
|
+
DEPTH: 21
|
|
7
|
+
W0: 80
|
|
8
|
+
WA: 42.63
|
|
9
|
+
WM: 2.66
|
|
10
|
+
GROUP_W: 24
|
|
11
|
+
OPTIM:
|
|
12
|
+
LR_POLICY: cos
|
|
13
|
+
BASE_LR: 0.4
|
|
14
|
+
MAX_EPOCH: 100
|
|
15
|
+
MOMENTUM: 0.9
|
|
16
|
+
WEIGHT_DECAY: 5e-5
|
|
17
|
+
WARMUP_ITERS: 5
|
|
18
|
+
TRAIN:
|
|
19
|
+
DATASET: imagenet
|
|
20
|
+
IM_SIZE: 224
|
|
21
|
+
BATCH_SIZE: 512
|
|
22
|
+
TEST:
|
|
23
|
+
DATASET: imagenet
|
|
24
|
+
IM_SIZE: 256
|
|
25
|
+
BATCH_SIZE: 400
|
|
26
|
+
NUM_GPUS: 8
|
|
27
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-32GF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
SE_ON: True
|
|
6
|
+
DEPTH: 20
|
|
7
|
+
W0: 232
|
|
8
|
+
WA: 115.89
|
|
9
|
+
WM: 2.53
|
|
10
|
+
GROUP_W: 232
|
|
11
|
+
OPTIM:
|
|
12
|
+
LR_POLICY: cos
|
|
13
|
+
BASE_LR: 0.2
|
|
14
|
+
MAX_EPOCH: 100
|
|
15
|
+
MOMENTUM: 0.9
|
|
16
|
+
WEIGHT_DECAY: 5e-5
|
|
17
|
+
WARMUP_ITERS: 5
|
|
18
|
+
TRAIN:
|
|
19
|
+
DATASET: imagenet
|
|
20
|
+
IM_SIZE: 224
|
|
21
|
+
BATCH_SIZE: 256
|
|
22
|
+
TEST:
|
|
23
|
+
DATASET: imagenet
|
|
24
|
+
IM_SIZE: 256
|
|
25
|
+
BATCH_SIZE: 200
|
|
26
|
+
NUM_GPUS: 8
|
|
27
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-4.0GF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
SE_ON: True
|
|
6
|
+
DEPTH: 22
|
|
7
|
+
W0: 96
|
|
8
|
+
WA: 31.41
|
|
9
|
+
WM: 2.24
|
|
10
|
+
GROUP_W: 64
|
|
11
|
+
OPTIM:
|
|
12
|
+
LR_POLICY: cos
|
|
13
|
+
BASE_LR: 0.4
|
|
14
|
+
MAX_EPOCH: 100
|
|
15
|
+
MOMENTUM: 0.9
|
|
16
|
+
WEIGHT_DECAY: 5e-5
|
|
17
|
+
WARMUP_ITERS: 5
|
|
18
|
+
TRAIN:
|
|
19
|
+
DATASET: imagenet
|
|
20
|
+
IM_SIZE: 224
|
|
21
|
+
BATCH_SIZE: 512
|
|
22
|
+
TEST:
|
|
23
|
+
DATASET: imagenet
|
|
24
|
+
IM_SIZE: 256
|
|
25
|
+
BATCH_SIZE: 400
|
|
26
|
+
NUM_GPUS: 8
|
|
27
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-400MF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
SE_ON: True
|
|
6
|
+
DEPTH: 16
|
|
7
|
+
W0: 48
|
|
8
|
+
WA: 27.89
|
|
9
|
+
WM: 2.09
|
|
10
|
+
GROUP_W: 8
|
|
11
|
+
OPTIM:
|
|
12
|
+
LR_POLICY: cos
|
|
13
|
+
BASE_LR: 0.8
|
|
14
|
+
MAX_EPOCH: 100
|
|
15
|
+
MOMENTUM: 0.9
|
|
16
|
+
WEIGHT_DECAY: 5e-5
|
|
17
|
+
WARMUP_ITERS: 5
|
|
18
|
+
TRAIN:
|
|
19
|
+
DATASET: imagenet
|
|
20
|
+
IM_SIZE: 224
|
|
21
|
+
BATCH_SIZE: 1024
|
|
22
|
+
TEST:
|
|
23
|
+
DATASET: imagenet
|
|
24
|
+
IM_SIZE: 256
|
|
25
|
+
BATCH_SIZE: 800
|
|
26
|
+
NUM_GPUS: 8
|
|
27
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-6.4GF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
SE_ON: True
|
|
6
|
+
DEPTH: 25
|
|
7
|
+
W0: 112
|
|
8
|
+
WA: 33.22
|
|
9
|
+
WM: 2.27
|
|
10
|
+
GROUP_W: 72
|
|
11
|
+
OPTIM:
|
|
12
|
+
LR_POLICY: cos
|
|
13
|
+
BASE_LR: 0.4
|
|
14
|
+
MAX_EPOCH: 100
|
|
15
|
+
MOMENTUM: 0.9
|
|
16
|
+
WEIGHT_DECAY: 5e-5
|
|
17
|
+
WARMUP_ITERS: 5
|
|
18
|
+
TRAIN:
|
|
19
|
+
DATASET: imagenet
|
|
20
|
+
IM_SIZE: 224
|
|
21
|
+
BATCH_SIZE: 512
|
|
22
|
+
TEST:
|
|
23
|
+
DATASET: imagenet
|
|
24
|
+
IM_SIZE: 256
|
|
25
|
+
BATCH_SIZE: 400
|
|
26
|
+
NUM_GPUS: 8
|
|
27
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-600MF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
SE_ON: True
|
|
6
|
+
DEPTH: 15
|
|
7
|
+
W0: 48
|
|
8
|
+
WA: 32.54
|
|
9
|
+
WM: 2.32
|
|
10
|
+
GROUP_W: 16
|
|
11
|
+
OPTIM:
|
|
12
|
+
LR_POLICY: cos
|
|
13
|
+
BASE_LR: 0.8
|
|
14
|
+
MAX_EPOCH: 100
|
|
15
|
+
MOMENTUM: 0.9
|
|
16
|
+
WEIGHT_DECAY: 5e-5
|
|
17
|
+
WARMUP_ITERS: 5
|
|
18
|
+
TRAIN:
|
|
19
|
+
DATASET: imagenet
|
|
20
|
+
IM_SIZE: 224
|
|
21
|
+
BATCH_SIZE: 1024
|
|
22
|
+
TEST:
|
|
23
|
+
DATASET: imagenet
|
|
24
|
+
IM_SIZE: 256
|
|
25
|
+
BATCH_SIZE: 800
|
|
26
|
+
NUM_GPUS: 8
|
|
27
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-8.0GF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
SE_ON: true
|
|
6
|
+
DEPTH: 17
|
|
7
|
+
W0: 192
|
|
8
|
+
WA: 76.82
|
|
9
|
+
WM: 2.19
|
|
10
|
+
GROUP_W: 56
|
|
11
|
+
OPTIM:
|
|
12
|
+
LR_POLICY: cos
|
|
13
|
+
BASE_LR: 0.4
|
|
14
|
+
MAX_EPOCH: 100
|
|
15
|
+
MOMENTUM: 0.9
|
|
16
|
+
WEIGHT_DECAY: 5e-5
|
|
17
|
+
WARMUP_ITERS: 5
|
|
18
|
+
TRAIN:
|
|
19
|
+
DATASET: imagenet
|
|
20
|
+
IM_SIZE: 224
|
|
21
|
+
BATCH_SIZE: 512
|
|
22
|
+
TEST:
|
|
23
|
+
DATASET: imagenet
|
|
24
|
+
IM_SIZE: 256
|
|
25
|
+
BATCH_SIZE: 400
|
|
26
|
+
NUM_GPUS: 8
|
|
27
|
+
OUT_DIR: .
|
dnt/third_party/fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-800MF_dds_8gpu.yaml
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
MODEL:
|
|
2
|
+
TYPE: regnet
|
|
3
|
+
NUM_CLASSES: 1000
|
|
4
|
+
REGNET:
|
|
5
|
+
SE_ON: True
|
|
6
|
+
DEPTH: 14
|
|
7
|
+
W0: 56
|
|
8
|
+
WA: 38.84
|
|
9
|
+
WM: 2.4
|
|
10
|
+
GROUP_W: 16
|
|
11
|
+
OPTIM:
|
|
12
|
+
LR_POLICY: cos
|
|
13
|
+
BASE_LR: 0.8
|
|
14
|
+
MAX_EPOCH: 100
|
|
15
|
+
MOMENTUM: 0.9
|
|
16
|
+
WEIGHT_DECAY: 5e-5
|
|
17
|
+
WARMUP_ITERS: 5
|
|
18
|
+
TRAIN:
|
|
19
|
+
DATASET: imagenet
|
|
20
|
+
IM_SIZE: 224
|
|
21
|
+
BATCH_SIZE: 1024
|
|
22
|
+
TEST:
|
|
23
|
+
DATASET: imagenet
|
|
24
|
+
IM_SIZE: 256
|
|
25
|
+
BATCH_SIZE: 800
|
|
26
|
+
NUM_GPUS: 8
|
|
27
|
+
OUT_DIR: .
|