├── .gitignore ├── DATASET.md ├── LICENSE ├── README.md ├── configs ├── det │ ├── _base_ │ │ ├── datasets │ │ │ ├── cityscapes_panoptic.py │ │ │ ├── cityscapes_step.py │ │ │ ├── cityscapes_vps_clips.py │ │ │ ├── cityscapes_vps_clips_trainval.py │ │ │ ├── coco_instance.py │ │ │ ├── coco_panoptic.py │ │ │ ├── coco_panoptic_instance_annotations.py │ │ │ ├── kitti_step_dvps.py │ │ │ ├── kitti_step_vps.py │ │ │ ├── kitti_step_vps_trainval.py │ │ │ ├── mapillary_panoptic.py │ │ │ └── vipseg_dvps.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ ├── knet_citystep_s3_r50_fpn.py │ │ │ ├── knet_kitti_step_s3_r50_fpn.py │ │ │ ├── knet_s3_r50_deformable_fpn.py │ │ │ ├── knet_s3_r50_fpn.py │ │ │ ├── knet_s3_r50_fpn_panoptic.py │ │ │ ├── knet_vipseg_s3_r50_fpn.py │ │ │ └── video_knet_s3_r50_fpn_panoptic.py │ │ └── schedules │ │ │ ├── schedule_10e.py │ │ │ └── schedule_1x.py │ ├── coco │ │ ├── knet_s3_r50_deformable_fpn_ms-3x_coco.py │ │ ├── knet_s3_r50_fpn_ms-3x_coco-panoptic.py │ │ ├── knet_s3_r50_fpn_ms-3x_coco.py │ │ └── knet_s3_swin-b_deformable_fpn_ms-3x_coco.py │ ├── common │ │ ├── lsj_coco_panoptic_50e.py │ │ ├── mstrain_3x_coco_instance.py │ │ ├── mstrain_3x_coco_panoptic_inst_anno.py │ │ ├── mstrain_3x_coco_panoptic_inst_anno_detr_aug.py │ │ └── mstrain_64e_city_panoptic.py │ ├── knet_cityscapes_step │ │ ├── knet_s3_r50_fpn.py │ │ ├── knet_s3_swin_b_fpn.py │ │ └── knet_s3_swin_l_fpn.py │ ├── video_knet_kitti_step │ │ ├── video_knet_s3_r50_rpn_1x_kitti_step_sigmoid_stride2_mask_embed_link_ffn_joint_train.py │ │ ├── video_knet_s3_r50_rpn_1x_kitti_step_sigmoid_stride2_mask_embed_link_ffn_joint_train_8e.py │ │ ├── video_knet_s3_swinb_rpn_1x_kitti_step_sigmoid_stride2_mask_embed_link_ffn_joint_update.py │ │ ├── video_knet_s3_swinl_rpn_1x_kitti_step_sigmoid_stride2_mask_embed_link_ffn_joint_update.py │ │ └── video_knet_s3_swinl_rpn_1x_kitti_step_sigmoid_stride2_mask_embed_link_ffn_update_conv_short_track_fc.py │ └── video_knet_vipseg │ │ ├── video_knet_s3_r50_rpn_vipseg_mask_embed_link_ffn_joint_train.py │ │ └── video_knet_s3_swin_b_rpn_vipseg_mask_embed_link_ffn_joint_train_8e.py └── video_knet_vis │ ├── _base_ │ ├── datasets │ │ ├── coco_instance.py │ │ └── youtubevis_2019.py │ ├── default_runtime.py │ ├── models │ │ ├── knet_track_r50.py │ │ └── knet_track_r50_deformablefpn.py │ └── schedules │ │ ├── schedule_0.75x.py │ │ ├── schedule_1x.py │ │ └── schedule_8e.py │ ├── common │ └── mstrain_3x_coco_instance.py │ └── video_knet_vis │ ├── knet_track_r50_1x_youtubevis.py │ ├── knet_track_r50_deformable_fpn_1x_youtubevis.py │ ├── knet_track_swinb_1x_youtubevis_8e.py │ └── knet_track_swinb_deformable_1x_youtubevis.py ├── external ├── cityscape_panoptic.py ├── cityscapes_step.py ├── cityscapes_vps.py ├── coco_panoptic.py ├── dataset │ ├── dvps_pipelines │ │ ├── __init__.py │ │ ├── loading.py │ │ ├── transforms.py │ │ └── tricks.py │ ├── forecasting_pipelines │ │ ├── __init__.py │ │ ├── loading.py │ │ └── transforms.py │ ├── mIoU.py │ └── pipelines │ │ ├── __init__.py │ │ ├── formatting.py │ │ ├── loading.py │ │ ├── test_time_aug.py │ │ └── transforms.py ├── evalhooks.py ├── ext │ ├── mask.py │ └── ytvos.py ├── fcn_mask_head.py ├── kitti_step_dvps.py ├── panoptic_fpn.py ├── panoptic_head.py ├── semantic_seg_head.py ├── semkitti_dvps.py ├── test.py ├── train.py ├── utils.py └── vipseg_dvps.py ├── figs ├── cityscapes_vps_video_1_20220318131729.gif ├── cityscapes_vps_video_2_20220318132943.gif ├── result_vps.png ├── step_video_1_20220318133227.gif └── step_video_2_20220318133423.gif ├── knet ├── __init__.py ├── cross_entropy_loss.py ├── det │ ├── dice_loss.py │ ├── kernel_head.py │ ├── kernel_iter_head.py │ ├── kernel_update_head.py │ ├── knet.py │ ├── mask_hungarian_assigner.py │ ├── mask_pseudo_sampler.py │ ├── msdeformattn_decoder.py │ ├── semantic_fpn_wrapper.py │ └── utils.py ├── kernel_updator.py └── video │ ├── __init__.py │ ├── dice_loss.py │ ├── kernel_head.py │ ├── kernel_iter_head.py │ ├── kernel_update_head.py │ ├── knet.py │ ├── knet_quansi_dense.py │ ├── knet_quansi_dense_embed_fc.py │ ├── knet_quansi_dense_embed_fc_joint_train.py │ ├── knet_quansi_dense_embed_fc_toy_exp.py │ ├── knet_quansi_dense_roi_gt_box.py │ ├── knet_quansi_dense_roi_gt_box_joint_train.py │ ├── knet_track_head.py │ ├── knet_track_head_roi_align.py │ ├── knet_uni_track.py │ ├── mask_hungarian_assigner.py │ ├── mask_pseudo_sampler.py │ ├── qdtrack │ ├── builder.py │ ├── losses │ │ ├── __init__.py │ │ ├── l2_loss.py │ │ └── multipos_cross_entropy_loss.py │ ├── track │ │ ├── __init__.py │ │ ├── similarity.py │ │ └── transforms.py │ └── trackers │ │ ├── __init__.py │ │ ├── quasi_dense_embed_tracker.py │ │ └── tao_tracker.py │ ├── track_heads.py │ ├── tracker.py │ └── util.py ├── knet_vis ├── __init__.py ├── det │ ├── __init__.py │ ├── kernel_head.py │ ├── kernel_iter_head.py │ ├── kernel_update_head.py │ ├── knet.py │ ├── mask_hungarian_assigner.py │ ├── mask_pseudo_sampler.py │ ├── semantic_fpn_wrapper.py │ └── utils.py ├── kernel_updator.py └── tracker │ ├── __init__.py │ ├── kernel_frame_head.py │ ├── kernel_frame_iter_head.py │ ├── kernel_head.py │ ├── kernel_iter_head.py │ ├── kernel_update_head.py │ ├── mask_hungarian_assigner.py │ ├── positional_encoding.py │ ├── semantic_fpn_wrapper3D.py │ └── track.py ├── mmtrack ├── datasets │ ├── coco_video_dataset.py │ ├── parsers │ │ ├── __init__.py │ │ └── coco_video_parser.py │ └── youtube_vis_dataset.py ├── pipelines │ ├── __init__.py │ ├── formatting.py │ ├── loading.py │ ├── test_time_aug.py │ └── transforms.py └── transform.py ├── scripts ├── kitti_step_prepare.py └── visualizer.py ├── slides ├── Video-KNet-cvpr-slides-10-25-version.pdf └── cvpr22_poster_lxt_zww_pjm.pdf ├── swin ├── DetectRS.py ├── ckpt_convert.py ├── mix_transformer.py ├── swin_checkpoint.py ├── swin_transformer.py ├── swin_transformer_rfp.py └── transformer.py ├── tools ├── dataset │ ├── cityscapes_instance_idmap.py │ └── youtubevis2coco.py ├── dist_step_test.sh ├── dist_test.sh ├── dist_train.sh ├── dist_train_new.sh ├── dist_vps_test.sh ├── docker.sh ├── eval_dstq.py ├── eval_dstq_step.py ├── eval_dstq_vipseg.py ├── eval_dvpq_step.py ├── eval_dvpq_vipseg.py ├── flops_counter.py ├── get_flops.py ├── inference_kitti_step.sh ├── slurm_test.sh ├── slurm_test_dvps.sh ├── slurm_test_step.sh ├── slurm_test_vis.sh ├── slurm_test_vps.sh ├── slurm_train.sh ├── test.py ├── test_dvps.py ├── test_step.py ├── test_vps.py ├── train.py ├── utils │ ├── DSTQ.py │ ├── STQ.py │ └── cityscapesvps_eval.py └── visualization.py ├── tools_vis ├── apis │ ├── __init__.py │ └── test.py ├── dist_test_whole_video.sh ├── docker.sh ├── slurm_test_vis.sh ├── test.py └── test_whole_video.py └── unitrack ├── __init__.py ├── basetrack.py ├── box.py ├── core ├── __init__.py ├── association │ ├── __init__.py │ └── matching.py ├── motion │ └── kalman_filter.py └── propagation │ ├── __init__.py │ ├── propagate_box.py │ ├── propagate_mask.py │ └── propagate_pose.py ├── mask.py ├── mask_with_train_embs.py ├── model ├── __init__.py ├── functional.py ├── hrnet.py ├── model.py ├── random_feat_generator.py └── resnet.py ├── multitracker.py └── utils ├── __init__.py ├── box.py ├── io.py ├── log.py ├── mask.py ├── meter.py ├── palette.py └── visualize.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/.gitignore -------------------------------------------------------------------------------- /DATASET.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/DATASET.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/README.md -------------------------------------------------------------------------------- /configs/det/_base_/datasets/cityscapes_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/datasets/cityscapes_panoptic.py -------------------------------------------------------------------------------- /configs/det/_base_/datasets/cityscapes_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/datasets/cityscapes_step.py -------------------------------------------------------------------------------- /configs/det/_base_/datasets/cityscapes_vps_clips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/datasets/cityscapes_vps_clips.py -------------------------------------------------------------------------------- /configs/det/_base_/datasets/cityscapes_vps_clips_trainval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/datasets/cityscapes_vps_clips_trainval.py -------------------------------------------------------------------------------- /configs/det/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /configs/det/_base_/datasets/coco_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/datasets/coco_panoptic.py -------------------------------------------------------------------------------- /configs/det/_base_/datasets/coco_panoptic_instance_annotations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/datasets/coco_panoptic_instance_annotations.py -------------------------------------------------------------------------------- /configs/det/_base_/datasets/kitti_step_dvps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/datasets/kitti_step_dvps.py -------------------------------------------------------------------------------- /configs/det/_base_/datasets/kitti_step_vps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/datasets/kitti_step_vps.py -------------------------------------------------------------------------------- /configs/det/_base_/datasets/kitti_step_vps_trainval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/datasets/kitti_step_vps_trainval.py -------------------------------------------------------------------------------- /configs/det/_base_/datasets/mapillary_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/datasets/mapillary_panoptic.py -------------------------------------------------------------------------------- /configs/det/_base_/datasets/vipseg_dvps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/datasets/vipseg_dvps.py -------------------------------------------------------------------------------- /configs/det/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/det/_base_/models/knet_citystep_s3_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/models/knet_citystep_s3_r50_fpn.py -------------------------------------------------------------------------------- /configs/det/_base_/models/knet_kitti_step_s3_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/models/knet_kitti_step_s3_r50_fpn.py -------------------------------------------------------------------------------- /configs/det/_base_/models/knet_s3_r50_deformable_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/models/knet_s3_r50_deformable_fpn.py -------------------------------------------------------------------------------- /configs/det/_base_/models/knet_s3_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/models/knet_s3_r50_fpn.py -------------------------------------------------------------------------------- /configs/det/_base_/models/knet_s3_r50_fpn_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/models/knet_s3_r50_fpn_panoptic.py -------------------------------------------------------------------------------- /configs/det/_base_/models/knet_vipseg_s3_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/models/knet_vipseg_s3_r50_fpn.py -------------------------------------------------------------------------------- /configs/det/_base_/models/video_knet_s3_r50_fpn_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/models/video_knet_s3_r50_fpn_panoptic.py -------------------------------------------------------------------------------- /configs/det/_base_/schedules/schedule_10e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/schedules/schedule_10e.py -------------------------------------------------------------------------------- /configs/det/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /configs/det/coco/knet_s3_r50_deformable_fpn_ms-3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/coco/knet_s3_r50_deformable_fpn_ms-3x_coco.py -------------------------------------------------------------------------------- /configs/det/coco/knet_s3_r50_fpn_ms-3x_coco-panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/coco/knet_s3_r50_fpn_ms-3x_coco-panoptic.py -------------------------------------------------------------------------------- /configs/det/coco/knet_s3_r50_fpn_ms-3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/coco/knet_s3_r50_fpn_ms-3x_coco.py -------------------------------------------------------------------------------- /configs/det/coco/knet_s3_swin-b_deformable_fpn_ms-3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/coco/knet_s3_swin-b_deformable_fpn_ms-3x_coco.py -------------------------------------------------------------------------------- /configs/det/common/lsj_coco_panoptic_50e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/common/lsj_coco_panoptic_50e.py -------------------------------------------------------------------------------- /configs/det/common/mstrain_3x_coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/common/mstrain_3x_coco_instance.py -------------------------------------------------------------------------------- /configs/det/common/mstrain_3x_coco_panoptic_inst_anno.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/common/mstrain_3x_coco_panoptic_inst_anno.py -------------------------------------------------------------------------------- /configs/det/common/mstrain_3x_coco_panoptic_inst_anno_detr_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/common/mstrain_3x_coco_panoptic_inst_anno_detr_aug.py -------------------------------------------------------------------------------- /configs/det/common/mstrain_64e_city_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/common/mstrain_64e_city_panoptic.py -------------------------------------------------------------------------------- /configs/det/knet_cityscapes_step/knet_s3_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/knet_cityscapes_step/knet_s3_r50_fpn.py -------------------------------------------------------------------------------- /configs/det/knet_cityscapes_step/knet_s3_swin_b_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/knet_cityscapes_step/knet_s3_swin_b_fpn.py -------------------------------------------------------------------------------- /configs/det/knet_cityscapes_step/knet_s3_swin_l_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/knet_cityscapes_step/knet_s3_swin_l_fpn.py -------------------------------------------------------------------------------- /configs/det/video_knet_kitti_step/video_knet_s3_r50_rpn_1x_kitti_step_sigmoid_stride2_mask_embed_link_ffn_joint_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/video_knet_kitti_step/video_knet_s3_r50_rpn_1x_kitti_step_sigmoid_stride2_mask_embed_link_ffn_joint_train.py -------------------------------------------------------------------------------- /configs/det/video_knet_kitti_step/video_knet_s3_r50_rpn_1x_kitti_step_sigmoid_stride2_mask_embed_link_ffn_joint_train_8e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/video_knet_kitti_step/video_knet_s3_r50_rpn_1x_kitti_step_sigmoid_stride2_mask_embed_link_ffn_joint_train_8e.py -------------------------------------------------------------------------------- /configs/det/video_knet_kitti_step/video_knet_s3_swinb_rpn_1x_kitti_step_sigmoid_stride2_mask_embed_link_ffn_joint_update.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/video_knet_kitti_step/video_knet_s3_swinb_rpn_1x_kitti_step_sigmoid_stride2_mask_embed_link_ffn_joint_update.py -------------------------------------------------------------------------------- /configs/det/video_knet_kitti_step/video_knet_s3_swinl_rpn_1x_kitti_step_sigmoid_stride2_mask_embed_link_ffn_joint_update.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/video_knet_kitti_step/video_knet_s3_swinl_rpn_1x_kitti_step_sigmoid_stride2_mask_embed_link_ffn_joint_update.py -------------------------------------------------------------------------------- /configs/det/video_knet_kitti_step/video_knet_s3_swinl_rpn_1x_kitti_step_sigmoid_stride2_mask_embed_link_ffn_update_conv_short_track_fc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/video_knet_kitti_step/video_knet_s3_swinl_rpn_1x_kitti_step_sigmoid_stride2_mask_embed_link_ffn_update_conv_short_track_fc.py -------------------------------------------------------------------------------- /configs/det/video_knet_vipseg/video_knet_s3_r50_rpn_vipseg_mask_embed_link_ffn_joint_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/video_knet_vipseg/video_knet_s3_r50_rpn_vipseg_mask_embed_link_ffn_joint_train.py -------------------------------------------------------------------------------- /configs/det/video_knet_vipseg/video_knet_s3_swin_b_rpn_vipseg_mask_embed_link_ffn_joint_train_8e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/det/video_knet_vipseg/video_knet_s3_swin_b_rpn_vipseg_mask_embed_link_ffn_joint_train_8e.py -------------------------------------------------------------------------------- /configs/video_knet_vis/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/video_knet_vis/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /configs/video_knet_vis/_base_/datasets/youtubevis_2019.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/video_knet_vis/_base_/datasets/youtubevis_2019.py -------------------------------------------------------------------------------- /configs/video_knet_vis/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/video_knet_vis/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/video_knet_vis/_base_/models/knet_track_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/video_knet_vis/_base_/models/knet_track_r50.py -------------------------------------------------------------------------------- /configs/video_knet_vis/_base_/models/knet_track_r50_deformablefpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/video_knet_vis/_base_/models/knet_track_r50_deformablefpn.py -------------------------------------------------------------------------------- /configs/video_knet_vis/_base_/schedules/schedule_0.75x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/video_knet_vis/_base_/schedules/schedule_0.75x.py -------------------------------------------------------------------------------- /configs/video_knet_vis/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/video_knet_vis/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /configs/video_knet_vis/_base_/schedules/schedule_8e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/video_knet_vis/_base_/schedules/schedule_8e.py -------------------------------------------------------------------------------- /configs/video_knet_vis/common/mstrain_3x_coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/video_knet_vis/common/mstrain_3x_coco_instance.py -------------------------------------------------------------------------------- /configs/video_knet_vis/video_knet_vis/knet_track_r50_1x_youtubevis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/video_knet_vis/video_knet_vis/knet_track_r50_1x_youtubevis.py -------------------------------------------------------------------------------- /configs/video_knet_vis/video_knet_vis/knet_track_r50_deformable_fpn_1x_youtubevis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/video_knet_vis/video_knet_vis/knet_track_r50_deformable_fpn_1x_youtubevis.py -------------------------------------------------------------------------------- /configs/video_knet_vis/video_knet_vis/knet_track_swinb_1x_youtubevis_8e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/video_knet_vis/video_knet_vis/knet_track_swinb_1x_youtubevis_8e.py -------------------------------------------------------------------------------- /configs/video_knet_vis/video_knet_vis/knet_track_swinb_deformable_1x_youtubevis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/configs/video_knet_vis/video_knet_vis/knet_track_swinb_deformable_1x_youtubevis.py -------------------------------------------------------------------------------- /external/cityscape_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/cityscape_panoptic.py -------------------------------------------------------------------------------- /external/cityscapes_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/cityscapes_step.py -------------------------------------------------------------------------------- /external/cityscapes_vps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/cityscapes_vps.py -------------------------------------------------------------------------------- /external/coco_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/coco_panoptic.py -------------------------------------------------------------------------------- /external/dataset/dvps_pipelines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /external/dataset/dvps_pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/dataset/dvps_pipelines/loading.py -------------------------------------------------------------------------------- /external/dataset/dvps_pipelines/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/dataset/dvps_pipelines/transforms.py -------------------------------------------------------------------------------- /external/dataset/dvps_pipelines/tricks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/dataset/dvps_pipelines/tricks.py -------------------------------------------------------------------------------- /external/dataset/forecasting_pipelines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /external/dataset/forecasting_pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/dataset/forecasting_pipelines/loading.py -------------------------------------------------------------------------------- /external/dataset/forecasting_pipelines/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/dataset/forecasting_pipelines/transforms.py -------------------------------------------------------------------------------- /external/dataset/mIoU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/dataset/mIoU.py -------------------------------------------------------------------------------- /external/dataset/pipelines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /external/dataset/pipelines/formatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/dataset/pipelines/formatting.py -------------------------------------------------------------------------------- /external/dataset/pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/dataset/pipelines/loading.py -------------------------------------------------------------------------------- /external/dataset/pipelines/test_time_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/dataset/pipelines/test_time_aug.py -------------------------------------------------------------------------------- /external/dataset/pipelines/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/dataset/pipelines/transforms.py -------------------------------------------------------------------------------- /external/evalhooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/evalhooks.py -------------------------------------------------------------------------------- /external/ext/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/ext/mask.py -------------------------------------------------------------------------------- /external/ext/ytvos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/ext/ytvos.py -------------------------------------------------------------------------------- /external/fcn_mask_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/fcn_mask_head.py -------------------------------------------------------------------------------- /external/kitti_step_dvps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/kitti_step_dvps.py -------------------------------------------------------------------------------- /external/panoptic_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/panoptic_fpn.py -------------------------------------------------------------------------------- /external/panoptic_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/panoptic_head.py -------------------------------------------------------------------------------- /external/semantic_seg_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/semantic_seg_head.py -------------------------------------------------------------------------------- /external/semkitti_dvps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/semkitti_dvps.py -------------------------------------------------------------------------------- /external/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/test.py -------------------------------------------------------------------------------- /external/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/train.py -------------------------------------------------------------------------------- /external/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/utils.py -------------------------------------------------------------------------------- /external/vipseg_dvps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/external/vipseg_dvps.py -------------------------------------------------------------------------------- /figs/cityscapes_vps_video_1_20220318131729.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/figs/cityscapes_vps_video_1_20220318131729.gif -------------------------------------------------------------------------------- /figs/cityscapes_vps_video_2_20220318132943.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/figs/cityscapes_vps_video_2_20220318132943.gif -------------------------------------------------------------------------------- /figs/result_vps.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/figs/result_vps.png -------------------------------------------------------------------------------- /figs/step_video_1_20220318133227.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/figs/step_video_1_20220318133227.gif -------------------------------------------------------------------------------- /figs/step_video_2_20220318133423.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/figs/step_video_2_20220318133423.gif -------------------------------------------------------------------------------- /knet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /knet/cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/cross_entropy_loss.py -------------------------------------------------------------------------------- /knet/det/dice_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/det/dice_loss.py -------------------------------------------------------------------------------- /knet/det/kernel_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/det/kernel_head.py -------------------------------------------------------------------------------- /knet/det/kernel_iter_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/det/kernel_iter_head.py -------------------------------------------------------------------------------- /knet/det/kernel_update_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/det/kernel_update_head.py -------------------------------------------------------------------------------- /knet/det/knet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/det/knet.py -------------------------------------------------------------------------------- /knet/det/mask_hungarian_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/det/mask_hungarian_assigner.py -------------------------------------------------------------------------------- /knet/det/mask_pseudo_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/det/mask_pseudo_sampler.py -------------------------------------------------------------------------------- /knet/det/msdeformattn_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/det/msdeformattn_decoder.py -------------------------------------------------------------------------------- /knet/det/semantic_fpn_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/det/semantic_fpn_wrapper.py -------------------------------------------------------------------------------- /knet/det/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/det/utils.py -------------------------------------------------------------------------------- /knet/kernel_updator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/kernel_updator.py -------------------------------------------------------------------------------- /knet/video/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /knet/video/dice_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/dice_loss.py -------------------------------------------------------------------------------- /knet/video/kernel_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/kernel_head.py -------------------------------------------------------------------------------- /knet/video/kernel_iter_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/kernel_iter_head.py -------------------------------------------------------------------------------- /knet/video/kernel_update_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/kernel_update_head.py -------------------------------------------------------------------------------- /knet/video/knet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/knet.py -------------------------------------------------------------------------------- /knet/video/knet_quansi_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/knet_quansi_dense.py -------------------------------------------------------------------------------- /knet/video/knet_quansi_dense_embed_fc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/knet_quansi_dense_embed_fc.py -------------------------------------------------------------------------------- /knet/video/knet_quansi_dense_embed_fc_joint_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/knet_quansi_dense_embed_fc_joint_train.py -------------------------------------------------------------------------------- /knet/video/knet_quansi_dense_embed_fc_toy_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/knet_quansi_dense_embed_fc_toy_exp.py -------------------------------------------------------------------------------- /knet/video/knet_quansi_dense_roi_gt_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/knet_quansi_dense_roi_gt_box.py -------------------------------------------------------------------------------- /knet/video/knet_quansi_dense_roi_gt_box_joint_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/knet_quansi_dense_roi_gt_box_joint_train.py -------------------------------------------------------------------------------- /knet/video/knet_track_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/knet_track_head.py -------------------------------------------------------------------------------- /knet/video/knet_track_head_roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/knet_track_head_roi_align.py -------------------------------------------------------------------------------- /knet/video/knet_uni_track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/knet_uni_track.py -------------------------------------------------------------------------------- /knet/video/mask_hungarian_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/mask_hungarian_assigner.py -------------------------------------------------------------------------------- /knet/video/mask_pseudo_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/mask_pseudo_sampler.py -------------------------------------------------------------------------------- /knet/video/qdtrack/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/qdtrack/builder.py -------------------------------------------------------------------------------- /knet/video/qdtrack/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/qdtrack/losses/__init__.py -------------------------------------------------------------------------------- /knet/video/qdtrack/losses/l2_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/qdtrack/losses/l2_loss.py -------------------------------------------------------------------------------- /knet/video/qdtrack/losses/multipos_cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/qdtrack/losses/multipos_cross_entropy_loss.py -------------------------------------------------------------------------------- /knet/video/qdtrack/track/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/qdtrack/track/__init__.py -------------------------------------------------------------------------------- /knet/video/qdtrack/track/similarity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/qdtrack/track/similarity.py -------------------------------------------------------------------------------- /knet/video/qdtrack/track/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/qdtrack/track/transforms.py -------------------------------------------------------------------------------- /knet/video/qdtrack/trackers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/qdtrack/trackers/__init__.py -------------------------------------------------------------------------------- /knet/video/qdtrack/trackers/quasi_dense_embed_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/qdtrack/trackers/quasi_dense_embed_tracker.py -------------------------------------------------------------------------------- /knet/video/qdtrack/trackers/tao_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/qdtrack/trackers/tao_tracker.py -------------------------------------------------------------------------------- /knet/video/track_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/track_heads.py -------------------------------------------------------------------------------- /knet/video/tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/tracker.py -------------------------------------------------------------------------------- /knet/video/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet/video/util.py -------------------------------------------------------------------------------- /knet_vis/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /knet_vis/det/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /knet_vis/det/kernel_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/det/kernel_head.py -------------------------------------------------------------------------------- /knet_vis/det/kernel_iter_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/det/kernel_iter_head.py -------------------------------------------------------------------------------- /knet_vis/det/kernel_update_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/det/kernel_update_head.py -------------------------------------------------------------------------------- /knet_vis/det/knet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/det/knet.py -------------------------------------------------------------------------------- /knet_vis/det/mask_hungarian_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/det/mask_hungarian_assigner.py -------------------------------------------------------------------------------- /knet_vis/det/mask_pseudo_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/det/mask_pseudo_sampler.py -------------------------------------------------------------------------------- /knet_vis/det/semantic_fpn_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/det/semantic_fpn_wrapper.py -------------------------------------------------------------------------------- /knet_vis/det/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/det/utils.py -------------------------------------------------------------------------------- /knet_vis/kernel_updator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/kernel_updator.py -------------------------------------------------------------------------------- /knet_vis/tracker/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /knet_vis/tracker/kernel_frame_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/tracker/kernel_frame_head.py -------------------------------------------------------------------------------- /knet_vis/tracker/kernel_frame_iter_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/tracker/kernel_frame_iter_head.py -------------------------------------------------------------------------------- /knet_vis/tracker/kernel_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/tracker/kernel_head.py -------------------------------------------------------------------------------- /knet_vis/tracker/kernel_iter_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/tracker/kernel_iter_head.py -------------------------------------------------------------------------------- /knet_vis/tracker/kernel_update_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/tracker/kernel_update_head.py -------------------------------------------------------------------------------- /knet_vis/tracker/mask_hungarian_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/tracker/mask_hungarian_assigner.py -------------------------------------------------------------------------------- /knet_vis/tracker/positional_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/tracker/positional_encoding.py -------------------------------------------------------------------------------- /knet_vis/tracker/semantic_fpn_wrapper3D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/tracker/semantic_fpn_wrapper3D.py -------------------------------------------------------------------------------- /knet_vis/tracker/track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/knet_vis/tracker/track.py -------------------------------------------------------------------------------- /mmtrack/datasets/coco_video_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/mmtrack/datasets/coco_video_dataset.py -------------------------------------------------------------------------------- /mmtrack/datasets/parsers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/mmtrack/datasets/parsers/__init__.py -------------------------------------------------------------------------------- /mmtrack/datasets/parsers/coco_video_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/mmtrack/datasets/parsers/coco_video_parser.py -------------------------------------------------------------------------------- /mmtrack/datasets/youtube_vis_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/mmtrack/datasets/youtube_vis_dataset.py -------------------------------------------------------------------------------- /mmtrack/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/mmtrack/pipelines/__init__.py -------------------------------------------------------------------------------- /mmtrack/pipelines/formatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/mmtrack/pipelines/formatting.py -------------------------------------------------------------------------------- /mmtrack/pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/mmtrack/pipelines/loading.py -------------------------------------------------------------------------------- /mmtrack/pipelines/test_time_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/mmtrack/pipelines/test_time_aug.py -------------------------------------------------------------------------------- /mmtrack/pipelines/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/mmtrack/pipelines/transforms.py -------------------------------------------------------------------------------- /mmtrack/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/mmtrack/transform.py -------------------------------------------------------------------------------- /scripts/kitti_step_prepare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/scripts/kitti_step_prepare.py -------------------------------------------------------------------------------- /scripts/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/scripts/visualizer.py -------------------------------------------------------------------------------- /slides/Video-KNet-cvpr-slides-10-25-version.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/slides/Video-KNet-cvpr-slides-10-25-version.pdf -------------------------------------------------------------------------------- /slides/cvpr22_poster_lxt_zww_pjm.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/slides/cvpr22_poster_lxt_zww_pjm.pdf -------------------------------------------------------------------------------- /swin/DetectRS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/swin/DetectRS.py -------------------------------------------------------------------------------- /swin/ckpt_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/swin/ckpt_convert.py -------------------------------------------------------------------------------- /swin/mix_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/swin/mix_transformer.py -------------------------------------------------------------------------------- /swin/swin_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/swin/swin_checkpoint.py -------------------------------------------------------------------------------- /swin/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/swin/swin_transformer.py -------------------------------------------------------------------------------- /swin/swin_transformer_rfp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/swin/swin_transformer_rfp.py -------------------------------------------------------------------------------- /swin/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/swin/transformer.py -------------------------------------------------------------------------------- /tools/dataset/cityscapes_instance_idmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/dataset/cityscapes_instance_idmap.py -------------------------------------------------------------------------------- /tools/dataset/youtubevis2coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/dataset/youtubevis2coco.py -------------------------------------------------------------------------------- /tools/dist_step_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/dist_step_test.sh -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/dist_train_new.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/dist_train_new.sh -------------------------------------------------------------------------------- /tools/dist_vps_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/dist_vps_test.sh -------------------------------------------------------------------------------- /tools/docker.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/docker.sh -------------------------------------------------------------------------------- /tools/eval_dstq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/eval_dstq.py -------------------------------------------------------------------------------- /tools/eval_dstq_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/eval_dstq_step.py -------------------------------------------------------------------------------- /tools/eval_dstq_vipseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/eval_dstq_vipseg.py -------------------------------------------------------------------------------- /tools/eval_dvpq_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/eval_dvpq_step.py -------------------------------------------------------------------------------- /tools/eval_dvpq_vipseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/eval_dvpq_vipseg.py -------------------------------------------------------------------------------- /tools/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/flops_counter.py -------------------------------------------------------------------------------- /tools/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/get_flops.py -------------------------------------------------------------------------------- /tools/inference_kitti_step.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/inference_kitti_step.sh -------------------------------------------------------------------------------- /tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/slurm_test.sh -------------------------------------------------------------------------------- /tools/slurm_test_dvps.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/slurm_test_dvps.sh -------------------------------------------------------------------------------- /tools/slurm_test_step.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/slurm_test_step.sh -------------------------------------------------------------------------------- /tools/slurm_test_vis.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/slurm_test_vis.sh -------------------------------------------------------------------------------- /tools/slurm_test_vps.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/slurm_test_vps.sh -------------------------------------------------------------------------------- /tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/slurm_train.sh -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/test_dvps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/test_dvps.py -------------------------------------------------------------------------------- /tools/test_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/test_step.py -------------------------------------------------------------------------------- /tools/test_vps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/test_vps.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/train.py -------------------------------------------------------------------------------- /tools/utils/DSTQ.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/utils/DSTQ.py -------------------------------------------------------------------------------- /tools/utils/STQ.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/utils/STQ.py -------------------------------------------------------------------------------- /tools/utils/cityscapesvps_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/utils/cityscapesvps_eval.py -------------------------------------------------------------------------------- /tools/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools/visualization.py -------------------------------------------------------------------------------- /tools_vis/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools_vis/apis/__init__.py -------------------------------------------------------------------------------- /tools_vis/apis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools_vis/apis/test.py -------------------------------------------------------------------------------- /tools_vis/dist_test_whole_video.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools_vis/dist_test_whole_video.sh -------------------------------------------------------------------------------- /tools_vis/docker.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools_vis/docker.sh -------------------------------------------------------------------------------- /tools_vis/slurm_test_vis.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools_vis/slurm_test_vis.sh -------------------------------------------------------------------------------- /tools_vis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools_vis/test.py -------------------------------------------------------------------------------- /tools_vis/test_whole_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/tools_vis/test_whole_video.py -------------------------------------------------------------------------------- /unitrack/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import * -------------------------------------------------------------------------------- /unitrack/basetrack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/basetrack.py -------------------------------------------------------------------------------- /unitrack/box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/box.py -------------------------------------------------------------------------------- /unitrack/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /unitrack/core/association/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /unitrack/core/association/matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/core/association/matching.py -------------------------------------------------------------------------------- /unitrack/core/motion/kalman_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/core/motion/kalman_filter.py -------------------------------------------------------------------------------- /unitrack/core/propagation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/core/propagation/__init__.py -------------------------------------------------------------------------------- /unitrack/core/propagation/propagate_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/core/propagation/propagate_box.py -------------------------------------------------------------------------------- /unitrack/core/propagation/propagate_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/core/propagation/propagate_mask.py -------------------------------------------------------------------------------- /unitrack/core/propagation/propagate_pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/core/propagation/propagate_pose.py -------------------------------------------------------------------------------- /unitrack/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/mask.py -------------------------------------------------------------------------------- /unitrack/mask_with_train_embs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/mask_with_train_embs.py -------------------------------------------------------------------------------- /unitrack/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/model/__init__.py -------------------------------------------------------------------------------- /unitrack/model/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/model/functional.py -------------------------------------------------------------------------------- /unitrack/model/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/model/hrnet.py -------------------------------------------------------------------------------- /unitrack/model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/model/model.py -------------------------------------------------------------------------------- /unitrack/model/random_feat_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/model/random_feat_generator.py -------------------------------------------------------------------------------- /unitrack/model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/model/resnet.py -------------------------------------------------------------------------------- /unitrack/multitracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/multitracker.py -------------------------------------------------------------------------------- /unitrack/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/utils/__init__.py -------------------------------------------------------------------------------- /unitrack/utils/box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/utils/box.py -------------------------------------------------------------------------------- /unitrack/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/utils/io.py -------------------------------------------------------------------------------- /unitrack/utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/utils/log.py -------------------------------------------------------------------------------- /unitrack/utils/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/utils/mask.py -------------------------------------------------------------------------------- /unitrack/utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/utils/meter.py -------------------------------------------------------------------------------- /unitrack/utils/palette.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/utils/palette.py -------------------------------------------------------------------------------- /unitrack/utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lxtGH/Video-K-Net/HEAD/unitrack/utils/visualize.py --------------------------------------------------------------------------------