├── .gitignore ├── .idea ├── .gitignore ├── YOLOv6_pro.iml ├── inspectionProfiles │ └── profiles_settings.xml ├── misc.xml ├── modules.xml └── vcs.xml ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── configs ├── BotNet │ ├── yolov6l_BepbotC3.yaml │ ├── yolov6l_BepbotC3_yaml.py │ ├── yolov6l_BoT3.yaml │ └── yolov6l_BotNet.yaml ├── CoAtNet │ ├── coat.py │ ├── coat.yaml │ ├── coat_block.py │ ├── coat_block.yaml │ ├── yolov6l_MBConvC3.py │ └── yolov6l_MBConvC3.yaml ├── DAMO-YOLO │ ├── yaml │ │ ├── yolov6l_RepGFPN_M.yaml │ │ ├── yolov6l_RepGFPN_S.yaml │ │ ├── yolov6s_RepGFPN_T.yaml │ │ ├── yolov6t_RepGFPN_T.yaml │ │ └── yolov6t_RepGFPN_T_SepBNHead.yaml │ ├── yolov6l_RepGFPN_M_yaml.py │ ├── yolov6l_RepGFPN_S_yaml.py │ ├── yolov6s_RepGFPN_T_yaml.py │ ├── yolov6t_RepGFPN_T_SepBNHead_yaml.py │ └── yolov6t_RepGFPN_T_yaml.py ├── FocalTransformer │ ├── yolov6l_Focal.yaml │ ├── yolov6l_FocalC3.yaml │ └── yolov6l_focal_yaml.py ├── RTMDet_net │ ├── RTMDet_l.py │ └── RTMDet_l.yaml ├── RepGhost │ ├── RepGhostNet1.0×.py │ ├── RepGhostNet1.0×.yaml │ ├── yolov6t_repghost.py │ ├── yolov6t_repghost.yaml │ ├── yolov6t_repghostC3.py │ └── yolov6t_repghostC3.yaml ├── SimOTA │ ├── yolov6l.yaml │ ├── yolov6l_yaml.py │ ├── yolov6s.yaml │ └── yolov6s_yaml.py ├── experiment │ ├── eval_640_repro.py │ ├── yolov6n_with_eval_params.py │ ├── yolov6s_csp_scaled.py │ └── yolov6t_csp_scaled.py ├── model_yaml │ ├── yolov5l_yaml.py │ ├── yolov6l6_p2_yaml.py │ ├── yolov6l6_p7_yaml.py │ ├── yolov6l6_yaml.py │ ├── yolov6l_yaml.py │ ├── yolov6m_yaml.py │ ├── yolov6n6_yaml.py │ ├── yolov6n_yaml.py │ ├── yolov6s_yaml.py │ ├── yolov6t_GiraffeNeckV2_yaml.py │ └── yolov6t_yaml.py ├── office │ ├── yolov6l.py │ ├── yolov6l_finetune.py │ ├── yolov6l_relu.py │ ├── yolov6l_relu_finetune.py │ ├── yolov6m.py │ ├── yolov6m_finetune.py │ ├── yolov6n.py │ ├── yolov6n_finetune.py │ ├── yolov6s.py │ ├── yolov6s_finetune.py │ ├── yolov6t.py │ └── yolov6t_finetune.py ├── repopt │ ├── yolov6_tiny_hs.py │ ├── yolov6_tiny_opt.py │ ├── yolov6_tiny_opt_qat.py │ ├── yolov6n_hs.py │ ├── yolov6n_opt.py │ ├── yolov6n_opt_qat.py │ ├── yolov6s_hs.py │ ├── yolov6s_opt.py │ └── yolov6s_opt_qat.py ├── without_weights │ ├── yolov6l6_yaml.py │ ├── yolov6l_yaml.py │ ├── yolov6m_yaml.py │ ├── yolov6n6_yaml.py │ ├── yolov6n_yaml.py │ ├── yolov6s_yaml.py │ └── yolov6t_yaml.py ├── yaml │ ├── yolov5l.yaml │ ├── yolov6l.yaml │ ├── yolov6l6.yaml │ ├── yolov6l6_p2.yaml │ ├── yolov6l6_p7.yaml │ ├── yolov6m.yaml │ ├── yolov6n.yaml │ ├── yolov6n6.yaml │ ├── yolov6s.yaml │ ├── yolov6t.yaml │ └── yolov6t_GiraffeNeckV2.yaml └── yolov7_net │ ├── yolov7.yaml │ ├── yolov7_net.yaml │ └── yolov7_net_yaml.py ├── data ├── Visdrone.yaml ├── algae.yaml ├── coco.yaml └── data.yaml ├── docs ├── Test_speed.md ├── Train_custom_data.md ├── Tutorial of Quantization.md ├── tutorial_repopt.md └── tutorial_voc.ipynb ├── models ├── Detect │ └── MuitlHead.py ├── Models │ ├── Attention │ │ ├── CrissCrossAttention.py │ │ ├── GAMAttention.py │ │ ├── NAMAttention.py │ │ ├── S2Attention.py │ │ ├── SEAttention.py │ │ ├── SKAttention.py │ │ ├── SOCA.py │ │ ├── ShuffleAttention.py │ │ └── SimAM.py │ ├── FaceV2.py │ ├── Litemodel.py │ ├── SwinTransformer.py │ ├── densenet.py │ ├── mobileone.py │ ├── muitlbackbone.py │ ├── ppyolo.py │ ├── research.py │ ├── resnet.py │ ├── slimneck.py │ ├── yolor.py │ ├── yolov4.py │ └── yolovii.py ├── __init__.py ├── algae │ ├── yolov5l6_cbnetv2.yaml │ ├── yolov5l6_cbnetv2_four.yaml │ └── yolov6.py ├── cbnetv2.py ├── common.py ├── dyhead.py ├── experimental.py ├── modulated_deform_conv.py ├── module.py ├── swin_transformer.py ├── swin_transformer2.py ├── tf.py ├── yolo.py ├── yolov6.py └── yolox.py ├── requirements.txt ├── timm ├── __init__.py ├── data │ ├── __init__.py │ ├── auto_augment.py │ ├── config.py │ ├── constants.py │ ├── dataset.py │ ├── dataset_factory.py │ ├── distributed_sampler.py │ ├── loader.py │ ├── mixup.py │ ├── random_erasing.py │ ├── readers │ │ ├── __init__.py │ │ ├── class_map.py │ │ ├── img_extensions.py │ │ ├── reader.py │ │ ├── reader_factory.py │ │ ├── reader_hfds.py │ │ ├── reader_image_folder.py │ │ ├── reader_image_in_tar.py │ │ ├── reader_image_tar.py │ │ ├── reader_tfds.py │ │ ├── reader_wds.py │ │ └── shared_count.py │ ├── real_labels.py │ ├── tf_preprocessing.py │ ├── transforms.py │ └── transforms_factory.py ├── layers │ ├── __init__.py │ ├── activations.py │ ├── activations_jit.py │ ├── activations_me.py │ ├── adaptive_avgmax_pool.py │ ├── attention_pool2d.py │ ├── blur_pool.py │ ├── bottleneck_attn.py │ ├── cbam.py │ ├── classifier.py │ ├── cond_conv2d.py │ ├── config.py │ ├── conv2d_same.py │ ├── conv_bn_act.py │ ├── create_act.py │ ├── create_attn.py │ ├── create_conv2d.py │ ├── create_norm.py │ ├── create_norm_act.py │ ├── drop.py │ ├── eca.py │ ├── evo_norm.py │ ├── fast_norm.py │ ├── filter_response_norm.py │ ├── gather_excite.py │ ├── global_context.py │ ├── halo_attn.py │ ├── helpers.py │ ├── inplace_abn.py │ ├── lambda_layer.py │ ├── linear.py │ ├── median_pool.py │ ├── mixed_conv2d.py │ ├── ml_decoder.py │ ├── mlp.py │ ├── non_local_attn.py │ ├── norm.py │ ├── norm_act.py │ ├── padding.py │ ├── patch_embed.py │ ├── pool2d_same.py │ ├── pos_embed.py │ ├── selective_kernel.py │ ├── separable_conv.py │ ├── space_to_depth.py │ ├── split_attn.py │ ├── split_batchnorm.py │ ├── squeeze_excite.py │ ├── std_conv.py │ ├── test_time_pool.py │ ├── trace_utils.py │ └── weight_init.py ├── loss │ ├── __init__.py │ ├── asymmetric_loss.py │ ├── binary_cross_entropy.py │ ├── cross_entropy.py │ └── jsd.py ├── models │ ├── __init__.py │ ├── _builder.py │ ├── _efficientnet_blocks.py │ ├── _efficientnet_builder.py │ ├── _factory.py │ ├── _features.py │ ├── _features_fx.py │ ├── _helpers.py │ ├── _hub.py │ ├── _manipulate.py │ ├── _pretrained.py │ ├── _prune.py │ ├── _pruned │ │ ├── ecaresnet101d_pruned.txt │ │ ├── ecaresnet50d_pruned.txt │ │ ├── efficientnet_b1_pruned.txt │ │ ├── efficientnet_b2_pruned.txt │ │ └── efficientnet_b3_pruned.txt │ ├── _registry.py │ ├── beit.py │ ├── byoanet.py │ ├── byobnet.py │ ├── cait.py │ ├── coat.py │ ├── convit.py │ ├── convmixer.py │ ├── convnext.py │ ├── crossvit.py │ ├── cspnet.py │ ├── deit.py │ ├── densenet.py │ ├── dla.py │ ├── dpn.py │ ├── edgenext.py │ ├── efficientformer.py │ ├── efficientnet.py │ ├── factory.py │ ├── features.py │ ├── fx_features.py │ ├── gcvit.py │ ├── ghostnet.py │ ├── gluon_resnet.py │ ├── gluon_xception.py │ ├── hardcorenas.py │ ├── helpers.py │ ├── hrnet.py │ ├── hub.py │ ├── inception_resnet_v2.py │ ├── inception_v3.py │ ├── inception_v4.py │ ├── layers │ │ └── __init__.py │ ├── levit.py │ ├── maxxvit.py │ ├── mlp_mixer.py │ ├── mobilenetv3.py │ ├── mobilevit.py │ ├── mvitv2.py │ ├── nasnet.py │ ├── nest.py │ ├── nfnet.py │ ├── pit.py │ ├── pnasnet.py │ ├── poolformer.py │ ├── pvt_v2.py │ ├── registry.py │ ├── regnet.py │ ├── res2net.py │ ├── resnest.py │ ├── resnet.py │ ├── resnetv2.py │ ├── rexnet.py │ ├── selecsls.py │ ├── senet.py │ ├── sequencer.py │ ├── sknet.py │ ├── swin_transformer.py │ ├── swin_transformer_v2.py │ ├── swin_transformer_v2_cr.py │ ├── tnt.py │ ├── tresnet.py │ ├── twins.py │ ├── vgg.py │ ├── visformer.py │ ├── vision_transformer.py │ ├── vision_transformer_hybrid.py │ ├── vision_transformer_relpos.py │ ├── volo.py │ ├── vovnet.py │ ├── xception.py │ ├── xception_aligned.py │ └── xcit.py ├── optim │ ├── __init__.py │ ├── adabelief.py │ ├── adafactor.py │ ├── adahessian.py │ ├── adamp.py │ ├── adamw.py │ ├── adan.py │ ├── lamb.py │ ├── lars.py │ ├── lookahead.py │ ├── madgrad.py │ ├── nadam.py │ ├── nvnovograd.py │ ├── optim_factory.py │ ├── radam.py │ ├── rmsprop_tf.py │ └── sgdp.py ├── scheduler │ ├── __init__.py │ ├── cosine_lr.py │ ├── multistep_lr.py │ ├── plateau_lr.py │ ├── poly_lr.py │ ├── scheduler.py │ ├── scheduler_factory.py │ ├── step_lr.py │ └── tanh_lr.py ├── utils │ ├── __init__.py │ ├── agc.py │ ├── checkpoint_saver.py │ ├── clip_grad.py │ ├── cuda.py │ ├── decay_batch.py │ ├── distributed.py │ ├── jit.py │ ├── log.py │ ├── metrics.py │ ├── misc.py │ ├── model.py │ ├── model_ema.py │ ├── random.py │ └── summary.py └── version.py ├── tools ├── change_weights.py ├── eval.py ├── get_flops.py ├── infer.py ├── partial_quantization │ ├── README.md │ ├── eval.py │ ├── eval.yaml │ ├── partial_quant.py │ ├── ptq.py │ ├── sensitivity_analyse.py │ └── utils.py ├── qat │ ├── README.md │ ├── qat_export.py │ └── qat_utils.py ├── quantization │ ├── mnn │ │ └── README.md │ └── tensorrt │ │ ├── post_training │ │ ├── Calibrator.py │ │ ├── LICENSE │ │ ├── README.md │ │ └── onnx_to_tensorrt.py │ │ ├── requirements.txt │ │ └── training_aware │ │ └── QAT_quantizer.py └── train.py ├── utils ├── RepulsionLoss.py ├── __init__.py ├── activations.py ├── augmentations.py ├── autoanchor.py ├── autobatch.py ├── aws │ ├── __init__.py │ └── resume.py ├── benchmarks.py ├── callbacks.py ├── datasets.py ├── downloads.py ├── flask_rest_api │ ├── README.md │ ├── example_request.py │ └── restapi.py ├── general.py ├── google_app_engine │ ├── Dockerfile │ ├── additional_requirements.txt │ └── app.yaml ├── loggers │ ├── __init__.py │ └── wandb │ │ ├── README.md │ │ ├── __init__.py │ │ ├── log_dataset.py │ │ ├── sweep.py │ │ ├── sweep.yaml │ │ └── wandb_utils.py ├── loss.py ├── loss_ps.py ├── metrics.py ├── plots.py ├── torch_utils.py ├── yolo_with_plugins.py └── yolov6official.py └── yolov6 ├── __init__.py ├── assigners ├── __init__.py ├── anchor_generator.py ├── assigner_utils.py ├── atss_assigner.py ├── iou2d_calculator.py └── tal_assigner.py ├── core ├── engine.py ├── evaler.py └── inferer.py ├── data ├── data_augment.py ├── data_load.py ├── datasets.py ├── vis_dataset.py └── voc2yolo.py ├── layers ├── BotNet.py ├── CoAtNet.py ├── RTMDet.py ├── TAP.py ├── common.py ├── damo_yolo.py ├── dbb_transforms.py ├── focal_transformer.py ├── misc.py ├── tiny_nas_csp.py └── yolov7.py ├── models ├── efficientrep.py ├── effidehead.py ├── end2end.py ├── loss.py ├── loss_distill.py ├── reppan.py ├── simOTA.py └── yolo.py ├── solver └── build.py └── utils ├── Arial.ttf ├── RepOptimizer.py ├── checkpoint.py ├── config.py ├── ema.py ├── envs.py ├── events.py ├── figure_iou.py ├── general.py ├── metrics.py ├── nms.py └── torch_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/.gitignore -------------------------------------------------------------------------------- /.idea/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/.idea/.gitignore -------------------------------------------------------------------------------- /.idea/YOLOv6_pro.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/.idea/YOLOv6_pro.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/README.md -------------------------------------------------------------------------------- /configs/BotNet/yolov6l_BepbotC3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/BotNet/yolov6l_BepbotC3.yaml -------------------------------------------------------------------------------- /configs/BotNet/yolov6l_BepbotC3_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/BotNet/yolov6l_BepbotC3_yaml.py -------------------------------------------------------------------------------- /configs/BotNet/yolov6l_BoT3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/BotNet/yolov6l_BoT3.yaml -------------------------------------------------------------------------------- /configs/BotNet/yolov6l_BotNet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/BotNet/yolov6l_BotNet.yaml -------------------------------------------------------------------------------- /configs/CoAtNet/coat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/CoAtNet/coat.py -------------------------------------------------------------------------------- /configs/CoAtNet/coat.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/CoAtNet/coat.yaml -------------------------------------------------------------------------------- /configs/CoAtNet/coat_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/CoAtNet/coat_block.py -------------------------------------------------------------------------------- /configs/CoAtNet/coat_block.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/CoAtNet/coat_block.yaml -------------------------------------------------------------------------------- /configs/CoAtNet/yolov6l_MBConvC3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/CoAtNet/yolov6l_MBConvC3.py -------------------------------------------------------------------------------- /configs/CoAtNet/yolov6l_MBConvC3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/CoAtNet/yolov6l_MBConvC3.yaml -------------------------------------------------------------------------------- /configs/DAMO-YOLO/yaml/yolov6l_RepGFPN_M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/DAMO-YOLO/yaml/yolov6l_RepGFPN_M.yaml -------------------------------------------------------------------------------- /configs/DAMO-YOLO/yaml/yolov6l_RepGFPN_S.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/DAMO-YOLO/yaml/yolov6l_RepGFPN_S.yaml -------------------------------------------------------------------------------- /configs/DAMO-YOLO/yaml/yolov6s_RepGFPN_T.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/DAMO-YOLO/yaml/yolov6s_RepGFPN_T.yaml -------------------------------------------------------------------------------- /configs/DAMO-YOLO/yaml/yolov6t_RepGFPN_T.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/DAMO-YOLO/yaml/yolov6t_RepGFPN_T.yaml -------------------------------------------------------------------------------- /configs/DAMO-YOLO/yaml/yolov6t_RepGFPN_T_SepBNHead.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/DAMO-YOLO/yaml/yolov6t_RepGFPN_T_SepBNHead.yaml -------------------------------------------------------------------------------- /configs/DAMO-YOLO/yolov6l_RepGFPN_M_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/DAMO-YOLO/yolov6l_RepGFPN_M_yaml.py -------------------------------------------------------------------------------- /configs/DAMO-YOLO/yolov6l_RepGFPN_S_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/DAMO-YOLO/yolov6l_RepGFPN_S_yaml.py -------------------------------------------------------------------------------- /configs/DAMO-YOLO/yolov6s_RepGFPN_T_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/DAMO-YOLO/yolov6s_RepGFPN_T_yaml.py -------------------------------------------------------------------------------- /configs/DAMO-YOLO/yolov6t_RepGFPN_T_SepBNHead_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/DAMO-YOLO/yolov6t_RepGFPN_T_SepBNHead_yaml.py -------------------------------------------------------------------------------- /configs/DAMO-YOLO/yolov6t_RepGFPN_T_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/DAMO-YOLO/yolov6t_RepGFPN_T_yaml.py -------------------------------------------------------------------------------- /configs/FocalTransformer/yolov6l_Focal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/FocalTransformer/yolov6l_Focal.yaml -------------------------------------------------------------------------------- /configs/FocalTransformer/yolov6l_FocalC3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/FocalTransformer/yolov6l_FocalC3.yaml -------------------------------------------------------------------------------- /configs/FocalTransformer/yolov6l_focal_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/FocalTransformer/yolov6l_focal_yaml.py -------------------------------------------------------------------------------- /configs/RTMDet_net/RTMDet_l.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/RTMDet_net/RTMDet_l.py -------------------------------------------------------------------------------- /configs/RTMDet_net/RTMDet_l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/RTMDet_net/RTMDet_l.yaml -------------------------------------------------------------------------------- /configs/RepGhost/RepGhostNet1.0×.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/RepGhost/RepGhostNet1.0×.py -------------------------------------------------------------------------------- /configs/RepGhost/RepGhostNet1.0×.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/RepGhost/RepGhostNet1.0×.yaml -------------------------------------------------------------------------------- /configs/RepGhost/yolov6t_repghost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/RepGhost/yolov6t_repghost.py -------------------------------------------------------------------------------- /configs/RepGhost/yolov6t_repghost.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/RepGhost/yolov6t_repghost.yaml -------------------------------------------------------------------------------- /configs/RepGhost/yolov6t_repghostC3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/RepGhost/yolov6t_repghostC3.py -------------------------------------------------------------------------------- /configs/RepGhost/yolov6t_repghostC3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/RepGhost/yolov6t_repghostC3.yaml -------------------------------------------------------------------------------- /configs/SimOTA/yolov6l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/SimOTA/yolov6l.yaml -------------------------------------------------------------------------------- /configs/SimOTA/yolov6l_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/SimOTA/yolov6l_yaml.py -------------------------------------------------------------------------------- /configs/SimOTA/yolov6s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/SimOTA/yolov6s.yaml -------------------------------------------------------------------------------- /configs/SimOTA/yolov6s_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/SimOTA/yolov6s_yaml.py -------------------------------------------------------------------------------- /configs/experiment/eval_640_repro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/experiment/eval_640_repro.py -------------------------------------------------------------------------------- /configs/experiment/yolov6n_with_eval_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/experiment/yolov6n_with_eval_params.py -------------------------------------------------------------------------------- /configs/experiment/yolov6s_csp_scaled.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/experiment/yolov6s_csp_scaled.py -------------------------------------------------------------------------------- /configs/experiment/yolov6t_csp_scaled.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/experiment/yolov6t_csp_scaled.py -------------------------------------------------------------------------------- /configs/model_yaml/yolov5l_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/model_yaml/yolov5l_yaml.py -------------------------------------------------------------------------------- /configs/model_yaml/yolov6l6_p2_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/model_yaml/yolov6l6_p2_yaml.py -------------------------------------------------------------------------------- /configs/model_yaml/yolov6l6_p7_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/model_yaml/yolov6l6_p7_yaml.py -------------------------------------------------------------------------------- /configs/model_yaml/yolov6l6_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/model_yaml/yolov6l6_yaml.py -------------------------------------------------------------------------------- /configs/model_yaml/yolov6l_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/model_yaml/yolov6l_yaml.py -------------------------------------------------------------------------------- /configs/model_yaml/yolov6m_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/model_yaml/yolov6m_yaml.py -------------------------------------------------------------------------------- /configs/model_yaml/yolov6n6_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/model_yaml/yolov6n6_yaml.py -------------------------------------------------------------------------------- /configs/model_yaml/yolov6n_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/model_yaml/yolov6n_yaml.py -------------------------------------------------------------------------------- /configs/model_yaml/yolov6s_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/model_yaml/yolov6s_yaml.py -------------------------------------------------------------------------------- /configs/model_yaml/yolov6t_GiraffeNeckV2_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/model_yaml/yolov6t_GiraffeNeckV2_yaml.py -------------------------------------------------------------------------------- /configs/model_yaml/yolov6t_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/model_yaml/yolov6t_yaml.py -------------------------------------------------------------------------------- /configs/office/yolov6l.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/office/yolov6l.py -------------------------------------------------------------------------------- /configs/office/yolov6l_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/office/yolov6l_finetune.py -------------------------------------------------------------------------------- /configs/office/yolov6l_relu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/office/yolov6l_relu.py -------------------------------------------------------------------------------- /configs/office/yolov6l_relu_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/office/yolov6l_relu_finetune.py -------------------------------------------------------------------------------- /configs/office/yolov6m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/office/yolov6m.py -------------------------------------------------------------------------------- /configs/office/yolov6m_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/office/yolov6m_finetune.py -------------------------------------------------------------------------------- /configs/office/yolov6n.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/office/yolov6n.py -------------------------------------------------------------------------------- /configs/office/yolov6n_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/office/yolov6n_finetune.py -------------------------------------------------------------------------------- /configs/office/yolov6s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/office/yolov6s.py -------------------------------------------------------------------------------- /configs/office/yolov6s_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/office/yolov6s_finetune.py -------------------------------------------------------------------------------- /configs/office/yolov6t.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/office/yolov6t.py -------------------------------------------------------------------------------- /configs/office/yolov6t_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/office/yolov6t_finetune.py -------------------------------------------------------------------------------- /configs/repopt/yolov6_tiny_hs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/repopt/yolov6_tiny_hs.py -------------------------------------------------------------------------------- /configs/repopt/yolov6_tiny_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/repopt/yolov6_tiny_opt.py -------------------------------------------------------------------------------- /configs/repopt/yolov6_tiny_opt_qat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/repopt/yolov6_tiny_opt_qat.py -------------------------------------------------------------------------------- /configs/repopt/yolov6n_hs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/repopt/yolov6n_hs.py -------------------------------------------------------------------------------- /configs/repopt/yolov6n_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/repopt/yolov6n_opt.py -------------------------------------------------------------------------------- /configs/repopt/yolov6n_opt_qat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/repopt/yolov6n_opt_qat.py -------------------------------------------------------------------------------- /configs/repopt/yolov6s_hs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/repopt/yolov6s_hs.py -------------------------------------------------------------------------------- /configs/repopt/yolov6s_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/repopt/yolov6s_opt.py -------------------------------------------------------------------------------- /configs/repopt/yolov6s_opt_qat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/repopt/yolov6s_opt_qat.py -------------------------------------------------------------------------------- /configs/without_weights/yolov6l6_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/without_weights/yolov6l6_yaml.py -------------------------------------------------------------------------------- /configs/without_weights/yolov6l_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/without_weights/yolov6l_yaml.py -------------------------------------------------------------------------------- /configs/without_weights/yolov6m_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/without_weights/yolov6m_yaml.py -------------------------------------------------------------------------------- /configs/without_weights/yolov6n6_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/without_weights/yolov6n6_yaml.py -------------------------------------------------------------------------------- /configs/without_weights/yolov6n_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/without_weights/yolov6n_yaml.py -------------------------------------------------------------------------------- /configs/without_weights/yolov6s_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/without_weights/yolov6s_yaml.py -------------------------------------------------------------------------------- /configs/without_weights/yolov6t_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/without_weights/yolov6t_yaml.py -------------------------------------------------------------------------------- /configs/yaml/yolov5l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/yaml/yolov5l.yaml -------------------------------------------------------------------------------- /configs/yaml/yolov6l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/yaml/yolov6l.yaml -------------------------------------------------------------------------------- /configs/yaml/yolov6l6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/yaml/yolov6l6.yaml -------------------------------------------------------------------------------- /configs/yaml/yolov6l6_p2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/yaml/yolov6l6_p2.yaml -------------------------------------------------------------------------------- /configs/yaml/yolov6l6_p7.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/yaml/yolov6l6_p7.yaml -------------------------------------------------------------------------------- /configs/yaml/yolov6m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/yaml/yolov6m.yaml -------------------------------------------------------------------------------- /configs/yaml/yolov6n.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/yaml/yolov6n.yaml -------------------------------------------------------------------------------- /configs/yaml/yolov6n6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/yaml/yolov6n6.yaml -------------------------------------------------------------------------------- /configs/yaml/yolov6s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/yaml/yolov6s.yaml -------------------------------------------------------------------------------- /configs/yaml/yolov6t.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/yaml/yolov6t.yaml -------------------------------------------------------------------------------- /configs/yaml/yolov6t_GiraffeNeckV2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/yaml/yolov6t_GiraffeNeckV2.yaml -------------------------------------------------------------------------------- /configs/yolov7_net/yolov7.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/yolov7_net/yolov7.yaml -------------------------------------------------------------------------------- /configs/yolov7_net/yolov7_net.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/yolov7_net/yolov7_net.yaml -------------------------------------------------------------------------------- /configs/yolov7_net/yolov7_net_yaml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/configs/yolov7_net/yolov7_net_yaml.py -------------------------------------------------------------------------------- /data/Visdrone.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/data/Visdrone.yaml -------------------------------------------------------------------------------- /data/algae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/data/algae.yaml -------------------------------------------------------------------------------- /data/coco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/data/coco.yaml -------------------------------------------------------------------------------- /data/data.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/data/data.yaml -------------------------------------------------------------------------------- /docs/Test_speed.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/docs/Test_speed.md -------------------------------------------------------------------------------- /docs/Train_custom_data.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/docs/Train_custom_data.md -------------------------------------------------------------------------------- /docs/Tutorial of Quantization.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/docs/Tutorial of Quantization.md -------------------------------------------------------------------------------- /docs/tutorial_repopt.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/docs/tutorial_repopt.md -------------------------------------------------------------------------------- /docs/tutorial_voc.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/docs/tutorial_voc.ipynb -------------------------------------------------------------------------------- /models/Detect/MuitlHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Detect/MuitlHead.py -------------------------------------------------------------------------------- /models/Models/Attention/CrissCrossAttention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/Attention/CrissCrossAttention.py -------------------------------------------------------------------------------- /models/Models/Attention/GAMAttention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/Attention/GAMAttention.py -------------------------------------------------------------------------------- /models/Models/Attention/NAMAttention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/Attention/NAMAttention.py -------------------------------------------------------------------------------- /models/Models/Attention/S2Attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/Attention/S2Attention.py -------------------------------------------------------------------------------- /models/Models/Attention/SEAttention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/Attention/SEAttention.py -------------------------------------------------------------------------------- /models/Models/Attention/SKAttention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/Attention/SKAttention.py -------------------------------------------------------------------------------- /models/Models/Attention/SOCA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/Attention/SOCA.py -------------------------------------------------------------------------------- /models/Models/Attention/ShuffleAttention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/Attention/ShuffleAttention.py -------------------------------------------------------------------------------- /models/Models/Attention/SimAM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/Attention/SimAM.py -------------------------------------------------------------------------------- /models/Models/FaceV2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/FaceV2.py -------------------------------------------------------------------------------- /models/Models/Litemodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/Litemodel.py -------------------------------------------------------------------------------- /models/Models/SwinTransformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/SwinTransformer.py -------------------------------------------------------------------------------- /models/Models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/densenet.py -------------------------------------------------------------------------------- /models/Models/mobileone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/mobileone.py -------------------------------------------------------------------------------- /models/Models/muitlbackbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/muitlbackbone.py -------------------------------------------------------------------------------- /models/Models/ppyolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/ppyolo.py -------------------------------------------------------------------------------- /models/Models/research.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/research.py -------------------------------------------------------------------------------- /models/Models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/resnet.py -------------------------------------------------------------------------------- /models/Models/slimneck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/slimneck.py -------------------------------------------------------------------------------- /models/Models/yolor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/yolor.py -------------------------------------------------------------------------------- /models/Models/yolov4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/yolov4.py -------------------------------------------------------------------------------- /models/Models/yolovii.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/Models/yolovii.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/algae/yolov5l6_cbnetv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/algae/yolov5l6_cbnetv2.yaml -------------------------------------------------------------------------------- /models/algae/yolov5l6_cbnetv2_four.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/algae/yolov5l6_cbnetv2_four.yaml -------------------------------------------------------------------------------- /models/algae/yolov6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/algae/yolov6.py -------------------------------------------------------------------------------- /models/cbnetv2.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/common.py -------------------------------------------------------------------------------- /models/dyhead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/dyhead.py -------------------------------------------------------------------------------- /models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/experimental.py -------------------------------------------------------------------------------- /models/modulated_deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/modulated_deform_conv.py -------------------------------------------------------------------------------- /models/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/module.py -------------------------------------------------------------------------------- /models/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/swin_transformer.py -------------------------------------------------------------------------------- /models/swin_transformer2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/swin_transformer2.py -------------------------------------------------------------------------------- /models/tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/tf.py -------------------------------------------------------------------------------- /models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/yolo.py -------------------------------------------------------------------------------- /models/yolov6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/yolov6.py -------------------------------------------------------------------------------- /models/yolox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/models/yolox.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/requirements.txt -------------------------------------------------------------------------------- /timm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/__init__.py -------------------------------------------------------------------------------- /timm/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/__init__.py -------------------------------------------------------------------------------- /timm/data/auto_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/auto_augment.py -------------------------------------------------------------------------------- /timm/data/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/config.py -------------------------------------------------------------------------------- /timm/data/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/constants.py -------------------------------------------------------------------------------- /timm/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/dataset.py -------------------------------------------------------------------------------- /timm/data/dataset_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/dataset_factory.py -------------------------------------------------------------------------------- /timm/data/distributed_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/distributed_sampler.py -------------------------------------------------------------------------------- /timm/data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/loader.py -------------------------------------------------------------------------------- /timm/data/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/mixup.py -------------------------------------------------------------------------------- /timm/data/random_erasing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/random_erasing.py -------------------------------------------------------------------------------- /timm/data/readers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/readers/__init__.py -------------------------------------------------------------------------------- /timm/data/readers/class_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/readers/class_map.py -------------------------------------------------------------------------------- /timm/data/readers/img_extensions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/readers/img_extensions.py -------------------------------------------------------------------------------- /timm/data/readers/reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/readers/reader.py -------------------------------------------------------------------------------- /timm/data/readers/reader_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/readers/reader_factory.py -------------------------------------------------------------------------------- /timm/data/readers/reader_hfds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/readers/reader_hfds.py -------------------------------------------------------------------------------- /timm/data/readers/reader_image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/readers/reader_image_folder.py -------------------------------------------------------------------------------- /timm/data/readers/reader_image_in_tar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/readers/reader_image_in_tar.py -------------------------------------------------------------------------------- /timm/data/readers/reader_image_tar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/readers/reader_image_tar.py -------------------------------------------------------------------------------- /timm/data/readers/reader_tfds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/readers/reader_tfds.py -------------------------------------------------------------------------------- /timm/data/readers/reader_wds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/readers/reader_wds.py -------------------------------------------------------------------------------- /timm/data/readers/shared_count.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/readers/shared_count.py -------------------------------------------------------------------------------- /timm/data/real_labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/real_labels.py -------------------------------------------------------------------------------- /timm/data/tf_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/tf_preprocessing.py -------------------------------------------------------------------------------- /timm/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/transforms.py -------------------------------------------------------------------------------- /timm/data/transforms_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/data/transforms_factory.py -------------------------------------------------------------------------------- /timm/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/__init__.py -------------------------------------------------------------------------------- /timm/layers/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/activations.py -------------------------------------------------------------------------------- /timm/layers/activations_jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/activations_jit.py -------------------------------------------------------------------------------- /timm/layers/activations_me.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/activations_me.py -------------------------------------------------------------------------------- /timm/layers/adaptive_avgmax_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/adaptive_avgmax_pool.py -------------------------------------------------------------------------------- /timm/layers/attention_pool2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/attention_pool2d.py -------------------------------------------------------------------------------- /timm/layers/blur_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/blur_pool.py -------------------------------------------------------------------------------- /timm/layers/bottleneck_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/bottleneck_attn.py -------------------------------------------------------------------------------- /timm/layers/cbam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/cbam.py -------------------------------------------------------------------------------- /timm/layers/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/classifier.py -------------------------------------------------------------------------------- /timm/layers/cond_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/cond_conv2d.py -------------------------------------------------------------------------------- /timm/layers/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/config.py -------------------------------------------------------------------------------- /timm/layers/conv2d_same.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/conv2d_same.py -------------------------------------------------------------------------------- /timm/layers/conv_bn_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/conv_bn_act.py -------------------------------------------------------------------------------- /timm/layers/create_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/create_act.py -------------------------------------------------------------------------------- /timm/layers/create_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/create_attn.py -------------------------------------------------------------------------------- /timm/layers/create_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/create_conv2d.py -------------------------------------------------------------------------------- /timm/layers/create_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/create_norm.py -------------------------------------------------------------------------------- /timm/layers/create_norm_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/create_norm_act.py -------------------------------------------------------------------------------- /timm/layers/drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/drop.py -------------------------------------------------------------------------------- /timm/layers/eca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/eca.py -------------------------------------------------------------------------------- /timm/layers/evo_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/evo_norm.py -------------------------------------------------------------------------------- /timm/layers/fast_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/fast_norm.py -------------------------------------------------------------------------------- /timm/layers/filter_response_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/filter_response_norm.py -------------------------------------------------------------------------------- /timm/layers/gather_excite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/gather_excite.py -------------------------------------------------------------------------------- /timm/layers/global_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/global_context.py -------------------------------------------------------------------------------- /timm/layers/halo_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/halo_attn.py -------------------------------------------------------------------------------- /timm/layers/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/helpers.py -------------------------------------------------------------------------------- /timm/layers/inplace_abn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/inplace_abn.py -------------------------------------------------------------------------------- /timm/layers/lambda_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/lambda_layer.py -------------------------------------------------------------------------------- /timm/layers/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/linear.py -------------------------------------------------------------------------------- /timm/layers/median_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/median_pool.py -------------------------------------------------------------------------------- /timm/layers/mixed_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/mixed_conv2d.py -------------------------------------------------------------------------------- /timm/layers/ml_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/ml_decoder.py -------------------------------------------------------------------------------- /timm/layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/mlp.py -------------------------------------------------------------------------------- /timm/layers/non_local_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/non_local_attn.py -------------------------------------------------------------------------------- /timm/layers/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/norm.py -------------------------------------------------------------------------------- /timm/layers/norm_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/norm_act.py -------------------------------------------------------------------------------- /timm/layers/padding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/padding.py -------------------------------------------------------------------------------- /timm/layers/patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/patch_embed.py -------------------------------------------------------------------------------- /timm/layers/pool2d_same.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/pool2d_same.py -------------------------------------------------------------------------------- /timm/layers/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/pos_embed.py -------------------------------------------------------------------------------- /timm/layers/selective_kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/selective_kernel.py -------------------------------------------------------------------------------- /timm/layers/separable_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/separable_conv.py -------------------------------------------------------------------------------- /timm/layers/space_to_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/space_to_depth.py -------------------------------------------------------------------------------- /timm/layers/split_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/split_attn.py -------------------------------------------------------------------------------- /timm/layers/split_batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/split_batchnorm.py -------------------------------------------------------------------------------- /timm/layers/squeeze_excite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/squeeze_excite.py -------------------------------------------------------------------------------- /timm/layers/std_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/std_conv.py -------------------------------------------------------------------------------- /timm/layers/test_time_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/test_time_pool.py -------------------------------------------------------------------------------- /timm/layers/trace_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/trace_utils.py -------------------------------------------------------------------------------- /timm/layers/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/layers/weight_init.py -------------------------------------------------------------------------------- /timm/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/loss/__init__.py -------------------------------------------------------------------------------- /timm/loss/asymmetric_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/loss/asymmetric_loss.py -------------------------------------------------------------------------------- /timm/loss/binary_cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/loss/binary_cross_entropy.py -------------------------------------------------------------------------------- /timm/loss/cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/loss/cross_entropy.py -------------------------------------------------------------------------------- /timm/loss/jsd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/loss/jsd.py -------------------------------------------------------------------------------- /timm/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/__init__.py -------------------------------------------------------------------------------- /timm/models/_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_builder.py -------------------------------------------------------------------------------- /timm/models/_efficientnet_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_efficientnet_blocks.py -------------------------------------------------------------------------------- /timm/models/_efficientnet_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_efficientnet_builder.py -------------------------------------------------------------------------------- /timm/models/_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_factory.py -------------------------------------------------------------------------------- /timm/models/_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_features.py -------------------------------------------------------------------------------- /timm/models/_features_fx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_features_fx.py -------------------------------------------------------------------------------- /timm/models/_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_helpers.py -------------------------------------------------------------------------------- /timm/models/_hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_hub.py -------------------------------------------------------------------------------- /timm/models/_manipulate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_manipulate.py -------------------------------------------------------------------------------- /timm/models/_pretrained.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_pretrained.py -------------------------------------------------------------------------------- /timm/models/_prune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_prune.py -------------------------------------------------------------------------------- /timm/models/_pruned/ecaresnet101d_pruned.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_pruned/ecaresnet101d_pruned.txt -------------------------------------------------------------------------------- /timm/models/_pruned/ecaresnet50d_pruned.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_pruned/ecaresnet50d_pruned.txt -------------------------------------------------------------------------------- /timm/models/_pruned/efficientnet_b1_pruned.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_pruned/efficientnet_b1_pruned.txt -------------------------------------------------------------------------------- /timm/models/_pruned/efficientnet_b2_pruned.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_pruned/efficientnet_b2_pruned.txt -------------------------------------------------------------------------------- /timm/models/_pruned/efficientnet_b3_pruned.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_pruned/efficientnet_b3_pruned.txt -------------------------------------------------------------------------------- /timm/models/_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/_registry.py -------------------------------------------------------------------------------- /timm/models/beit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/beit.py -------------------------------------------------------------------------------- /timm/models/byoanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/byoanet.py -------------------------------------------------------------------------------- /timm/models/byobnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/byobnet.py -------------------------------------------------------------------------------- /timm/models/cait.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/cait.py -------------------------------------------------------------------------------- /timm/models/coat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/coat.py -------------------------------------------------------------------------------- /timm/models/convit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/convit.py -------------------------------------------------------------------------------- /timm/models/convmixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/convmixer.py -------------------------------------------------------------------------------- /timm/models/convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/convnext.py -------------------------------------------------------------------------------- /timm/models/crossvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/crossvit.py -------------------------------------------------------------------------------- /timm/models/cspnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/cspnet.py -------------------------------------------------------------------------------- /timm/models/deit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/deit.py -------------------------------------------------------------------------------- /timm/models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/densenet.py -------------------------------------------------------------------------------- /timm/models/dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/dla.py -------------------------------------------------------------------------------- /timm/models/dpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/dpn.py -------------------------------------------------------------------------------- /timm/models/edgenext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/edgenext.py -------------------------------------------------------------------------------- /timm/models/efficientformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/efficientformer.py -------------------------------------------------------------------------------- /timm/models/efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/efficientnet.py -------------------------------------------------------------------------------- /timm/models/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/factory.py -------------------------------------------------------------------------------- /timm/models/features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/features.py -------------------------------------------------------------------------------- /timm/models/fx_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/fx_features.py -------------------------------------------------------------------------------- /timm/models/gcvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/gcvit.py -------------------------------------------------------------------------------- /timm/models/ghostnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/ghostnet.py -------------------------------------------------------------------------------- /timm/models/gluon_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/gluon_resnet.py -------------------------------------------------------------------------------- /timm/models/gluon_xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/gluon_xception.py -------------------------------------------------------------------------------- /timm/models/hardcorenas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/hardcorenas.py -------------------------------------------------------------------------------- /timm/models/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/helpers.py -------------------------------------------------------------------------------- /timm/models/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/hrnet.py -------------------------------------------------------------------------------- /timm/models/hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/hub.py -------------------------------------------------------------------------------- /timm/models/inception_resnet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/inception_resnet_v2.py -------------------------------------------------------------------------------- /timm/models/inception_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/inception_v3.py -------------------------------------------------------------------------------- /timm/models/inception_v4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/inception_v4.py -------------------------------------------------------------------------------- /timm/models/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/layers/__init__.py -------------------------------------------------------------------------------- /timm/models/levit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/levit.py -------------------------------------------------------------------------------- /timm/models/maxxvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/maxxvit.py -------------------------------------------------------------------------------- /timm/models/mlp_mixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/mlp_mixer.py -------------------------------------------------------------------------------- /timm/models/mobilenetv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/mobilenetv3.py -------------------------------------------------------------------------------- /timm/models/mobilevit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/mobilevit.py -------------------------------------------------------------------------------- /timm/models/mvitv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/mvitv2.py -------------------------------------------------------------------------------- /timm/models/nasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/nasnet.py -------------------------------------------------------------------------------- /timm/models/nest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/nest.py -------------------------------------------------------------------------------- /timm/models/nfnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/nfnet.py -------------------------------------------------------------------------------- /timm/models/pit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/pit.py -------------------------------------------------------------------------------- /timm/models/pnasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/pnasnet.py -------------------------------------------------------------------------------- /timm/models/poolformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/poolformer.py -------------------------------------------------------------------------------- /timm/models/pvt_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/pvt_v2.py -------------------------------------------------------------------------------- /timm/models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/registry.py -------------------------------------------------------------------------------- /timm/models/regnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/regnet.py -------------------------------------------------------------------------------- /timm/models/res2net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/res2net.py -------------------------------------------------------------------------------- /timm/models/resnest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/resnest.py -------------------------------------------------------------------------------- /timm/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/resnet.py -------------------------------------------------------------------------------- /timm/models/resnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/resnetv2.py -------------------------------------------------------------------------------- /timm/models/rexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/rexnet.py -------------------------------------------------------------------------------- /timm/models/selecsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/selecsls.py -------------------------------------------------------------------------------- /timm/models/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/senet.py -------------------------------------------------------------------------------- /timm/models/sequencer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/sequencer.py -------------------------------------------------------------------------------- /timm/models/sknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/sknet.py -------------------------------------------------------------------------------- /timm/models/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/swin_transformer.py -------------------------------------------------------------------------------- /timm/models/swin_transformer_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/swin_transformer_v2.py -------------------------------------------------------------------------------- /timm/models/swin_transformer_v2_cr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/swin_transformer_v2_cr.py -------------------------------------------------------------------------------- /timm/models/tnt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/tnt.py -------------------------------------------------------------------------------- /timm/models/tresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/tresnet.py -------------------------------------------------------------------------------- /timm/models/twins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/twins.py -------------------------------------------------------------------------------- /timm/models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/vgg.py -------------------------------------------------------------------------------- /timm/models/visformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/visformer.py -------------------------------------------------------------------------------- /timm/models/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/vision_transformer.py -------------------------------------------------------------------------------- /timm/models/vision_transformer_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/vision_transformer_hybrid.py -------------------------------------------------------------------------------- /timm/models/vision_transformer_relpos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/vision_transformer_relpos.py -------------------------------------------------------------------------------- /timm/models/volo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/volo.py -------------------------------------------------------------------------------- /timm/models/vovnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/vovnet.py -------------------------------------------------------------------------------- /timm/models/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/xception.py -------------------------------------------------------------------------------- /timm/models/xception_aligned.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/xception_aligned.py -------------------------------------------------------------------------------- /timm/models/xcit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/models/xcit.py -------------------------------------------------------------------------------- /timm/optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/__init__.py -------------------------------------------------------------------------------- /timm/optim/adabelief.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/adabelief.py -------------------------------------------------------------------------------- /timm/optim/adafactor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/adafactor.py -------------------------------------------------------------------------------- /timm/optim/adahessian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/adahessian.py -------------------------------------------------------------------------------- /timm/optim/adamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/adamp.py -------------------------------------------------------------------------------- /timm/optim/adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/adamw.py -------------------------------------------------------------------------------- /timm/optim/adan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/adan.py -------------------------------------------------------------------------------- /timm/optim/lamb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/lamb.py -------------------------------------------------------------------------------- /timm/optim/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/lars.py -------------------------------------------------------------------------------- /timm/optim/lookahead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/lookahead.py -------------------------------------------------------------------------------- /timm/optim/madgrad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/madgrad.py -------------------------------------------------------------------------------- /timm/optim/nadam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/nadam.py -------------------------------------------------------------------------------- /timm/optim/nvnovograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/nvnovograd.py -------------------------------------------------------------------------------- /timm/optim/optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/optim_factory.py -------------------------------------------------------------------------------- /timm/optim/radam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/radam.py -------------------------------------------------------------------------------- /timm/optim/rmsprop_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/rmsprop_tf.py -------------------------------------------------------------------------------- /timm/optim/sgdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/optim/sgdp.py -------------------------------------------------------------------------------- /timm/scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/scheduler/__init__.py -------------------------------------------------------------------------------- /timm/scheduler/cosine_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/scheduler/cosine_lr.py -------------------------------------------------------------------------------- /timm/scheduler/multistep_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/scheduler/multistep_lr.py -------------------------------------------------------------------------------- /timm/scheduler/plateau_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/scheduler/plateau_lr.py -------------------------------------------------------------------------------- /timm/scheduler/poly_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/scheduler/poly_lr.py -------------------------------------------------------------------------------- /timm/scheduler/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/scheduler/scheduler.py -------------------------------------------------------------------------------- /timm/scheduler/scheduler_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/scheduler/scheduler_factory.py -------------------------------------------------------------------------------- /timm/scheduler/step_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/scheduler/step_lr.py -------------------------------------------------------------------------------- /timm/scheduler/tanh_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/scheduler/tanh_lr.py -------------------------------------------------------------------------------- /timm/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/utils/__init__.py -------------------------------------------------------------------------------- /timm/utils/agc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/utils/agc.py -------------------------------------------------------------------------------- /timm/utils/checkpoint_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/utils/checkpoint_saver.py -------------------------------------------------------------------------------- /timm/utils/clip_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/utils/clip_grad.py -------------------------------------------------------------------------------- /timm/utils/cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/utils/cuda.py -------------------------------------------------------------------------------- /timm/utils/decay_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/utils/decay_batch.py -------------------------------------------------------------------------------- /timm/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/utils/distributed.py -------------------------------------------------------------------------------- /timm/utils/jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/utils/jit.py -------------------------------------------------------------------------------- /timm/utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/utils/log.py -------------------------------------------------------------------------------- /timm/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/utils/metrics.py -------------------------------------------------------------------------------- /timm/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/utils/misc.py -------------------------------------------------------------------------------- /timm/utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/utils/model.py -------------------------------------------------------------------------------- /timm/utils/model_ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/utils/model_ema.py -------------------------------------------------------------------------------- /timm/utils/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/utils/random.py -------------------------------------------------------------------------------- /timm/utils/summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/timm/utils/summary.py -------------------------------------------------------------------------------- /timm/version.py: -------------------------------------------------------------------------------- 1 | __version__ = '0.8.1dev0' 2 | -------------------------------------------------------------------------------- /tools/change_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/change_weights.py -------------------------------------------------------------------------------- /tools/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/eval.py -------------------------------------------------------------------------------- /tools/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/get_flops.py -------------------------------------------------------------------------------- /tools/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/infer.py -------------------------------------------------------------------------------- /tools/partial_quantization/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/partial_quantization/README.md -------------------------------------------------------------------------------- /tools/partial_quantization/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/partial_quantization/eval.py -------------------------------------------------------------------------------- /tools/partial_quantization/eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/partial_quantization/eval.yaml -------------------------------------------------------------------------------- /tools/partial_quantization/partial_quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/partial_quantization/partial_quant.py -------------------------------------------------------------------------------- /tools/partial_quantization/ptq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/partial_quantization/ptq.py -------------------------------------------------------------------------------- /tools/partial_quantization/sensitivity_analyse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/partial_quantization/sensitivity_analyse.py -------------------------------------------------------------------------------- /tools/partial_quantization/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/partial_quantization/utils.py -------------------------------------------------------------------------------- /tools/qat/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/qat/README.md -------------------------------------------------------------------------------- /tools/qat/qat_export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/qat/qat_export.py -------------------------------------------------------------------------------- /tools/qat/qat_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/qat/qat_utils.py -------------------------------------------------------------------------------- /tools/quantization/mnn/README.md: -------------------------------------------------------------------------------- 1 | # Coming soon 2 | -------------------------------------------------------------------------------- /tools/quantization/tensorrt/post_training/Calibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/quantization/tensorrt/post_training/Calibrator.py -------------------------------------------------------------------------------- /tools/quantization/tensorrt/post_training/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/quantization/tensorrt/post_training/LICENSE -------------------------------------------------------------------------------- /tools/quantization/tensorrt/post_training/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/quantization/tensorrt/post_training/README.md -------------------------------------------------------------------------------- /tools/quantization/tensorrt/post_training/onnx_to_tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/quantization/tensorrt/post_training/onnx_to_tensorrt.py -------------------------------------------------------------------------------- /tools/quantization/tensorrt/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/quantization/tensorrt/requirements.txt -------------------------------------------------------------------------------- /tools/quantization/tensorrt/training_aware/QAT_quantizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/quantization/tensorrt/training_aware/QAT_quantizer.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/tools/train.py -------------------------------------------------------------------------------- /utils/RepulsionLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/RepulsionLoss.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/activations.py -------------------------------------------------------------------------------- /utils/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/augmentations.py -------------------------------------------------------------------------------- /utils/autoanchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/autoanchor.py -------------------------------------------------------------------------------- /utils/autobatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/autobatch.py -------------------------------------------------------------------------------- /utils/aws/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/aws/resume.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/aws/resume.py -------------------------------------------------------------------------------- /utils/benchmarks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/benchmarks.py -------------------------------------------------------------------------------- /utils/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/callbacks.py -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/downloads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/downloads.py -------------------------------------------------------------------------------- /utils/flask_rest_api/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/flask_rest_api/README.md -------------------------------------------------------------------------------- /utils/flask_rest_api/example_request.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/flask_rest_api/example_request.py -------------------------------------------------------------------------------- /utils/flask_rest_api/restapi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/flask_rest_api/restapi.py -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/google_app_engine/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/google_app_engine/Dockerfile -------------------------------------------------------------------------------- /utils/google_app_engine/additional_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/google_app_engine/additional_requirements.txt -------------------------------------------------------------------------------- /utils/google_app_engine/app.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/google_app_engine/app.yaml -------------------------------------------------------------------------------- /utils/loggers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/loggers/__init__.py -------------------------------------------------------------------------------- /utils/loggers/wandb/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/loggers/wandb/README.md -------------------------------------------------------------------------------- /utils/loggers/wandb/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/loggers/wandb/log_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/loggers/wandb/log_dataset.py -------------------------------------------------------------------------------- /utils/loggers/wandb/sweep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/loggers/wandb/sweep.py -------------------------------------------------------------------------------- /utils/loggers/wandb/sweep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/loggers/wandb/sweep.yaml -------------------------------------------------------------------------------- /utils/loggers/wandb/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/loggers/wandb/wandb_utils.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/loss_ps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/loss_ps.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/plots.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /utils/yolo_with_plugins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/yolo_with_plugins.py -------------------------------------------------------------------------------- /utils/yolov6official.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/utils/yolov6official.py -------------------------------------------------------------------------------- /yolov6/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /yolov6/assigners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/assigners/__init__.py -------------------------------------------------------------------------------- /yolov6/assigners/anchor_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/assigners/anchor_generator.py -------------------------------------------------------------------------------- /yolov6/assigners/assigner_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/assigners/assigner_utils.py -------------------------------------------------------------------------------- /yolov6/assigners/atss_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/assigners/atss_assigner.py -------------------------------------------------------------------------------- /yolov6/assigners/iou2d_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/assigners/iou2d_calculator.py -------------------------------------------------------------------------------- /yolov6/assigners/tal_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/assigners/tal_assigner.py -------------------------------------------------------------------------------- /yolov6/core/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/core/engine.py -------------------------------------------------------------------------------- /yolov6/core/evaler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/core/evaler.py -------------------------------------------------------------------------------- /yolov6/core/inferer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/core/inferer.py -------------------------------------------------------------------------------- /yolov6/data/data_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/data/data_augment.py -------------------------------------------------------------------------------- /yolov6/data/data_load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/data/data_load.py -------------------------------------------------------------------------------- /yolov6/data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/data/datasets.py -------------------------------------------------------------------------------- /yolov6/data/vis_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/data/vis_dataset.py -------------------------------------------------------------------------------- /yolov6/data/voc2yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/data/voc2yolo.py -------------------------------------------------------------------------------- /yolov6/layers/BotNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/layers/BotNet.py -------------------------------------------------------------------------------- /yolov6/layers/CoAtNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/layers/CoAtNet.py -------------------------------------------------------------------------------- /yolov6/layers/RTMDet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/layers/RTMDet.py -------------------------------------------------------------------------------- /yolov6/layers/TAP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/layers/TAP.py -------------------------------------------------------------------------------- /yolov6/layers/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/layers/common.py -------------------------------------------------------------------------------- /yolov6/layers/damo_yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/layers/damo_yolo.py -------------------------------------------------------------------------------- /yolov6/layers/dbb_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/layers/dbb_transforms.py -------------------------------------------------------------------------------- /yolov6/layers/focal_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/layers/focal_transformer.py -------------------------------------------------------------------------------- /yolov6/layers/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/layers/misc.py -------------------------------------------------------------------------------- /yolov6/layers/tiny_nas_csp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/layers/tiny_nas_csp.py -------------------------------------------------------------------------------- /yolov6/layers/yolov7.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/layers/yolov7.py -------------------------------------------------------------------------------- /yolov6/models/efficientrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/models/efficientrep.py -------------------------------------------------------------------------------- /yolov6/models/effidehead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/models/effidehead.py -------------------------------------------------------------------------------- /yolov6/models/end2end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/models/end2end.py -------------------------------------------------------------------------------- /yolov6/models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/models/loss.py -------------------------------------------------------------------------------- /yolov6/models/loss_distill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/models/loss_distill.py -------------------------------------------------------------------------------- /yolov6/models/reppan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/models/reppan.py -------------------------------------------------------------------------------- /yolov6/models/simOTA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/models/simOTA.py -------------------------------------------------------------------------------- /yolov6/models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/models/yolo.py -------------------------------------------------------------------------------- /yolov6/solver/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/solver/build.py -------------------------------------------------------------------------------- /yolov6/utils/Arial.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/utils/Arial.ttf -------------------------------------------------------------------------------- /yolov6/utils/RepOptimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/utils/RepOptimizer.py -------------------------------------------------------------------------------- /yolov6/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/utils/checkpoint.py -------------------------------------------------------------------------------- /yolov6/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/utils/config.py -------------------------------------------------------------------------------- /yolov6/utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/utils/ema.py -------------------------------------------------------------------------------- /yolov6/utils/envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/utils/envs.py -------------------------------------------------------------------------------- /yolov6/utils/events.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/utils/events.py -------------------------------------------------------------------------------- /yolov6/utils/figure_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/utils/figure_iou.py -------------------------------------------------------------------------------- /yolov6/utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/utils/general.py -------------------------------------------------------------------------------- /yolov6/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/utils/metrics.py -------------------------------------------------------------------------------- /yolov6/utils/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/utils/nms.py -------------------------------------------------------------------------------- /yolov6/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yang-0201/YOLOv6_pro/HEAD/yolov6/utils/torch_utils.py --------------------------------------------------------------------------------