├── .dockerignore ├── .gitattributes ├── .gitignore ├── DAv2 ├── DA-2K.md ├── LICENSE ├── README.md ├── app.py ├── assets │ └── examples │ │ ├── demo01.jpg │ │ ├── demo02.jpg │ │ ├── demo03.jpg │ │ ├── demo04.jpg │ │ ├── demo05.jpg │ │ ├── demo06.jpg │ │ ├── demo07.jpg │ │ ├── demo08.jpg │ │ ├── demo09.jpg │ │ ├── demo10.jpg │ │ ├── demo11.jpg │ │ ├── demo12.jpg │ │ ├── demo13.jpg │ │ ├── demo14.jpg │ │ ├── demo15.jpg │ │ ├── demo16.jpg │ │ ├── demo17.jpg │ │ ├── demo18.jpg │ │ ├── demo19.jpg │ │ └── demo20.jpg ├── depth_anything_v2 │ ├── dinov2.py │ ├── dinov2_layers │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── block.py │ │ ├── drop_path.py │ │ ├── layer_scale.py │ │ ├── mlp.py │ │ ├── patch_embed.py │ │ └── swiglu_ffn.py │ ├── dpt.py │ └── util │ │ ├── blocks.py │ │ └── transform.py ├── metric_depth │ ├── README.md │ ├── dataset │ │ ├── hypersim.py │ │ ├── kitti.py │ │ ├── splits │ │ │ ├── hypersim │ │ │ │ └── val.txt │ │ │ ├── kitti │ │ │ │ └── val.txt │ │ │ └── vkitti2 │ │ │ │ └── train.txt │ │ ├── transform.py │ │ └── vkitti2.py │ ├── depth_anything_v2 │ │ ├── dinov2.py │ │ ├── dinov2_layers │ │ │ ├── __init__.py │ │ │ ├── attention.py │ │ │ ├── block.py │ │ │ ├── drop_path.py │ │ │ ├── layer_scale.py │ │ │ ├── mlp.py │ │ │ ├── patch_embed.py │ │ │ └── swiglu_ffn.py │ │ ├── dpt.py │ │ └── util │ │ │ ├── blocks.py │ │ │ └── transform.py │ ├── depth_to_pointcloud.py │ ├── dist_train.sh │ ├── requirements.txt │ ├── run.py │ ├── train.py │ └── util │ │ ├── dist_helper.py │ │ ├── loss.py │ │ ├── metric.py │ │ └── utils.py ├── requirements.txt ├── run.py └── run_video.py ├── LICENSE.md ├── README.md ├── aios_repo ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── build_helper.py ├── config │ ├── __init__.py │ ├── aios_smplx.py │ ├── aios_smplx_agora_val.py │ ├── aios_smplx_bedlam.py │ ├── aios_smplx_controlnet.py │ ├── aios_smplx_demo.py │ ├── aios_smplx_inference.py │ ├── aios_smplx_pretrain.py │ ├── aios_smplx_train.py │ ├── config.py │ ├── test_agora.py │ ├── test_arctic.py │ ├── test_egobody.py │ ├── test_ehf.py │ └── test_ubody.py ├── datasets │ ├── AGORA_MM.py │ ├── ARCTIC.py │ ├── BEDLAM.py │ ├── COCO_NA.py │ ├── EHF.py │ ├── EgoBody_Egocentric.py │ ├── INFERENCE.py │ ├── INFERENCE_AGORA.py │ ├── INFERENCE_BEDLAM.py │ ├── INFERENCE_demo.py │ ├── UBody_MM.py │ ├── __init__.py │ ├── dataset.py │ └── humandata.py ├── detrsmpl │ ├── __init__.py │ ├── apis │ │ ├── __init__.py │ │ ├── inference.py │ │ ├── test.py │ │ └── train.py │ ├── configs │ │ └── _base_ │ │ │ ├── datasets │ │ │ └── pw3d.py │ │ │ ├── default_runtime.py │ │ │ ├── models │ │ │ └── resnet50.py │ │ │ ├── post_processing │ │ │ ├── README.md │ │ │ ├── deciwatch_interval10_q1.py │ │ │ ├── deciwatch_interval10_q2.py │ │ │ ├── deciwatch_interval10_q3.py │ │ │ ├── deciwatch_interval10_q4.py │ │ │ ├── deciwatch_interval10_q5.py │ │ │ ├── deciwatch_interval5_q1.py │ │ │ ├── deciwatch_interval5_q2.py │ │ │ ├── deciwatch_interval5_q3.py │ │ │ ├── deciwatch_interval5_q4.py │ │ │ ├── deciwatch_interval5_q5.py │ │ │ ├── gaus1d.py │ │ │ ├── oneeuro.py │ │ │ ├── savgol.py │ │ │ ├── smoothnet_windowsize16.py │ │ │ ├── smoothnet_windowsize32.py │ │ │ ├── smoothnet_windowsize64.py │ │ │ └── smoothnet_windowsize8.py │ │ │ └── schedulers │ │ │ └── hmr_bs256.py │ ├── core │ │ ├── __init__.py │ │ ├── cameras │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ ├── camera_parameters.py │ │ │ └── cameras.py │ │ ├── conventions │ │ │ ├── __init__.py │ │ │ ├── cameras │ │ │ │ ├── __init__.py │ │ │ │ ├── convert_convention.py │ │ │ │ └── convert_projection.py │ │ │ ├── joints_mapping │ │ │ │ ├── __init__.py │ │ │ │ └── standard_joint_angles.py │ │ │ ├── keypoints_mapping │ │ │ │ ├── __init__.py │ │ │ │ ├── agora.py │ │ │ │ ├── coco.py │ │ │ │ ├── coco_wholebody.py │ │ │ │ ├── crowdpose.py │ │ │ │ ├── face3d.py │ │ │ │ ├── flame.py │ │ │ │ ├── gta.py │ │ │ │ ├── h36m.py │ │ │ │ ├── human_data.py │ │ │ │ ├── hybrik.py │ │ │ │ ├── instavariety.py │ │ │ │ ├── lsp.py │ │ │ │ ├── mano.py │ │ │ │ ├── mpi_inf_3dhp.py │ │ │ │ ├── mpii.py │ │ │ │ ├── openpose.py │ │ │ │ ├── penn_action.py │ │ │ │ ├── posetrack.py │ │ │ │ ├── pw3d.py │ │ │ │ ├── smpl.py │ │ │ │ ├── smplx.py │ │ │ │ ├── spin_smplx.py │ │ │ │ └── star.py │ │ │ └── segmentation │ │ │ │ ├── __init__.py │ │ │ │ ├── smpl.py │ │ │ │ └── smplx.py │ │ ├── distributed_wrapper.py │ │ ├── evaluation │ │ │ ├── __init__.py │ │ │ ├── eval_hooks.py │ │ │ ├── eval_utils.py │ │ │ └── mesh_eval.py │ │ ├── optimizer │ │ │ ├── __init__.py │ │ │ └── builder.py │ │ ├── post_processing │ │ │ ├── __init__.py │ │ │ ├── bbox │ │ │ │ ├── __init__.py │ │ │ │ ├── assigners │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── assign_result.py │ │ │ │ │ ├── base_assigner.py │ │ │ │ │ ├── builder.py │ │ │ │ │ └── hungarian_assigner.py │ │ │ │ ├── coder │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_bbox_coder.py │ │ │ │ │ ├── builder.py │ │ │ │ │ └── distance_point_bbox_coder.py │ │ │ │ ├── match_costs │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── builder.py │ │ │ │ │ └── match_cost.py │ │ │ │ ├── samplers │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_sampler.py │ │ │ │ │ ├── builder.py │ │ │ │ │ ├── pseudo_sampler.py │ │ │ │ │ └── sampling_result.py │ │ │ │ └── transforms.py │ │ │ ├── builder.py │ │ │ ├── smooth │ │ │ │ ├── __init__.py │ │ │ │ ├── gaus1d_filter.py │ │ │ │ ├── oneeuro_filter.py │ │ │ │ ├── savgol_filter.py │ │ │ │ └── smoothnet.py │ │ │ └── speed_up │ │ │ │ ├── __init__.py │ │ │ │ └── deciwatch.py │ │ ├── renderer │ │ │ ├── __init__.py │ │ │ ├── matplotlib3d_renderer.py │ │ │ ├── mpr_renderer │ │ │ │ ├── __init__.py │ │ │ │ ├── camera.py │ │ │ │ ├── cuda │ │ │ │ │ ├── rasterizer.cpp │ │ │ │ │ └── rasterizer_kernel.cu │ │ │ │ ├── rasterizer.py │ │ │ │ ├── smpl_realrender.py │ │ │ │ └── utils.py │ │ │ ├── torch3d_renderer │ │ │ │ ├── __init__.py │ │ │ │ ├── base_renderer.py │ │ │ │ ├── builder.py │ │ │ │ ├── depth_renderer.py │ │ │ │ ├── lights │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── builder.py │ │ │ │ │ └── lights.py │ │ │ │ ├── mesh_renderer.py │ │ │ │ ├── meshes.py │ │ │ │ ├── normal_renderer.py │ │ │ │ ├── pointcloud_renderer.py │ │ │ │ ├── render_runner.py │ │ │ │ ├── render_smpl_config.py │ │ │ │ ├── segmentation_renderer.py │ │ │ │ ├── shader │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── builder.py │ │ │ │ │ └── shader.py │ │ │ │ ├── silhouette_renderer.py │ │ │ │ ├── smpl_renderer.py │ │ │ │ ├── textures │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── builder.py │ │ │ │ │ └── textures.py │ │ │ │ ├── utils.py │ │ │ │ └── uv_renderer.py │ │ │ └── vedo_render.py │ │ └── visualization │ │ │ ├── __init__.py │ │ │ ├── visualize_cameras.py │ │ │ ├── visualize_keypoints2d.py │ │ │ ├── visualize_keypoints3d.py │ │ │ └── visualize_smpl.py │ ├── data │ │ ├── __init__.py │ │ ├── data_structures │ │ │ ├── __init__.py │ │ │ ├── human_data.py │ │ │ ├── human_data_cache.py │ │ │ ├── multi_human_data.py │ │ │ └── smc_reader.py │ │ └── datasets │ │ │ ├── __init__.py │ │ │ ├── adversarial_dataset.py │ │ │ ├── base_dataset.py │ │ │ ├── builder.py │ │ │ ├── dataset_wrappers.py │ │ │ ├── human_hybrik_dataset.py │ │ │ ├── human_image_dataset.py │ │ │ ├── human_image_smplx_dataset.py │ │ │ ├── human_video_dataset.py │ │ │ ├── mesh_dataset.py │ │ │ ├── mixed_dataset.py │ │ │ ├── multi_human_image_dataset.py │ │ │ ├── pipelines │ │ │ ├── __init__.py │ │ │ ├── compose.py │ │ │ ├── formatting.py │ │ │ ├── hybrik_transforms.py │ │ │ ├── loading.py │ │ │ ├── synthetic_occlusion_augmentation.py │ │ │ └── transforms.py │ │ │ └── samplers │ │ │ ├── __init__.py │ │ │ └── distributed_sampler.py │ ├── models │ │ ├── __init__.py │ │ ├── architectures │ │ │ ├── DetrSMPL.py │ │ │ ├── DetrSMPLloss.py │ │ │ ├── __init__.py │ │ │ ├── base_architecture.py │ │ │ ├── builder.py │ │ │ ├── expressive_mesh_estimator.py │ │ │ ├── hybrik.py │ │ │ └── mesh_estimator.py │ │ ├── backbones │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ ├── hrnet.py │ │ │ └── resnet.py │ │ ├── body_models │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ ├── flame.py │ │ │ ├── mano.py │ │ │ ├── smpl.py │ │ │ ├── smplx.py │ │ │ ├── star.py │ │ │ └── utils.py │ │ ├── discriminators │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ └── pose_discriminator.py │ │ ├── heads │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ ├── detr_head.py │ │ │ ├── expose_head.py │ │ │ ├── hmr_head.py │ │ │ ├── hybrik_head.py │ │ │ └── pare_head.py │ │ ├── losses │ │ │ ├── __init__.py │ │ │ ├── balanced_mse_loss.py │ │ │ ├── builder.py │ │ │ ├── cross_entropy_loss.py │ │ │ ├── focal_loss.py │ │ │ ├── gan_loss.py │ │ │ ├── iou_loss.py │ │ │ ├── mse_loss.py │ │ │ ├── prior_loss.py │ │ │ ├── rotaion_distance_loss.py │ │ │ ├── smooth_l1_loss.py │ │ │ └── utils.py │ │ ├── necks │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ ├── channel_mapper.py │ │ │ └── temporal_encoder.py │ │ ├── registrants │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ ├── smplify.py │ │ │ └── smplifyx.py │ │ └── utils │ │ │ ├── SMPLX.py │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ ├── fits_dict.py │ │ │ ├── inverse_kinematics.py │ │ │ ├── positional_encoding.py │ │ │ ├── res_layer.py │ │ │ └── transformer.py │ ├── utils │ │ ├── __init__.py │ │ ├── camera_utils.py │ │ ├── collect_env.py │ │ ├── demo_utils.py │ │ ├── dist_utils.py │ │ ├── ffmpeg_utils.py │ │ ├── geometry.py │ │ ├── keypoint_utils.py │ │ ├── logger.py │ │ ├── mesh_utils.py │ │ ├── misc.py │ │ ├── path_utils.py │ │ ├── transforms.py │ │ └── util_mixins.py │ └── version.py ├── engine.py ├── main.py ├── models │ ├── __init__.py │ ├── aios │ │ ├── __init__.py │ │ ├── aios_smplx.py │ │ ├── aios_smplx_controlnet.py │ │ ├── backbones │ │ │ ├── __init__.py │ │ │ ├── backbone.py │ │ │ └── swin_transformer.py │ │ ├── criterion_smplx.py │ │ ├── matcher.py │ │ ├── ops │ │ │ ├── functions │ │ │ │ ├── __init__.py │ │ │ │ └── ms_deform_attn_func.py │ │ │ ├── make.sh │ │ │ ├── modules │ │ │ │ ├── __init__.py │ │ │ │ ├── ms_deform_attn.py │ │ │ │ └── ms_deform_attn_key_aware.py │ │ │ ├── setup.py │ │ │ ├── src │ │ │ │ ├── cpu │ │ │ │ │ ├── ms_deform_attn_cpu.cpp │ │ │ │ │ └── ms_deform_attn_cpu.h │ │ │ │ ├── cuda │ │ │ │ │ ├── ms_deform_attn_cuda.cu │ │ │ │ │ ├── ms_deform_attn_cuda.h │ │ │ │ │ └── ms_deform_im2col_cuda.cuh │ │ │ │ ├── ms_deform_attn.h │ │ │ │ └── vision.cpp │ │ │ └── test.py │ │ ├── position_encoding.py │ │ ├── postprocesses.py │ │ ├── transformer.py │ │ ├── transformer_deformable.py │ │ └── utils.py │ ├── registry.py │ └── utils.py ├── pyrightconfig.json ├── requirements.txt ├── scripts │ ├── inference.sh │ ├── test_agora.sh │ ├── test_agora_val.sh │ ├── test_bedlam.sh │ └── train.sh └── util │ ├── __init__.py │ ├── box_loss.py │ ├── box_ops.py │ ├── config.py │ ├── dir.py │ ├── distribute_utils.py │ ├── formatting.py │ ├── get_param_dicts.py │ ├── human_models.py │ ├── keypoint_ops.py │ ├── logger.py │ ├── misc.py │ ├── plot_utils.py │ ├── preprocessing.py │ ├── smplx │ ├── README.md │ ├── examples │ │ ├── demo.py │ │ ├── demo_layers.py │ │ ├── vis_flame_vertices.py │ │ └── vis_mano_vertices.py │ ├── setup.py │ ├── smplx │ │ ├── __init__.py │ │ ├── body_models.py │ │ ├── joint_names.py │ │ ├── lbs.py │ │ ├── utils.py │ │ ├── vertex_ids.py │ │ └── vertex_joint_selector.py │ └── tools │ │ ├── README.md │ │ ├── __init__.py │ │ ├── clean_ch.py │ │ └── merge_smplh_mano.py │ ├── time_counter.py │ ├── transforms.py │ ├── utils.py │ └── vis.py ├── api ├── BLADE_API.py └── test_api.py ├── assets ├── demo_results.png ├── pipeline.png └── teaser_blade.png ├── blade ├── __init__.py ├── cameras │ ├── __init__.py │ ├── builder.py │ ├── convert_convention.py │ ├── convert_projection.py │ ├── pytorch3d_wrapper │ │ ├── builder.py │ │ └── cameras.py │ └── utils.py ├── configs │ ├── base.py │ ├── blade_depthnet.py │ ├── blade_inthewild.py │ └── blade_posenet.py ├── custom_hooks │ ├── __init__.py │ └── wandb_hook.py ├── datasets │ ├── __init__.py │ ├── batch_run_preprocess_ourdata.sh │ ├── batch_run_preprocess_smpl2smplx.sh │ ├── builder.py │ ├── combine_preprocessed_ourdata.py │ ├── combine_preprocessed_smpldata.py │ ├── gender_lookup.py │ ├── human_image_dataset.py │ ├── human_image_dataset_smplx.py │ ├── human_image_dataset_tz.py │ ├── image_dataset.py │ ├── itw_dataset.py │ ├── our_dataset.py │ ├── our_dataset_smplx.py │ ├── pipelines │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── compose.py │ │ ├── loading.py │ │ └── transforms.py │ ├── preprocess_ourdata_chunked.py │ ├── preprocess_smpldata_chunked.py │ └── utils.py ├── models │ ├── __init__.py │ ├── architectures │ │ ├── blade.py │ │ ├── blade_helper.py │ │ ├── blade_imports.py │ │ └── builder.py │ ├── backbones │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── hrnet.py │ │ ├── mlp.py │ │ └── resnet.py │ ├── bert │ │ ├── graphormer.py │ │ └── mlp_mixer.py │ ├── body_models │ │ ├── __init__.py │ │ ├── base.py │ │ ├── builder.py │ │ ├── flame.py │ │ ├── mano.py │ │ ├── mappings │ │ │ ├── __init__.py │ │ │ ├── coco.py │ │ │ ├── coco_hand.py │ │ │ ├── flame.py │ │ │ ├── h36m.py │ │ │ ├── mano.py │ │ │ ├── mmpose_hand.py │ │ │ ├── people_snapshot.py │ │ │ ├── segmentation │ │ │ │ ├── __init__.py │ │ │ │ └── smpl_1723.py │ │ │ ├── smpl.py │ │ │ ├── smplh.py │ │ │ └── smplx.py │ │ ├── smpl.py │ │ ├── smpl_d.py │ │ ├── smpl_x.py │ │ ├── smplh.py │ │ └── utils.py │ ├── heads │ │ ├── builder.py │ │ └── depthonly_head.py │ ├── layers │ │ ├── conv.py │ │ ├── fpn.py │ │ ├── gcn_lift.py │ │ ├── graph_cnn.py │ │ ├── mlp_lift.py │ │ ├── realnvp.py │ │ └── stgcn_lift.py │ ├── losses │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── cross_entropy_loss.py │ │ ├── mesh_loss.py │ │ ├── mse_loss.py │ │ ├── norm_loss.py │ │ ├── prior_loss.py │ │ ├── regression_loss.py │ │ ├── render_loss.py │ │ ├── smooth_loss.py │ │ └── utils.py │ ├── necks │ │ └── builder.py │ ├── transformer │ │ ├── __init__.py │ │ ├── position_encoding.py │ │ └── transformer.py │ ├── utils │ │ ├── __init__.py │ │ ├── basic_block.py │ │ ├── embed.py │ │ ├── encoding.py │ │ ├── inverted_residual.py │ │ ├── make_divisible.py │ │ ├── point_sample.py │ │ ├── ppm.py │ │ ├── res_layer.py │ │ ├── san_layers.py │ │ ├── se_layer.py │ │ ├── self_attention_block.py │ │ ├── shape_convert.py │ │ ├── up_conv_block.py │ │ └── wrappers.py │ └── visualizers │ │ ├── builder.py │ │ └── smpl_visualizer.py ├── render │ ├── __init__.py │ ├── builder.py │ ├── explicit │ │ ├── __init__.py │ │ ├── opengl_wrapper │ │ │ └── __init__.py │ │ ├── pyrender_wrapper │ │ │ └── __init__.py │ │ └── pytorch3d_wrapper │ │ │ ├── __init__.py │ │ │ ├── lights │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ └── lights.py │ │ │ ├── render_runner.py │ │ │ ├── renderers │ │ │ ├── base_renderer.py │ │ │ ├── builder.py │ │ │ ├── depth_renderer.py │ │ │ ├── mesh_renderer.py │ │ │ ├── normal_renderer.py │ │ │ ├── opticalflow_renderer.py │ │ │ ├── pointcloud_renderer.py │ │ │ ├── segmentation_renderer.py │ │ │ ├── silhouette_renderer.py │ │ │ ├── smpl_renderer.py │ │ │ ├── utils.py │ │ │ └── uv_renderer.py │ │ │ ├── shaders │ │ │ ├── builder.py │ │ │ └── shader.py │ │ │ └── textures │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ └── textures.py │ └── implicit │ │ └── __init__.py ├── structures │ └── meshes │ │ ├── meshes.py │ │ └── utils.py ├── transforms │ ├── __init__.py │ ├── transform2d │ │ ├── __init__.py │ │ ├── affine.py │ │ ├── perspective.py │ │ ├── thin_plate_spline.py │ │ └── transform2d.py │ └── transform3d │ │ ├── __init__.py │ │ ├── convert_rotation.py │ │ └── geometry.py └── utils │ ├── __init__.py │ ├── bbox_utils.py │ ├── cylindar.py │ ├── demo_utils.py │ ├── frame_utils.py │ ├── ftxty_solve.py │ ├── helpers.py │ ├── keypoint_utils.py │ ├── mesh_utils.py │ ├── torch_utils.py │ ├── visualize_keypoints2d.py │ ├── visualize_smpl.py │ └── visualize_voxel.py ├── body_models ├── smpl │ └── .gitkeep └── smplx │ └── .gitkeep ├── cache └── .gitkeep ├── demo.sh ├── demo_images ├── 000012.jpg ├── 5682889770_b707c270cf_b.jpg ├── pexels-alejandro-reyes-1970474111-31445232.jpg ├── pexels-antonio-prado-1050855-3250746.jpg ├── pexels-dauphotographer-31452085.jpg ├── pexels-davidgari-11961763.jpg ├── pexels-jean-daniel-12337963.jpg ├── pexels-jean-daniel-21544681.jpg ├── pexels-jean-daniel-21544687.jpg ├── pexels-prime-cinematics-1005175-2057477.jpg ├── pexels-prime-cinematics-1005175-2057675.jpg ├── pexels-tkirkgoz-11481204.jpg ├── pexels-wendelmoretti-1777479.jpg ├── pexels-wesley-souza-1045799-2027492.jpg └── pexels-zaid-mohammed-86842527-19862946.jpg ├── docs ├── BEDLAMCC_GENERATION.md ├── DATASETS.md ├── INSTALL_CUDA13.md ├── MULTI_NODE_TRAINING.md └── bedlamcc_assets.zip ├── mmcv ├── .circleci │ └── config.yml ├── .dev_scripts │ ├── check_installation.py │ └── visualize_lr.py ├── .dockerignore ├── .github │ ├── ISSUE_TEMPLATE │ │ ├── config.yml │ │ ├── feature_request.md │ │ ├── general_questions.md │ │ └── unexpected_report.md │ ├── pull_request_template.md │ └── workflows │ │ ├── build.yml │ │ ├── build_pat.yml │ │ ├── lint.yml │ │ └── publish-to-pypi.yml ├── .gitignore ├── .owners.yml ├── .pre-commit-config.yaml ├── .readthedocs.yml ├── CITATION.cff ├── CONTRIBUTING.md ├── Dockerfile ├── Jenkinsfile ├── LICENSE ├── LICENSES.md ├── MANIFEST.in ├── README.md ├── README_zh-CN.md ├── TERMINOLOGY.md ├── docs │ ├── en │ │ ├── Makefile │ │ ├── _static │ │ │ ├── css │ │ │ │ └── readthedocs.css │ │ │ ├── qq_group_qrcode.jpg │ │ │ ├── wechat_qrcode.jpg │ │ │ └── zhihu_qrcode.jpg │ │ ├── api.rst │ │ ├── community │ │ │ ├── contributing.md │ │ │ └── pr.md │ │ ├── compatibility.md │ │ ├── conf.py │ │ ├── deployment │ │ │ ├── mmcv_ops_definition.md │ │ │ ├── onnx.md │ │ │ ├── onnxruntime_custom_ops.md │ │ │ ├── onnxruntime_op.md │ │ │ ├── tensorrt_custom_ops.md │ │ │ └── tensorrt_plugin.md │ │ ├── faq.md │ │ ├── get_started │ │ │ ├── build.md │ │ │ ├── installation.md │ │ │ ├── introduction.md │ │ │ └── previous_versions.md │ │ ├── index.rst │ │ ├── make.bat │ │ └── understand_mmcv │ │ │ ├── cnn.md │ │ │ ├── config.md │ │ │ ├── data_process.md │ │ │ ├── io.md │ │ │ ├── ops.md │ │ │ ├── registry.md │ │ │ ├── runner.md │ │ │ ├── utils.md │ │ │ └── visualization.md │ └── zh_cn │ │ ├── Makefile │ │ ├── _static │ │ └── css │ │ │ └── readthedocs.css │ │ ├── api.rst │ │ ├── community │ │ ├── contributing.md │ │ └── pr.md │ │ ├── compatibility.md │ │ ├── conf.py │ │ ├── deployment │ │ ├── onnx.md │ │ ├── onnxruntime_custom_ops.md │ │ ├── onnxruntime_op.md │ │ ├── tensorrt_custom_ops.md │ │ └── tensorrt_plugin.md │ │ ├── faq.md │ │ ├── get_started │ │ ├── build.md │ │ ├── installation.md │ │ ├── introduction.md │ │ └── previous_versions.md │ │ ├── index.rst │ │ ├── make.bat │ │ └── understand_mmcv │ │ ├── cnn.md │ │ ├── config.md │ │ ├── data_process.md │ │ ├── io.md │ │ ├── ops.md │ │ ├── registry.md │ │ ├── runner.md │ │ ├── utils.md │ │ └── visualization.md ├── examples │ └── train.py ├── mmcv │ ├── __init__.py │ ├── arraymisc │ │ ├── __init__.py │ │ └── quantization.py │ ├── cnn │ │ ├── __init__.py │ │ ├── alexnet.py │ │ ├── bricks │ │ │ ├── __init__.py │ │ │ ├── activation.py │ │ │ ├── context_block.py │ │ │ ├── conv.py │ │ │ ├── conv2d_adaptive_padding.py │ │ │ ├── conv_module.py │ │ │ ├── conv_ws.py │ │ │ ├── depthwise_separable_conv_module.py │ │ │ ├── drop.py │ │ │ ├── generalized_attention.py │ │ │ ├── hsigmoid.py │ │ │ ├── hswish.py │ │ │ ├── non_local.py │ │ │ ├── norm.py │ │ │ ├── padding.py │ │ │ ├── plugin.py │ │ │ ├── registry.py │ │ │ ├── scale.py │ │ │ ├── swish.py │ │ │ ├── transformer.py │ │ │ ├── upsample.py │ │ │ └── wrappers.py │ │ ├── builder.py │ │ ├── resnet.py │ │ ├── utils │ │ │ ├── __init__.py │ │ │ ├── flops_counter.py │ │ │ ├── fuse_conv_bn.py │ │ │ ├── sync_bn.py │ │ │ └── weight_init.py │ │ └── vgg.py │ ├── device │ │ ├── __init__.py │ │ ├── ipu │ │ │ ├── __init__.py │ │ │ ├── dataloader.py │ │ │ ├── hierarchical_data_manager.py │ │ │ ├── hook_wrapper.py │ │ │ ├── model_wrapper.py │ │ │ ├── runner.py │ │ │ └── utils.py │ │ └── mlu │ │ │ ├── __init__.py │ │ │ ├── _functions.py │ │ │ ├── data_parallel.py │ │ │ ├── distributed.py │ │ │ └── scatter_gather.py │ ├── engine │ │ ├── __init__.py │ │ └── test.py │ ├── fileio │ │ ├── __init__.py │ │ ├── file_client.py │ │ ├── handlers │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── json_handler.py │ │ │ ├── pickle_handler.py │ │ │ └── yaml_handler.py │ │ ├── io.py │ │ └── parse.py │ ├── image │ │ ├── __init__.py │ │ ├── colorspace.py │ │ ├── geometric.py │ │ ├── io.py │ │ ├── misc.py │ │ └── photometric.py │ ├── model_zoo │ │ ├── deprecated.json │ │ ├── mmcls.json │ │ ├── open_mmlab.json │ │ └── torchvision_0.12.json │ ├── onnx │ │ ├── __init__.py │ │ ├── info.py │ │ ├── onnx_utils │ │ │ ├── __init__.py │ │ │ └── symbolic_helper.py │ │ └── symbolic.py │ ├── ops │ │ ├── __init__.py │ │ ├── active_rotated_filter.py │ │ ├── assign_score_withk.py │ │ ├── ball_query.py │ │ ├── bbox.py │ │ ├── border_align.py │ │ ├── box_iou_rotated.py │ │ ├── carafe.py │ │ ├── cc_attention.py │ │ ├── chamfer_distance.py │ │ ├── contour_expand.py │ │ ├── convex_iou.py │ │ ├── corner_pool.py │ │ ├── correlation.py │ │ ├── csrc │ │ │ ├── README.md │ │ │ ├── common │ │ │ │ ├── box_iou_rotated_utils.hpp │ │ │ │ ├── cuda │ │ │ │ │ ├── active_rotated_filter_cuda_kernel.cuh │ │ │ │ │ ├── assign_score_withk_cuda_kernel.cuh │ │ │ │ │ ├── ball_query_cuda_kernel.cuh │ │ │ │ │ ├── bbox_overlaps_cuda_kernel.cuh │ │ │ │ │ ├── border_align_cuda_kernel.cuh │ │ │ │ │ ├── box_iou_rotated_cuda.cuh │ │ │ │ │ ├── carafe_cuda_kernel.cuh │ │ │ │ │ ├── carafe_naive_cuda_kernel.cuh │ │ │ │ │ ├── chamfer_distance_cuda_kernel.cuh │ │ │ │ │ ├── common_cuda_helper.hpp │ │ │ │ │ ├── convex_iou_cuda_kernel.cuh │ │ │ │ │ ├── correlation_cuda.cuh │ │ │ │ │ ├── deform_conv_cuda_kernel.cuh │ │ │ │ │ ├── deform_roi_pool_cuda_kernel.cuh │ │ │ │ │ ├── diff_iou_rotated_cuda_kernel.cuh │ │ │ │ │ ├── furthest_point_sample_cuda_kernel.cuh │ │ │ │ │ ├── gather_points_cuda_kernel.cuh │ │ │ │ │ ├── group_points_cuda_kernel.cuh │ │ │ │ │ ├── iou3d_cuda_kernel.cuh │ │ │ │ │ ├── knn_cuda_kernel.cuh │ │ │ │ │ ├── masked_conv2d_cuda_kernel.cuh │ │ │ │ │ ├── min_area_polygons_cuda.cuh │ │ │ │ │ ├── modulated_deform_conv_cuda_kernel.cuh │ │ │ │ │ ├── ms_deform_attn_cuda_kernel.cuh │ │ │ │ │ ├── nms_cuda_kernel.cuh │ │ │ │ │ ├── nms_rotated_cuda.cuh │ │ │ │ │ ├── parrots_cudawarpfunction.cuh │ │ │ │ │ ├── points_in_boxes_cuda_kernel.cuh │ │ │ │ │ ├── points_in_polygons_cuda_kernel.cuh │ │ │ │ │ ├── psamask_cuda_kernel.cuh │ │ │ │ │ ├── riroi_align_rotated_cuda_kernel.cuh │ │ │ │ │ ├── roi_align_cuda_kernel.cuh │ │ │ │ │ ├── roi_align_rotated_cuda_kernel.cuh │ │ │ │ │ ├── roi_pool_cuda_kernel.cuh │ │ │ │ │ ├── roiaware_pool3d_cuda_kernel.cuh │ │ │ │ │ ├── roipoint_pool3d_cuda_kernel.cuh │ │ │ │ │ ├── rotated_feature_align_cuda_kernel.cuh │ │ │ │ │ ├── scatter_points_cuda_kernel.cuh │ │ │ │ │ ├── sigmoid_focal_loss_cuda_kernel.cuh │ │ │ │ │ ├── softmax_focal_loss_cuda_kernel.cuh │ │ │ │ │ ├── spconv │ │ │ │ │ │ ├── indice.cuh │ │ │ │ │ │ └── reordering.cuh │ │ │ │ │ ├── sync_bn_cuda_kernel.cuh │ │ │ │ │ ├── three_interpolate_cuda_kernel.cuh │ │ │ │ │ ├── three_nn_cuda_kernel.cuh │ │ │ │ │ ├── tin_shift_cuda_kernel.cuh │ │ │ │ │ └── voxelization_cuda_kernel.cuh │ │ │ │ ├── mlu │ │ │ │ │ ├── bbox_overlaps_mlu_kernel.mlu │ │ │ │ │ ├── common_mlu_helper.hpp │ │ │ │ │ ├── focal_loss_sigmoid_mlu_kernel.mlu │ │ │ │ │ ├── nms_mlu_kernel.mlu │ │ │ │ │ ├── psamask_mlu_kernel.mlu │ │ │ │ │ ├── psamask_utils.hpp │ │ │ │ │ ├── roi_align_mlu_kernel.mlu │ │ │ │ │ ├── roi_align_rotated_mlu_kernel.mlu │ │ │ │ │ ├── roi_align_rotated_utils.hpp │ │ │ │ │ └── tin_shift_mlu_kernel.mlu │ │ │ │ ├── parrots_cpp_helper.hpp │ │ │ │ ├── parrots_cuda_helper.hpp │ │ │ │ ├── pytorch_cpp_helper.hpp │ │ │ │ ├── pytorch_cuda_helper.hpp │ │ │ │ ├── pytorch_device_registry.hpp │ │ │ │ ├── pytorch_mlu_helper.hpp │ │ │ │ └── utils │ │ │ │ │ └── spconv │ │ │ │ │ ├── paramsgrid.h │ │ │ │ │ ├── prettyprint.h │ │ │ │ │ ├── pybind11_utils.h │ │ │ │ │ ├── spconv │ │ │ │ │ ├── geometry.h │ │ │ │ │ ├── indice.h │ │ │ │ │ ├── maxpool.h │ │ │ │ │ ├── mp_helper.h │ │ │ │ │ ├── point2voxel.h │ │ │ │ │ └── reordering.h │ │ │ │ │ └── tensorview │ │ │ │ │ ├── helper_kernel.cuh │ │ │ │ │ ├── helper_launch.h │ │ │ │ │ └── tensorview.h │ │ │ ├── onnxruntime │ │ │ │ ├── corner_pool.h │ │ │ │ ├── cpu │ │ │ │ │ ├── corner_pool.cpp │ │ │ │ │ ├── deform_conv.cpp │ │ │ │ │ ├── gridSample.cpp │ │ │ │ │ ├── modulated_deform_conv.cpp │ │ │ │ │ ├── nms.cpp │ │ │ │ │ ├── onnxruntime_register.cpp │ │ │ │ │ ├── reduce_ops.cpp │ │ │ │ │ ├── roi_align.cpp │ │ │ │ │ ├── roi_align_rotated.cpp │ │ │ │ │ ├── rotated_feature_align.cpp │ │ │ │ │ └── soft_nms.cpp │ │ │ │ ├── deform_conv.h │ │ │ │ ├── grid_sample.h │ │ │ │ ├── modulated_deform_conv.h │ │ │ │ ├── nms.h │ │ │ │ ├── onnxruntime_register.h │ │ │ │ ├── onnxruntime_session_options_config_keys.h │ │ │ │ ├── ort_mmcv_utils.h │ │ │ │ ├── reduce_ops.h │ │ │ │ ├── roi_align.h │ │ │ │ ├── roi_align_rotated.h │ │ │ │ ├── rotated_feature_align.h │ │ │ │ └── soft_nms.h │ │ │ ├── parrots │ │ │ │ ├── active_rotated_filter.cpp │ │ │ │ ├── active_rotated_filter_parrots.cpp │ │ │ │ ├── active_rotated_filter_pytorch.h │ │ │ │ ├── assign_score_withk.cpp │ │ │ │ ├── assign_score_withk_parrots.cpp │ │ │ │ ├── assign_score_withk_pytorch.h │ │ │ │ ├── ball_query._parrots.cpp │ │ │ │ ├── ball_query.cpp │ │ │ │ ├── ball_query_pytorch.h │ │ │ │ ├── bbox_overlaps.cpp │ │ │ │ ├── bbox_overlaps_parrots.cpp │ │ │ │ ├── bbox_overlaps_pytorch.h │ │ │ │ ├── border_align.cpp │ │ │ │ ├── border_align_parrots.cpp │ │ │ │ ├── border_align_pytorch.h │ │ │ │ ├── box_iou_rotated.cpp │ │ │ │ ├── box_iou_rotated_parrots.cpp │ │ │ │ ├── box_iou_rotated_pytorch.h │ │ │ │ ├── carafe.cpp │ │ │ │ ├── carafe_naive.cpp │ │ │ │ ├── carafe_naive_parrots.cpp │ │ │ │ ├── carafe_naive_pytorch.h │ │ │ │ ├── carafe_parrots.cpp │ │ │ │ ├── carafe_pytorch.h │ │ │ │ ├── contour_expand.cpp │ │ │ │ ├── contour_expand_parrots.cpp │ │ │ │ ├── contour_expand_pytorch.h │ │ │ │ ├── convex_iou.cpp │ │ │ │ ├── convex_iou_parrots.cpp │ │ │ │ ├── convex_iou_pytorch.h │ │ │ │ ├── correlation.cpp │ │ │ │ ├── correlation_parrots.cpp │ │ │ │ ├── correlation_pytorch.h │ │ │ │ ├── cudabind.cpp │ │ │ │ ├── deform_conv.cpp │ │ │ │ ├── deform_conv_parrots.cpp │ │ │ │ ├── deform_conv_pytorch.h │ │ │ │ ├── deform_roi_pool.cpp │ │ │ │ ├── deform_roi_pool_parrots.cpp │ │ │ │ ├── deform_roi_pool_pytorch.h │ │ │ │ ├── diff_iou_rotated.cpp │ │ │ │ ├── diff_iou_rotated_parrots.cpp │ │ │ │ ├── diff_iou_rotated_pytorch.h │ │ │ │ ├── focal_loss.cpp │ │ │ │ ├── focal_loss_parrots.cpp │ │ │ │ ├── focal_loss_pytorch.h │ │ │ │ ├── furthest_point_sample.cpp │ │ │ │ ├── furthest_point_sample_parrots.cpp │ │ │ │ ├── furthest_point_sample_pytorch.h │ │ │ │ ├── fused_bias_leakyrelu.cpp │ │ │ │ ├── fused_bias_parrots.cpp │ │ │ │ ├── gather_points.cpp │ │ │ │ ├── gather_points_parrots.cpp │ │ │ │ ├── gather_points_pytorch.h │ │ │ │ ├── group_points.cpp │ │ │ │ ├── group_points_parrots.cpp │ │ │ │ ├── group_points_pytorch.h │ │ │ │ ├── info.cpp │ │ │ │ ├── iou3d.cpp │ │ │ │ ├── iou3d_parrots.cpp │ │ │ │ ├── iou3d_pytorch.h │ │ │ │ ├── knn.cpp │ │ │ │ ├── knn_parrots.cpp │ │ │ │ ├── knn_pytorch.h │ │ │ │ ├── masked_conv2d.cpp │ │ │ │ ├── masked_conv2d_parrots.cpp │ │ │ │ ├── masked_conv2d_pytorch.h │ │ │ │ ├── min_area_polygons.cpp │ │ │ │ ├── min_area_polygons_parrots.cpp │ │ │ │ ├── min_area_polygons_pytorch.h │ │ │ │ ├── modulated_deform_conv.cpp │ │ │ │ ├── modulated_deform_conv_parrots.cpp │ │ │ │ ├── modulated_deform_conv_pytorch.h │ │ │ │ ├── ms_deform_attn.cpp │ │ │ │ ├── ms_deform_attn_parrots.cpp │ │ │ │ ├── nms.cpp │ │ │ │ ├── nms_parrots.cpp │ │ │ │ ├── nms_pytorch.h │ │ │ │ ├── nms_rotated.cpp │ │ │ │ ├── pixel_group.cpp │ │ │ │ ├── pixel_group_parrots.cpp │ │ │ │ ├── pixel_group_pytorch.h │ │ │ │ ├── points_in_boxes.cpp │ │ │ │ ├── points_in_boxes_parrots.cpp │ │ │ │ ├── points_in_boxes_pytorch.h │ │ │ │ ├── points_in_polygons.cpp │ │ │ │ ├── points_in_polygons_parrots.cpp │ │ │ │ ├── points_in_polygons_pytorch.h │ │ │ │ ├── psamask.cpp │ │ │ │ ├── psamask_parrots.cpp │ │ │ │ ├── psamask_pytorch.h │ │ │ │ ├── riroi_align_rotated.cpp │ │ │ │ ├── riroi_align_rotated_parrots.cpp │ │ │ │ ├── riroi_align_rotated_pytorch.h │ │ │ │ ├── roi_align.cpp │ │ │ │ ├── roi_align_parrots.cpp │ │ │ │ ├── roi_align_pytorch.h │ │ │ │ ├── roi_align_rotated.cpp │ │ │ │ ├── roi_align_rotated_parrots.cpp │ │ │ │ ├── roi_align_rotated_pytorch.h │ │ │ │ ├── roi_pool.cpp │ │ │ │ ├── roi_pool_parrots.cpp │ │ │ │ ├── roi_pool_pytorch.h │ │ │ │ ├── roiaware_pool3d.cpp │ │ │ │ ├── roiaware_pool3d_parrots.cpp │ │ │ │ ├── roiaware_pool3d_pytorch.h │ │ │ │ ├── roipoint_pool3d.cpp │ │ │ │ ├── roipoint_pool3d_parrots.cpp │ │ │ │ ├── roipoint_pool3d_pytorch.h │ │ │ │ ├── rotated_feature_align.cpp │ │ │ │ ├── rotated_feature_align_parrots.cpp │ │ │ │ ├── rotated_feature_align_pytorch.h │ │ │ │ ├── sync_bn.cpp │ │ │ │ ├── sync_bn_parrots.cpp │ │ │ │ ├── sync_bn_pytorch.h │ │ │ │ ├── three_interpolate.cpp │ │ │ │ ├── three_interpolate_parrots.cpp │ │ │ │ ├── three_interpolate_pytorch.h │ │ │ │ ├── three_nn.cpp │ │ │ │ ├── three_nn_parrots.cpp │ │ │ │ ├── three_nn_pytorch.h │ │ │ │ ├── tin_shift.cpp │ │ │ │ ├── tin_shift_parrots.cpp │ │ │ │ ├── tin_shift_pytorch.h │ │ │ │ ├── upfirdn2d.cpp │ │ │ │ ├── upfirdn2d_parrots.cpp │ │ │ │ ├── voxelization.cpp │ │ │ │ ├── voxelization_parrots.cpp │ │ │ │ └── voxelization_pytorch.h │ │ │ ├── pytorch │ │ │ │ ├── active_rotated_filter.cpp │ │ │ │ ├── assign_score_withk.cpp │ │ │ │ ├── ball_query.cpp │ │ │ │ ├── bbox_overlaps.cpp │ │ │ │ ├── border_align.cpp │ │ │ │ ├── box_iou_rotated.cpp │ │ │ │ ├── carafe.cpp │ │ │ │ ├── carafe_naive.cpp │ │ │ │ ├── chamfer_distance.cpp │ │ │ │ ├── contour_expand.cpp │ │ │ │ ├── convex_iou.cpp │ │ │ │ ├── correlation.cpp │ │ │ │ ├── cpu │ │ │ │ │ ├── active_rotated_filter.cpp │ │ │ │ │ ├── box_iou_rotated.cpp │ │ │ │ │ ├── deform_conv.cpp │ │ │ │ │ ├── modulated_deform_conv.cpp │ │ │ │ │ ├── nms.cpp │ │ │ │ │ ├── nms_rotated.cpp │ │ │ │ │ ├── pixel_group.cpp │ │ │ │ │ ├── points_in_boxes.cpp │ │ │ │ │ ├── psamask.cpp │ │ │ │ │ ├── roi_align.cpp │ │ │ │ │ ├── roi_align_rotated.cpp │ │ │ │ │ ├── rotated_feature_align.cpp │ │ │ │ │ ├── sparse_indice.cpp │ │ │ │ │ ├── sparse_maxpool.cpp │ │ │ │ │ ├── sparse_reordering.cpp │ │ │ │ │ └── voxelization.cpp │ │ │ │ ├── cuda │ │ │ │ │ ├── active_rotated_filter_cuda.cu │ │ │ │ │ ├── assign_score_withk_cuda.cu │ │ │ │ │ ├── ball_query_cuda.cu │ │ │ │ │ ├── bbox_overlaps_cuda.cu │ │ │ │ │ ├── border_align_cuda.cu │ │ │ │ │ ├── box_iou_rotated_cuda.cu │ │ │ │ │ ├── carafe_cuda.cu │ │ │ │ │ ├── carafe_naive_cuda.cu │ │ │ │ │ ├── chamfer_distance_cuda.cu │ │ │ │ │ ├── convex_iou.cu │ │ │ │ │ ├── correlation_cuda.cu │ │ │ │ │ ├── cudabind.cpp │ │ │ │ │ ├── deform_conv_cuda.cu │ │ │ │ │ ├── deform_roi_pool_cuda.cu │ │ │ │ │ ├── diff_iou_rotated_cuda.cu │ │ │ │ │ ├── focal_loss_cuda.cu │ │ │ │ │ ├── furthest_point_sample_cuda.cu │ │ │ │ │ ├── fused_bias_leakyrelu_cuda.cu │ │ │ │ │ ├── fused_spconv_ops_cuda.cu │ │ │ │ │ ├── gather_points_cuda.cu │ │ │ │ │ ├── group_points_cuda.cu │ │ │ │ │ ├── iou3d_cuda.cu │ │ │ │ │ ├── knn_cuda.cu │ │ │ │ │ ├── masked_conv2d_cuda.cu │ │ │ │ │ ├── min_area_polygons.cu │ │ │ │ │ ├── modulated_deform_conv_cuda.cu │ │ │ │ │ ├── ms_deform_attn_cuda.cu │ │ │ │ │ ├── nms_cuda.cu │ │ │ │ │ ├── nms_rotated_cuda.cu │ │ │ │ │ ├── points_in_boxes_cuda.cu │ │ │ │ │ ├── points_in_polygons_cuda.cu │ │ │ │ │ ├── psamask_cuda.cu │ │ │ │ │ ├── riroi_align_rotated_cuda.cu │ │ │ │ │ ├── roi_align_cuda.cu │ │ │ │ │ ├── roi_align_rotated_cuda.cu │ │ │ │ │ ├── roi_pool_cuda.cu │ │ │ │ │ ├── roiaware_pool3d_cuda.cu │ │ │ │ │ ├── roipoint_pool3d_cuda.cu │ │ │ │ │ ├── rotated_feature_align_cuda.cu │ │ │ │ │ ├── scatter_points_cuda.cu │ │ │ │ │ ├── sparse_indice.cu │ │ │ │ │ ├── sparse_maxpool.cu │ │ │ │ │ ├── sparse_pool_ops_cuda.cu │ │ │ │ │ ├── sparse_reordering.cu │ │ │ │ │ ├── spconv_ops_cuda.cu │ │ │ │ │ ├── sync_bn_cuda.cu │ │ │ │ │ ├── three_interpolate_cuda.cu │ │ │ │ │ ├── three_nn_cuda.cu │ │ │ │ │ ├── tin_shift_cuda.cu │ │ │ │ │ ├── upfirdn2d_kernel.cu │ │ │ │ │ └── voxelization_cuda.cu │ │ │ │ ├── deform_conv.cpp │ │ │ │ ├── deform_roi_pool.cpp │ │ │ │ ├── diff_iou_rotated.cpp │ │ │ │ ├── focal_loss.cpp │ │ │ │ ├── furthest_point_sample.cpp │ │ │ │ ├── fused_bias_leakyrelu.cpp │ │ │ │ ├── fused_spconv_ops.cpp │ │ │ │ ├── gather_points.cpp │ │ │ │ ├── group_points.cpp │ │ │ │ ├── info.cpp │ │ │ │ ├── iou3d.cpp │ │ │ │ ├── knn.cpp │ │ │ │ ├── masked_conv2d.cpp │ │ │ │ ├── min_area_polygons.cpp │ │ │ │ ├── mlu │ │ │ │ │ ├── bbox_overlaps_mlu.cpp │ │ │ │ │ ├── focal_loss_sigmoid_mlu.cpp │ │ │ │ │ ├── nms_mlu.cpp │ │ │ │ │ ├── psamask_mlu.cpp │ │ │ │ │ ├── roi_align_mlu.cpp │ │ │ │ │ ├── roi_align_rotated_mlu.cpp │ │ │ │ │ └── tin_shift_mlu.cpp │ │ │ │ ├── modulated_deform_conv.cpp │ │ │ │ ├── ms_deform_attn.cpp │ │ │ │ ├── nms.cpp │ │ │ │ ├── nms_rotated.cpp │ │ │ │ ├── pixel_group.cpp │ │ │ │ ├── points_in_boxes.cpp │ │ │ │ ├── points_in_polygons.cpp │ │ │ │ ├── psamask.cpp │ │ │ │ ├── pybind.cpp │ │ │ │ ├── riroi_align_rotated.cpp │ │ │ │ ├── roi_align.cpp │ │ │ │ ├── roi_align_rotated.cpp │ │ │ │ ├── roi_pool.cpp │ │ │ │ ├── roiaware_pool3d.cpp │ │ │ │ ├── roipoint_pool3d.cpp │ │ │ │ ├── rotated_feature_align.cpp │ │ │ │ ├── scatter_points.cpp │ │ │ │ ├── sparse_pool_ops.cpp │ │ │ │ ├── spconv_ops.cpp │ │ │ │ ├── spconv_utils.h │ │ │ │ ├── sync_bn.cpp │ │ │ │ ├── three_interpolate.cpp │ │ │ │ ├── three_nn.cpp │ │ │ │ ├── tin_shift.cpp │ │ │ │ ├── upfirdn2d.cpp │ │ │ │ └── voxelization.cpp │ │ │ └── tensorrt │ │ │ │ ├── plugins │ │ │ │ ├── trt_corner_pool.cpp │ │ │ │ ├── trt_corner_pool_kernel.cu │ │ │ │ ├── trt_cuda_helper.cu │ │ │ │ ├── trt_cummaxmin.cpp │ │ │ │ ├── trt_cummaxmin_kernel.cu │ │ │ │ ├── trt_deform_conv.cpp │ │ │ │ ├── trt_deform_conv_kernel.cu │ │ │ │ ├── trt_grid_sampler.cpp │ │ │ │ ├── trt_grid_sampler_kernel.cu │ │ │ │ ├── trt_instance_norm.cpp │ │ │ │ ├── trt_modulated_deform_conv.cpp │ │ │ │ ├── trt_modulated_deform_conv_kernel.cu │ │ │ │ ├── trt_nms.cpp │ │ │ │ ├── trt_nms_kernel.cu │ │ │ │ ├── trt_plugin.cpp │ │ │ │ ├── trt_roi_align.cpp │ │ │ │ ├── trt_roi_align_kernel.cu │ │ │ │ ├── trt_scatternd.cpp │ │ │ │ └── trt_scatternd_kernel.cu │ │ │ │ ├── trt_corner_pool.hpp │ │ │ │ ├── trt_cuda_helper.cuh │ │ │ │ ├── trt_cummaxmin.hpp │ │ │ │ ├── trt_deform_conv.hpp │ │ │ │ ├── trt_grid_sampler.hpp │ │ │ │ ├── trt_instance_norm.hpp │ │ │ │ ├── trt_modulated_deform_conv.hpp │ │ │ │ ├── trt_nms.hpp │ │ │ │ ├── trt_plugin.hpp │ │ │ │ ├── trt_plugin_helper.hpp │ │ │ │ ├── trt_roi_align.hpp │ │ │ │ ├── trt_scatternd.hpp │ │ │ │ └── trt_serialize.hpp │ │ ├── deform_conv.py │ │ ├── deform_roi_pool.py │ │ ├── deprecated_wrappers.py │ │ ├── diff_iou_rotated.py │ │ ├── focal_loss.py │ │ ├── furthest_point_sample.py │ │ ├── fused_bias_leakyrelu.py │ │ ├── gather_points.py │ │ ├── group_points.py │ │ ├── info.py │ │ ├── iou3d.py │ │ ├── knn.py │ │ ├── masked_conv.py │ │ ├── merge_cells.py │ │ ├── min_area_polygons.py │ │ ├── modulated_deform_conv.py │ │ ├── multi_scale_deform_attn.py │ │ ├── nms.py │ │ ├── pixel_group.py │ │ ├── point_sample.py │ │ ├── points_in_boxes.py │ │ ├── points_in_polygons.py │ │ ├── points_sampler.py │ │ ├── psa_mask.py │ │ ├── riroi_align_rotated.py │ │ ├── roi_align.py │ │ ├── roi_align_rotated.py │ │ ├── roi_pool.py │ │ ├── roiaware_pool3d.py │ │ ├── roipoint_pool3d.py │ │ ├── rotated_feature_align.py │ │ ├── saconv.py │ │ ├── scatter_points.py │ │ ├── sparse_conv.py │ │ ├── sparse_functional.py │ │ ├── sparse_modules.py │ │ ├── sparse_ops.py │ │ ├── sparse_pool.py │ │ ├── sparse_structure.py │ │ ├── sync_bn.py │ │ ├── three_interpolate.py │ │ ├── three_nn.py │ │ ├── tin_shift.py │ │ ├── upfirdn2d.py │ │ └── voxelize.py │ ├── parallel │ │ ├── __init__.py │ │ ├── _functions.py │ │ ├── collate.py │ │ ├── data_container.py │ │ ├── data_parallel.py │ │ ├── distributed.py │ │ ├── distributed_deprecated.py │ │ ├── registry.py │ │ ├── scatter_gather.py │ │ └── utils.py │ ├── runner │ │ ├── __init__.py │ │ ├── base_module.py │ │ ├── base_runner.py │ │ ├── builder.py │ │ ├── checkpoint.py │ │ ├── default_constructor.py │ │ ├── dist_utils.py │ │ ├── epoch_based_runner.py │ │ ├── fp16_utils.py │ │ ├── hooks │ │ │ ├── __init__.py │ │ │ ├── checkpoint.py │ │ │ ├── closure.py │ │ │ ├── ema.py │ │ │ ├── evaluation.py │ │ │ ├── hook.py │ │ │ ├── iter_timer.py │ │ │ ├── logger │ │ │ │ ├── __init__.py │ │ │ │ ├── base.py │ │ │ │ ├── clearml.py │ │ │ │ ├── dvclive.py │ │ │ │ ├── mlflow.py │ │ │ │ ├── neptune.py │ │ │ │ ├── pavi.py │ │ │ │ ├── segmind.py │ │ │ │ ├── tensorboard.py │ │ │ │ ├── text.py │ │ │ │ └── wandb.py │ │ │ ├── lr_updater.py │ │ │ ├── memory.py │ │ │ ├── momentum_updater.py │ │ │ ├── optimizer.py │ │ │ ├── profiler.py │ │ │ ├── sampler_seed.py │ │ │ └── sync_buffer.py │ │ ├── iter_based_runner.py │ │ ├── log_buffer.py │ │ ├── optimizer │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ └── default_constructor.py │ │ ├── priority.py │ │ └── utils.py │ ├── tensorrt │ │ ├── __init__.py │ │ ├── init_plugins.py │ │ ├── preprocess.py │ │ └── tensorrt_utils.py │ ├── transforms │ │ ├── __init__.py │ │ ├── base.py │ │ ├── builder.py │ │ ├── formatting.py │ │ ├── loading.py │ │ ├── processing.py │ │ ├── utils.py │ │ └── wrappers.py │ ├── utils │ │ ├── __init__.py │ │ ├── config.py │ │ ├── device_type.py │ │ ├── env.py │ │ ├── ext_loader.py │ │ ├── hub.py │ │ ├── logging.py │ │ ├── misc.py │ │ ├── parrots_jit.py │ │ ├── parrots_wrapper.py │ │ ├── path.py │ │ ├── progressbar.py │ │ ├── registry.py │ │ ├── seed.py │ │ ├── testing.py │ │ ├── timer.py │ │ ├── trace.py │ │ └── version_utils.py │ ├── version.py │ ├── video │ │ ├── __init__.py │ │ ├── io.py │ │ ├── optflow.py │ │ └── processing.py │ └── visualization │ │ ├── __init__.py │ │ ├── color.py │ │ ├── image.py │ │ └── optflow.py ├── requirements.txt ├── requirements │ ├── build.txt │ ├── docs.txt │ ├── optional.txt │ ├── runtime.txt │ └── test.txt ├── setup.cfg ├── setup.py └── tests │ ├── test_arraymisc.py │ ├── test_cnn │ ├── test_build_layers.py │ ├── test_context_block.py │ ├── test_conv2d_adaptive_padding.py │ ├── test_conv_module.py │ ├── test_depthwise_seperable_conv_module.py │ ├── test_flops_counter.py │ ├── test_fuse_conv_bn.py │ ├── test_generalized_attention.py │ ├── test_hsigmoid.py │ ├── test_hswish.py │ ├── test_model_registry.py │ ├── test_non_local.py │ ├── test_revert_syncbn.py │ ├── test_scale.py │ ├── test_swish.py │ ├── test_transformer.py │ ├── test_weight_init.py │ └── test_wrappers.py │ ├── test_device │ ├── test_ipu │ │ ├── test_hierarchicaldatamanager.py │ │ ├── test_ipu_dataloder.py │ │ ├── test_ipu_hooks.py │ │ ├── test_ipu_model.py │ │ ├── test_ipu_runner.py │ │ └── test_ipu_utils.py │ └── test_mlu │ │ └── test_mlu_parallel.py │ ├── test_fileclient.py │ ├── test_fileio.py │ ├── test_image │ ├── test_colorspace.py │ ├── test_geometric.py │ ├── test_image_misc.py │ ├── test_io.py │ └── test_photometric.py │ ├── test_load_model_zoo.py │ ├── test_ops │ ├── test_active_rotated_filter.py │ ├── test_assign_score_withk.py │ ├── test_ball_query.py │ ├── test_bbox.py │ ├── test_bilinear_grid_sample.py │ ├── test_border_align.py │ ├── test_box_iou_rotated.py │ ├── test_carafe.py │ ├── test_cc_attention.py │ ├── test_chamfer_distance.py │ ├── test_contour_expand.py │ ├── test_convex_iou.py │ ├── test_corner_pool.py │ ├── test_correlation.py │ ├── test_deform_conv.py │ ├── test_deform_roi_pool.py │ ├── test_diff_iou_rotated.py │ ├── test_focal_loss.py │ ├── test_furthest_point_sample.py │ ├── test_fused_bias_leakyrelu.py │ ├── test_gather_points.py │ ├── test_group_points.py │ ├── test_info.py │ ├── test_iou3d.py │ ├── test_knn.py │ ├── test_masked_conv2d.py │ ├── test_merge_cells.py │ ├── test_min_area_polygons.py │ ├── test_modulated_deform_conv.py │ ├── test_ms_deformable_attn.py │ ├── test_nms.py │ ├── test_nms_rotated.py │ ├── test_onnx.py │ ├── test_pixel_group.py │ ├── test_points_in_polygons.py │ ├── test_psa_mask.py │ ├── test_riroi_align_rotated.py │ ├── test_roi_align.py │ ├── test_roi_align_rotated.py │ ├── test_roi_pool.py │ ├── test_roiaware_pool3d.py │ ├── test_roipoint_pool3d.py │ ├── test_rotated_feature_align.py │ ├── test_saconv.py │ ├── test_scatter_points.py │ ├── test_spconv.py │ ├── test_syncbn.py │ ├── test_tensorrt.py │ ├── test_tensorrt_preprocess.py │ ├── test_three_interpolate.py │ ├── test_three_nn.py │ ├── test_tin_shift.py │ ├── test_upfirdn2d.py │ └── test_voxelization.py │ ├── test_parallel.py │ ├── test_runner │ ├── test_basemodule.py │ ├── test_checkpoint.py │ ├── test_dist_utils.py │ ├── test_eval_hook.py │ ├── test_fp16.py │ ├── test_hooks.py │ ├── test_optimizer.py │ ├── test_runner.py │ └── test_utils.py │ ├── test_utils │ ├── test_config.py │ ├── test_env.py │ ├── test_hub.py │ ├── test_logging.py │ ├── test_misc.py │ ├── test_parrots_jit.py │ ├── test_path.py │ ├── test_progressbar.py │ ├── test_registry.py │ ├── test_testing.py │ ├── test_timer.py │ ├── test_trace.py │ └── test_version_utils.py │ ├── test_video │ ├── test_optflow.py │ ├── test_processing.py │ └── test_reader.py │ └── test_visualization.py ├── mmhuman3d ├── __init__.py ├── apis │ ├── __init__.py │ ├── inference.py │ ├── test.py │ └── train.py ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ └── pw3d.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ └── resnet50.py │ │ ├── post_processing │ │ │ ├── README.md │ │ │ ├── deciwatch_interval10_q1.py │ │ │ ├── deciwatch_interval10_q2.py │ │ │ ├── deciwatch_interval10_q3.py │ │ │ ├── deciwatch_interval10_q4.py │ │ │ ├── deciwatch_interval10_q5.py │ │ │ ├── deciwatch_interval5_q1.py │ │ │ ├── deciwatch_interval5_q2.py │ │ │ ├── deciwatch_interval5_q3.py │ │ │ ├── deciwatch_interval5_q4.py │ │ │ ├── deciwatch_interval5_q5.py │ │ │ ├── guas1d.py │ │ │ ├── oneeuro.py │ │ │ ├── savgol.py │ │ │ ├── smoothnet_windowsize16.py │ │ │ ├── smoothnet_windowsize32.py │ │ │ ├── smoothnet_windowsize64.py │ │ │ └── smoothnet_windowsize8.py │ │ └── schedulers │ │ │ └── hmr_bs256.py │ ├── expose │ │ ├── README.md │ │ ├── expose.py │ │ ├── hrnet_hmr_expose_body.py │ │ ├── metafile.yml │ │ ├── resnet18_hmr_expose_face.py │ │ └── resnet18_hmr_expose_hand.py │ ├── gta_human │ │ ├── README.md │ │ ├── hmr │ │ │ ├── README.md │ │ │ ├── resnet50_hmr_gta_bt.py │ │ │ └── resnet50_hmr_gta_ft.py │ │ ├── pare │ │ │ ├── README.md │ │ │ └── hrnet_w32_conv_pare_gta_ft.py │ │ └── spin │ │ │ ├── README.md │ │ │ └── resnet50_spin_gta_ft.py │ ├── hmr │ │ ├── README.md │ │ ├── metafile.yml │ │ ├── resnet50_hmr_pw3d.py │ │ └── resnet50_hmr_pw3d_e50_cache.py │ ├── hybrik │ │ ├── README.md │ │ ├── metafile.yml │ │ └── resnet34_hybrik_mixed.py │ ├── pare │ │ ├── README.md │ │ ├── hrnet_w32_conv_pare_coco.py │ │ ├── hrnet_w32_conv_pare_coco_cache.py │ │ ├── hrnet_w32_conv_pare_mix.py │ │ ├── hrnet_w32_conv_pare_mix_cache.py │ │ ├── hrnet_w32_conv_pare_mix_no_mosh.py │ │ └── metafile.yml │ ├── render │ │ └── smpl.py │ ├── smplify │ │ ├── smplify.py │ │ ├── smplify3d.py │ │ └── smplifyx.py │ ├── spin │ │ ├── README.md │ │ ├── metafile.yml │ │ ├── resnet50_spin_pw3d.py │ │ └── resnet50_spin_pw3d_cache.py │ └── vibe │ │ ├── README.md │ │ ├── metafile.yml │ │ └── resnet50_vibe_pw3d.py ├── core │ ├── __init__.py │ ├── cameras │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── camera_parameters.py │ │ └── cameras.py │ ├── conventions │ │ ├── __init__.py │ │ ├── cameras │ │ │ ├── __init__.py │ │ │ ├── convert_convention.py │ │ │ └── convert_projection.py │ │ ├── joints_mapping │ │ │ └── standard_joint_angles.py │ │ ├── keypoints_mapping │ │ │ ├── __init__.py │ │ │ ├── agora.py │ │ │ ├── coco.py │ │ │ ├── coco_wholebody.py │ │ │ ├── crowdpose.py │ │ │ ├── face3d.py │ │ │ ├── flame.py │ │ │ ├── gta.py │ │ │ ├── h36m.py │ │ │ ├── human_data.py │ │ │ ├── hybrik.py │ │ │ ├── instavariety.py │ │ │ ├── lsp.py │ │ │ ├── mano.py │ │ │ ├── mpi_inf_3dhp.py │ │ │ ├── mpii.py │ │ │ ├── openpose.py │ │ │ ├── penn_action.py │ │ │ ├── posetrack.py │ │ │ ├── pw3d.py │ │ │ ├── smpl.py │ │ │ ├── smplx.py │ │ │ ├── spin_smplx.py │ │ │ └── star.py │ │ └── segmentation │ │ │ ├── __init__.py │ │ │ ├── smpl.py │ │ │ └── smplx.py │ ├── distributed_wrapper.py │ ├── evaluation │ │ ├── __init__.py │ │ ├── eval_hooks.py │ │ ├── eval_utils.py │ │ └── mesh_eval.py │ ├── optimizer │ │ ├── __init__.py │ │ └── builder.py │ ├── post_processing │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── smooth │ │ │ ├── gaus1d_filter.py │ │ │ ├── oneeuro_filter.py │ │ │ ├── savgol_filter.py │ │ │ └── smoothnet.py │ │ └── speed_up │ │ │ └── deciwatch.py │ ├── renderer │ │ ├── __init__.py │ │ ├── matplotlib3d_renderer.py │ │ ├── torch3d_renderer │ │ │ ├── base_renderer.py │ │ │ ├── builder.py │ │ │ ├── depth_renderer.py │ │ │ ├── lights │ │ │ │ ├── __init__.py │ │ │ │ ├── builder.py │ │ │ │ └── lights.py │ │ │ ├── mesh_renderer.py │ │ │ ├── meshes.py │ │ │ ├── normal_renderer.py │ │ │ ├── pointcloud_renderer.py │ │ │ ├── render_runner.py │ │ │ ├── segmentation_renderer.py │ │ │ ├── shader │ │ │ │ ├── __init__.py │ │ │ │ ├── builder.py │ │ │ │ └── shader.py │ │ │ ├── silhouette_renderer.py │ │ │ ├── smpl_renderer.py │ │ │ ├── textures │ │ │ │ ├── __init__.py │ │ │ │ ├── builder.py │ │ │ │ └── textures.py │ │ │ ├── utils.py │ │ │ └── uv_renderer.py │ │ └── vedo_render.py │ └── visualization │ │ ├── __init__.py │ │ ├── visualize_cameras.py │ │ ├── visualize_keypoints2d.py │ │ ├── visualize_keypoints3d.py │ │ └── visualize_smpl.py ├── data │ ├── __init__.py │ ├── data_converters │ │ ├── __init__.py │ │ ├── agora.py │ │ ├── amass.py │ │ ├── base_converter.py │ │ ├── builder.py │ │ ├── coco.py │ │ ├── coco_hybrik.py │ │ ├── coco_wholebody.py │ │ ├── crowdpose.py │ │ ├── eft.py │ │ ├── ehf.py │ │ ├── expose_curated_fits.py │ │ ├── expose_spin_smplx.py │ │ ├── ffhq_flame.py │ │ ├── freihand.py │ │ ├── gta_human.py │ │ ├── h36m.py │ │ ├── h36m_hybrik.py │ │ ├── h36m_smplx.py │ │ ├── humman.py │ │ ├── insta_vibe.py │ │ ├── lsp.py │ │ ├── lsp_extended.py │ │ ├── mpi_inf_3dhp.py │ │ ├── mpi_inf_3dhp_hybrik.py │ │ ├── mpii.py │ │ ├── penn_action.py │ │ ├── posetrack.py │ │ ├── pw3d.py │ │ ├── pw3d_hybrik.py │ │ ├── spin.py │ │ ├── stirling.py │ │ ├── surreal.py │ │ ├── up3d.py │ │ └── vibe.py │ ├── data_structures │ │ ├── __init__.py │ │ ├── human_data.py │ │ ├── human_data_cache.py │ │ └── smc_reader.py │ └── datasets │ │ ├── __init__.py │ │ ├── adversarial_dataset.py │ │ ├── base_dataset.py │ │ ├── builder.py │ │ ├── dataset_wrappers.py │ │ ├── human_hybrik_dataset.py │ │ ├── human_image_dataset.py │ │ ├── human_image_smplx_dataset.py │ │ ├── human_video_dataset.py │ │ ├── mesh_dataset.py │ │ ├── mixed_dataset.py │ │ ├── pipelines │ │ ├── __init__.py │ │ ├── compose.py │ │ ├── formatting.py │ │ ├── hybrik_transforms.py │ │ ├── loading.py │ │ ├── synthetic_occlusion_augmentation.py │ │ └── transforms.py │ │ └── samplers │ │ ├── __init__.py │ │ └── distributed_sampler.py ├── models │ ├── __init__.py │ ├── architectures │ │ ├── __init__.py │ │ ├── base_architecture.py │ │ ├── builder.py │ │ ├── expressive_mesh_estimator.py │ │ ├── hybrik.py │ │ └── mesh_estimator.py │ ├── backbones │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── hrnet.py │ │ └── resnet.py │ ├── body_models │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── flame.py │ │ ├── mano.py │ │ ├── smpl.py │ │ ├── smplx.py │ │ ├── star.py │ │ └── utils.py │ ├── discriminators │ │ ├── __init__.py │ │ ├── builder.py │ │ └── pose_discriminator.py │ ├── heads │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── expose_head.py │ │ ├── hmr_head.py │ │ ├── hybrik_head.py │ │ └── pare_head.py │ ├── losses │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── cross_entropy_loss.py │ │ ├── gan_loss.py │ │ ├── mse_loss.py │ │ ├── prior_loss.py │ │ ├── rotaion_distance_loss.py │ │ ├── smooth_l1_loss.py │ │ └── utils.py │ ├── necks │ │ ├── __init__.py │ │ ├── builder.py │ │ └── temporal_encoder.py │ ├── registrants │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── smplify.py │ │ └── smplifyx.py │ └── utils │ │ ├── SMPLX.py │ │ ├── __init__.py │ │ ├── fits_dict.py │ │ ├── inverse_kinematics.py │ │ └── res_layer.py ├── utils │ ├── __init__.py │ ├── camera_utils.py │ ├── collect_env.py │ ├── demo_utils.py │ ├── dist_utils.py │ ├── ffmpeg_utils.py │ ├── geometry.py │ ├── keypoint_utils.py │ ├── logger.py │ ├── mesh_utils.py │ ├── misc.py │ ├── path_utils.py │ └── transforms.py └── version.py ├── mmhuman_data └── .gitkeep ├── pretrained ├── model_init_weights │ └── .gitkeep ├── pose │ └── .gitkeep ├── rtmpose │ └── .gitkeep └── transfer_data │ └── .gitkeep ├── sapiens ├── .gitignore ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── _install │ ├── conda.sh │ └── make_links.sh ├── cv │ ├── .gitignore │ ├── MANIFEST.in │ ├── mmcv │ │ ├── __init__.py │ │ ├── arraymisc │ │ │ ├── __init__.py │ │ │ └── quantization.py │ │ ├── cnn │ │ │ ├── __init__.py │ │ │ ├── alexnet.py │ │ │ ├── bricks │ │ │ │ ├── __init__.py │ │ │ │ ├── activation.py │ │ │ │ ├── context_block.py │ │ │ │ ├── conv.py │ │ │ │ ├── conv2d_adaptive_padding.py │ │ │ │ ├── conv_module.py │ │ │ │ ├── conv_ws.py │ │ │ │ ├── depthwise_separable_conv_module.py │ │ │ │ ├── drop.py │ │ │ │ ├── generalized_attention.py │ │ │ │ ├── hsigmoid.py │ │ │ │ ├── hswish.py │ │ │ │ ├── non_local.py │ │ │ │ ├── norm.py │ │ │ │ ├── padding.py │ │ │ │ ├── plugin.py │ │ │ │ ├── scale.py │ │ │ │ ├── swish.py │ │ │ │ ├── transformer.py │ │ │ │ ├── upsample.py │ │ │ │ └── wrappers.py │ │ │ ├── resnet.py │ │ │ ├── rfsearch │ │ │ │ ├── __init__.py │ │ │ │ ├── operator.py │ │ │ │ ├── search.py │ │ │ │ └── utils.py │ │ │ ├── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── flops_counter.py │ │ │ │ └── fuse_conv_bn.py │ │ │ └── vgg.py │ │ ├── image │ │ │ ├── __init__.py │ │ │ ├── colorspace.py │ │ │ ├── geometric.py │ │ │ ├── io.py │ │ │ ├── misc.py │ │ │ └── photometric.py │ │ ├── ops │ │ │ ├── __init__.py │ │ │ ├── active_rotated_filter.py │ │ │ ├── assign_score_withk.py │ │ │ ├── ball_query.py │ │ │ ├── bbox.py │ │ │ ├── bezier_align.py │ │ │ ├── bias_act.py │ │ │ ├── border_align.py │ │ │ ├── box_iou_quadri.py │ │ │ ├── box_iou_rotated.py │ │ │ ├── carafe.py │ │ │ ├── cc_attention.py │ │ │ ├── chamfer_distance.py │ │ │ ├── contour_expand.py │ │ │ ├── conv2d_gradfix.py │ │ │ ├── convex_iou.py │ │ │ ├── corner_pool.py │ │ │ ├── correlation.py │ │ │ ├── csrc │ │ │ │ ├── README.md │ │ │ │ ├── common │ │ │ │ │ ├── box_iou_rotated_utils.hpp │ │ │ │ │ ├── cuda │ │ │ │ │ │ ├── active_rotated_filter_cuda_kernel.cuh │ │ │ │ │ │ ├── assign_score_withk_cuda_kernel.cuh │ │ │ │ │ │ ├── ball_query_cuda_kernel.cuh │ │ │ │ │ │ ├── bbox_overlaps_cuda_kernel.cuh │ │ │ │ │ │ ├── bezier_align_cuda_kernel.cuh │ │ │ │ │ │ ├── border_align_cuda_kernel.cuh │ │ │ │ │ │ ├── box_iou_quadri_cuda.cuh │ │ │ │ │ │ ├── box_iou_rotated_cuda.cuh │ │ │ │ │ │ ├── carafe_cuda_kernel.cuh │ │ │ │ │ │ ├── carafe_naive_cuda_kernel.cuh │ │ │ │ │ │ ├── chamfer_distance_cuda_kernel.cuh │ │ │ │ │ │ ├── common_cuda_helper.hpp │ │ │ │ │ │ ├── convex_iou_cuda_kernel.cuh │ │ │ │ │ │ ├── correlation_cuda.cuh │ │ │ │ │ │ ├── deform_conv_cuda_kernel.cuh │ │ │ │ │ │ ├── deform_roi_pool_cuda_kernel.cuh │ │ │ │ │ │ ├── diff_iou_rotated_cuda_kernel.cuh │ │ │ │ │ │ ├── furthest_point_sample_cuda_kernel.cuh │ │ │ │ │ │ ├── gather_points_cuda_kernel.cuh │ │ │ │ │ │ ├── group_points_cuda_kernel.cuh │ │ │ │ │ │ ├── iou3d_cuda_kernel.cuh │ │ │ │ │ │ ├── knn_cuda_kernel.cuh │ │ │ │ │ │ ├── masked_conv2d_cuda_kernel.cuh │ │ │ │ │ │ ├── min_area_polygons_cuda.cuh │ │ │ │ │ │ ├── modulated_deform_conv_cuda_kernel.cuh │ │ │ │ │ │ ├── ms_deform_attn_cuda_kernel.cuh │ │ │ │ │ │ ├── nms_cuda_kernel.cuh │ │ │ │ │ │ ├── nms_quadri_cuda.cuh │ │ │ │ │ │ ├── nms_rotated_cuda.cuh │ │ │ │ │ │ ├── parrots_cudawarpfunction.cuh │ │ │ │ │ │ ├── points_in_boxes_cuda_kernel.cuh │ │ │ │ │ │ ├── points_in_polygons_cuda_kernel.cuh │ │ │ │ │ │ ├── prroi_pool_cuda_kernel.cuh │ │ │ │ │ │ ├── psamask_cuda_kernel.cuh │ │ │ │ │ │ ├── riroi_align_rotated_cuda_kernel.cuh │ │ │ │ │ │ ├── roi_align_cuda_kernel.cuh │ │ │ │ │ │ ├── roi_align_rotated_cuda_kernel.cuh │ │ │ │ │ │ ├── roi_pool_cuda_kernel.cuh │ │ │ │ │ │ ├── roiaware_pool3d_cuda_kernel.cuh │ │ │ │ │ │ ├── roipoint_pool3d_cuda_kernel.cuh │ │ │ │ │ │ ├── rotated_feature_align_cuda_kernel.cuh │ │ │ │ │ │ ├── scatter_points_cuda_kernel.cuh │ │ │ │ │ │ ├── sigmoid_focal_loss_cuda_kernel.cuh │ │ │ │ │ │ ├── softmax_focal_loss_cuda_kernel.cuh │ │ │ │ │ │ ├── spconv │ │ │ │ │ │ │ ├── indice.cuh │ │ │ │ │ │ │ └── reordering.cuh │ │ │ │ │ │ ├── stack_ball_query_cuda_kernel.cuh │ │ │ │ │ │ ├── stack_group_points_cuda_kernel.cuh │ │ │ │ │ │ ├── sync_bn_cuda_kernel.cuh │ │ │ │ │ │ ├── three_interpolate_cuda_kernel.cuh │ │ │ │ │ │ ├── three_nn_cuda_kernel.cuh │ │ │ │ │ │ ├── tin_shift_cuda_kernel.cuh │ │ │ │ │ │ └── voxelization_cuda_kernel.cuh │ │ │ │ │ ├── mlu │ │ │ │ │ │ ├── common_mlu_helper.hpp │ │ │ │ │ │ ├── masked_conv2d_mlu_kernel.mlu │ │ │ │ │ │ └── roi_pool_mlu_kernel.mlu │ │ │ │ │ ├── mps │ │ │ │ │ │ ├── MPSDevice.h │ │ │ │ │ │ ├── MPSLibrary.h │ │ │ │ │ │ ├── MPSLibrary.mm │ │ │ │ │ │ ├── MPSStream.h │ │ │ │ │ │ └── MPSUtils.h │ │ │ │ │ ├── parrots_cpp_helper.hpp │ │ │ │ │ ├── parrots_cuda_helper.hpp │ │ │ │ │ ├── pytorch_cpp_helper.hpp │ │ │ │ │ ├── pytorch_cuda_helper.hpp │ │ │ │ │ ├── pytorch_device_registry.hpp │ │ │ │ │ ├── pytorch_mlu_helper.hpp │ │ │ │ │ ├── pytorch_npu_helper.hpp │ │ │ │ │ └── utils │ │ │ │ │ │ └── spconv │ │ │ │ │ │ ├── paramsgrid.h │ │ │ │ │ │ ├── prettyprint.h │ │ │ │ │ │ ├── pybind11_utils.h │ │ │ │ │ │ ├── spconv │ │ │ │ │ │ ├── geometry.h │ │ │ │ │ │ ├── indice.h │ │ │ │ │ │ ├── maxpool.h │ │ │ │ │ │ ├── mp_helper.h │ │ │ │ │ │ ├── point2voxel.h │ │ │ │ │ │ └── reordering.h │ │ │ │ │ │ └── tensorview │ │ │ │ │ │ ├── helper_kernel.cuh │ │ │ │ │ │ ├── helper_launch.h │ │ │ │ │ │ └── tensorview.h │ │ │ │ ├── parrots │ │ │ │ │ ├── active_rotated_filter.cpp │ │ │ │ │ ├── active_rotated_filter_parrots.cpp │ │ │ │ │ ├── active_rotated_filter_pytorch.h │ │ │ │ │ ├── assign_score_withk.cpp │ │ │ │ │ ├── assign_score_withk_parrots.cpp │ │ │ │ │ ├── assign_score_withk_pytorch.h │ │ │ │ │ ├── ball_query.cpp │ │ │ │ │ ├── ball_query_pytorch.h │ │ │ │ │ ├── bbox_overlaps.cpp │ │ │ │ │ ├── bbox_overlaps_parrots.cpp │ │ │ │ │ ├── bbox_overlaps_pytorch.h │ │ │ │ │ ├── border_align.cpp │ │ │ │ │ ├── border_align_parrots.cpp │ │ │ │ │ ├── border_align_pytorch.h │ │ │ │ │ ├── box_iou_rotated.cpp │ │ │ │ │ ├── box_iou_rotated_parrots.cpp │ │ │ │ │ ├── box_iou_rotated_pytorch.h │ │ │ │ │ ├── carafe.cpp │ │ │ │ │ ├── carafe_naive.cpp │ │ │ │ │ ├── carafe_naive_parrots.cpp │ │ │ │ │ ├── carafe_naive_pytorch.h │ │ │ │ │ ├── carafe_parrots.cpp │ │ │ │ │ ├── carafe_pytorch.h │ │ │ │ │ ├── chamfer_distance.cpp │ │ │ │ │ ├── chamfer_distance_parrots.cpp │ │ │ │ │ ├── chamfer_distance_pytorch.h │ │ │ │ │ ├── contour_expand.cpp │ │ │ │ │ ├── contour_expand_parrots.cpp │ │ │ │ │ ├── contour_expand_pytorch.h │ │ │ │ │ ├── convex_iou.cpp │ │ │ │ │ ├── convex_iou_parrots.cpp │ │ │ │ │ ├── convex_iou_pytorch.h │ │ │ │ │ ├── correlation.cpp │ │ │ │ │ ├── correlation_parrots.cpp │ │ │ │ │ ├── correlation_pytorch.h │ │ │ │ │ ├── cudabind.cpp │ │ │ │ │ ├── deform_conv.cpp │ │ │ │ │ ├── deform_conv_parrots.cpp │ │ │ │ │ ├── deform_conv_pytorch.h │ │ │ │ │ ├── deform_roi_pool.cpp │ │ │ │ │ ├── deform_roi_pool_parrots.cpp │ │ │ │ │ ├── deform_roi_pool_pytorch.h │ │ │ │ │ ├── diff_iou_rotated.cpp │ │ │ │ │ ├── diff_iou_rotated_parrots.cpp │ │ │ │ │ ├── diff_iou_rotated_pytorch.h │ │ │ │ │ ├── focal_loss.cpp │ │ │ │ │ ├── focal_loss_parrots.cpp │ │ │ │ │ ├── focal_loss_pytorch.h │ │ │ │ │ ├── furthest_point_sample.cpp │ │ │ │ │ ├── furthest_point_sample_parrots.cpp │ │ │ │ │ ├── furthest_point_sample_pytorch.h │ │ │ │ │ ├── fused_bias_leakyrelu.cpp │ │ │ │ │ ├── fused_bias_parrots.cpp │ │ │ │ │ ├── gather_points.cpp │ │ │ │ │ ├── gather_points_parrots.cpp │ │ │ │ │ ├── gather_points_pytorch.h │ │ │ │ │ ├── group_points.cpp │ │ │ │ │ ├── group_points_parrots.cpp │ │ │ │ │ ├── group_points_pytorch.h │ │ │ │ │ ├── info.cpp │ │ │ │ │ ├── iou3d.cpp │ │ │ │ │ ├── iou3d_parrots.cpp │ │ │ │ │ ├── iou3d_pytorch.h │ │ │ │ │ ├── knn.cpp │ │ │ │ │ ├── knn_parrots.cpp │ │ │ │ │ ├── knn_pytorch.h │ │ │ │ │ ├── masked_conv2d.cpp │ │ │ │ │ ├── masked_conv2d_parrots.cpp │ │ │ │ │ ├── masked_conv2d_pytorch.h │ │ │ │ │ ├── min_area_polygons.cpp │ │ │ │ │ ├── min_area_polygons_parrots.cpp │ │ │ │ │ ├── min_area_polygons_pytorch.h │ │ │ │ │ ├── modulated_deform_conv.cpp │ │ │ │ │ ├── modulated_deform_conv_parrots.cpp │ │ │ │ │ ├── modulated_deform_conv_pytorch.h │ │ │ │ │ ├── ms_deform_attn.cpp │ │ │ │ │ ├── ms_deform_attn_parrots.cpp │ │ │ │ │ ├── nms.cpp │ │ │ │ │ ├── nms_parrots.cpp │ │ │ │ │ ├── nms_pytorch.h │ │ │ │ │ ├── nms_rotated.cpp │ │ │ │ │ ├── pixel_group.cpp │ │ │ │ │ ├── pixel_group_parrots.cpp │ │ │ │ │ ├── pixel_group_pytorch.h │ │ │ │ │ ├── points_in_boxes.cpp │ │ │ │ │ ├── points_in_boxes_parrots.cpp │ │ │ │ │ ├── points_in_boxes_pytorch.h │ │ │ │ │ ├── points_in_polygons.cpp │ │ │ │ │ ├── points_in_polygons_parrots.cpp │ │ │ │ │ ├── points_in_polygons_pytorch.h │ │ │ │ │ ├── prroi_pool.cpp │ │ │ │ │ ├── prroi_pool_parrots.cpp │ │ │ │ │ ├── prroi_pool_pytorch.h │ │ │ │ │ ├── psamask.cpp │ │ │ │ │ ├── psamask_parrots.cpp │ │ │ │ │ ├── psamask_pytorch.h │ │ │ │ │ ├── riroi_align_rotated.cpp │ │ │ │ │ ├── riroi_align_rotated_parrots.cpp │ │ │ │ │ ├── riroi_align_rotated_pytorch.h │ │ │ │ │ ├── roi_align.cpp │ │ │ │ │ ├── roi_align_parrots.cpp │ │ │ │ │ ├── roi_align_pytorch.h │ │ │ │ │ ├── roi_align_rotated.cpp │ │ │ │ │ ├── roi_align_rotated_parrots.cpp │ │ │ │ │ ├── roi_align_rotated_pytorch.h │ │ │ │ │ ├── roi_pool.cpp │ │ │ │ │ ├── roi_pool_parrots.cpp │ │ │ │ │ ├── roi_pool_pytorch.h │ │ │ │ │ ├── roiaware_pool3d.cpp │ │ │ │ │ ├── roiaware_pool3d_parrots.cpp │ │ │ │ │ ├── roiaware_pool3d_pytorch.h │ │ │ │ │ ├── roipoint_pool3d.cpp │ │ │ │ │ ├── roipoint_pool3d_parrots.cpp │ │ │ │ │ ├── roipoint_pool3d_pytorch.h │ │ │ │ │ ├── rotated_feature_align.cpp │ │ │ │ │ ├── rotated_feature_align_parrots.cpp │ │ │ │ │ ├── rotated_feature_align_pytorch.h │ │ │ │ │ ├── sync_bn.cpp │ │ │ │ │ ├── sync_bn_parrots.cpp │ │ │ │ │ ├── sync_bn_pytorch.h │ │ │ │ │ ├── three_interpolate.cpp │ │ │ │ │ ├── three_interpolate_parrots.cpp │ │ │ │ │ ├── three_interpolate_pytorch.h │ │ │ │ │ ├── three_nn.cpp │ │ │ │ │ ├── three_nn_parrots.cpp │ │ │ │ │ ├── three_nn_pytorch.h │ │ │ │ │ ├── tin_shift.cpp │ │ │ │ │ ├── tin_shift_parrots.cpp │ │ │ │ │ ├── tin_shift_pytorch.h │ │ │ │ │ ├── upfirdn2d.cpp │ │ │ │ │ ├── upfirdn2d_parrots.cpp │ │ │ │ │ ├── voxelization.cpp │ │ │ │ │ ├── voxelization_parrots.cpp │ │ │ │ │ └── voxelization_pytorch.h │ │ │ │ └── pytorch │ │ │ │ │ ├── active_rotated_filter.cpp │ │ │ │ │ ├── assign_score_withk.cpp │ │ │ │ │ ├── ball_query.cpp │ │ │ │ │ ├── bbox_overlaps.cpp │ │ │ │ │ ├── bezier_align.cpp │ │ │ │ │ ├── bias_act.cpp │ │ │ │ │ ├── border_align.cpp │ │ │ │ │ ├── box_iou_quadri.cpp │ │ │ │ │ ├── box_iou_rotated.cpp │ │ │ │ │ ├── carafe.cpp │ │ │ │ │ ├── carafe_naive.cpp │ │ │ │ │ ├── chamfer_distance.cpp │ │ │ │ │ ├── contour_expand.cpp │ │ │ │ │ ├── convex_iou.cpp │ │ │ │ │ ├── correlation.cpp │ │ │ │ │ ├── cpu │ │ │ │ │ ├── active_rotated_filter.cpp │ │ │ │ │ ├── bbox_overlaps_cpu.cpp │ │ │ │ │ ├── bezier_align.cpp │ │ │ │ │ ├── box_iou_quadri.cpp │ │ │ │ │ ├── box_iou_rotated.cpp │ │ │ │ │ ├── deform_conv.cpp │ │ │ │ │ ├── modulated_deform_conv.cpp │ │ │ │ │ ├── nms.cpp │ │ │ │ │ ├── nms_quadri.cpp │ │ │ │ │ ├── nms_rotated.cpp │ │ │ │ │ ├── pixel_group.cpp │ │ │ │ │ ├── points_in_boxes.cpp │ │ │ │ │ ├── psamask.cpp │ │ │ │ │ ├── roi_align.cpp │ │ │ │ │ ├── roi_align_rotated.cpp │ │ │ │ │ ├── rotated_feature_align.cpp │ │ │ │ │ ├── sparse_indice.cpp │ │ │ │ │ ├── sparse_maxpool.cpp │ │ │ │ │ ├── sparse_reordering.cpp │ │ │ │ │ └── voxelization.cpp │ │ │ │ │ ├── cuda │ │ │ │ │ ├── active_rotated_filter_cuda.cu │ │ │ │ │ ├── assign_score_withk_cuda.cu │ │ │ │ │ ├── ball_query_cuda.cu │ │ │ │ │ ├── bbox_overlaps_cuda.cu │ │ │ │ │ ├── bezier_align_cuda.cu │ │ │ │ │ ├── bias_act_cuda.cu │ │ │ │ │ ├── border_align_cuda.cu │ │ │ │ │ ├── box_iou_quadri_cuda.cu │ │ │ │ │ ├── box_iou_rotated_cuda.cu │ │ │ │ │ ├── carafe_cuda.cu │ │ │ │ │ ├── carafe_naive_cuda.cu │ │ │ │ │ ├── chamfer_distance_cuda.cu │ │ │ │ │ ├── convex_iou.cu │ │ │ │ │ ├── correlation_cuda.cu │ │ │ │ │ ├── cudabind.cpp │ │ │ │ │ ├── deform_conv_cuda.cu │ │ │ │ │ ├── deform_roi_pool_cuda.cu │ │ │ │ │ ├── diff_iou_rotated_cuda.cu │ │ │ │ │ ├── filtered_lrelu.cu │ │ │ │ │ ├── focal_loss_cuda.cu │ │ │ │ │ ├── furthest_point_sample_cuda.cu │ │ │ │ │ ├── fused_bias_leakyrelu_cuda.cu │ │ │ │ │ ├── fused_spconv_ops_cuda.cu │ │ │ │ │ ├── gather_points_cuda.cu │ │ │ │ │ ├── group_points_cuda.cu │ │ │ │ │ ├── iou3d_cuda.cu │ │ │ │ │ ├── knn_cuda.cu │ │ │ │ │ ├── masked_conv2d_cuda.cu │ │ │ │ │ ├── min_area_polygons.cu │ │ │ │ │ ├── modulated_deform_conv_cuda.cu │ │ │ │ │ ├── ms_deform_attn_cuda.cu │ │ │ │ │ ├── nms_cuda.cu │ │ │ │ │ ├── nms_quadri_cuda.cu │ │ │ │ │ ├── nms_rotated_cuda.cu │ │ │ │ │ ├── points_in_boxes_cuda.cu │ │ │ │ │ ├── points_in_polygons_cuda.cu │ │ │ │ │ ├── prroi_pool_cuda.cu │ │ │ │ │ ├── psamask_cuda.cu │ │ │ │ │ ├── riroi_align_rotated_cuda.cu │ │ │ │ │ ├── roi_align_cuda.cu │ │ │ │ │ ├── roi_align_rotated_cuda.cu │ │ │ │ │ ├── roi_pool_cuda.cu │ │ │ │ │ ├── roiaware_pool3d_cuda.cu │ │ │ │ │ ├── roipoint_pool3d_cuda.cu │ │ │ │ │ ├── rotated_feature_align_cuda.cu │ │ │ │ │ ├── scatter_points_cuda.cu │ │ │ │ │ ├── sparse_indice.cu │ │ │ │ │ ├── sparse_maxpool.cu │ │ │ │ │ ├── sparse_pool_ops_cuda.cu │ │ │ │ │ ├── sparse_reordering.cu │ │ │ │ │ ├── spconv_ops_cuda.cu │ │ │ │ │ ├── stack_ball_query_cuda.cu │ │ │ │ │ ├── stack_group_points_cuda.cu │ │ │ │ │ ├── sync_bn_cuda.cu │ │ │ │ │ ├── three_interpolate_cuda.cu │ │ │ │ │ ├── three_nn_cuda.cu │ │ │ │ │ ├── tin_shift_cuda.cu │ │ │ │ │ ├── upfirdn2d_kernel.cu │ │ │ │ │ └── voxelization_cuda.cu │ │ │ │ │ ├── deform_conv.cpp │ │ │ │ │ ├── deform_roi_pool.cpp │ │ │ │ │ ├── diff_iou_rotated.cpp │ │ │ │ │ ├── filtered_lrelu.cpp │ │ │ │ │ ├── focal_loss.cpp │ │ │ │ │ ├── furthest_point_sample.cpp │ │ │ │ │ ├── fused_bias_leakyrelu.cpp │ │ │ │ │ ├── fused_spconv_ops.cpp │ │ │ │ │ ├── gather_points.cpp │ │ │ │ │ ├── group_points.cpp │ │ │ │ │ ├── info.cpp │ │ │ │ │ ├── iou3d.cpp │ │ │ │ │ ├── knn.cpp │ │ │ │ │ ├── masked_conv2d.cpp │ │ │ │ │ ├── min_area_polygons.cpp │ │ │ │ │ ├── mlu │ │ │ │ │ ├── ball_query_mlu.cpp │ │ │ │ │ ├── bbox_overlaps_mlu.cpp │ │ │ │ │ ├── box_iou_rotated.cpp │ │ │ │ │ ├── carafe_mlu.cpp │ │ │ │ │ ├── deform_roi_pool_mlu.cpp │ │ │ │ │ ├── diff_iou_rotated_mlu.cpp │ │ │ │ │ ├── focal_loss_sigmoid_mlu.cpp │ │ │ │ │ ├── iou3d_mlu.cpp │ │ │ │ │ ├── masked_conv2d_mlu.cpp │ │ │ │ │ ├── mlu_common_helper.cpp │ │ │ │ │ ├── mlu_common_helper.h │ │ │ │ │ ├── ms_deform_attn_mlu.cpp │ │ │ │ │ ├── nms_mlu.cpp │ │ │ │ │ ├── nms_rotated_mlu.cpp │ │ │ │ │ ├── psamask_mlu.cpp │ │ │ │ │ ├── roi_align_mlu.cpp │ │ │ │ │ ├── roi_align_rotated_mlu.cpp │ │ │ │ │ ├── roi_pool_mlu.cpp │ │ │ │ │ ├── roiaware_pool3d_mlu.cpp │ │ │ │ │ ├── roipoint_pool3d_mlu.cpp │ │ │ │ │ ├── rotated_feature_align_mlu.cpp │ │ │ │ │ ├── scatter_points_mlu.cpp │ │ │ │ │ ├── sparse_conv_mlu.cpp │ │ │ │ │ ├── three_nn_mlu.cpp │ │ │ │ │ ├── tin_shift_mlu.cpp │ │ │ │ │ └── voxelization_mlu.cpp │ │ │ │ │ ├── modulated_deform_conv.cpp │ │ │ │ │ ├── mps │ │ │ │ │ └── bbox_overlaps_mps.mm │ │ │ │ │ ├── ms_deform_attn.cpp │ │ │ │ │ ├── nms.cpp │ │ │ │ │ ├── nms_quadri.cpp │ │ │ │ │ ├── nms_rotated.cpp │ │ │ │ │ ├── npu │ │ │ │ │ ├── active_rotated_filter_npu.cpp │ │ │ │ │ ├── bbox_overlaps_npu.cpp │ │ │ │ │ ├── box_iou_rotated_npu.cpp │ │ │ │ │ ├── deform_roi_pool.cpp │ │ │ │ │ ├── focal_loss_npu.cpp │ │ │ │ │ ├── fused_bias_leakyrelu_npu.cpp │ │ │ │ │ ├── gather_points_npu.cpp │ │ │ │ │ ├── group_points_npu.cpp │ │ │ │ │ ├── nms_npu.cpp │ │ │ │ │ ├── nms_rotated_npu.cpp │ │ │ │ │ ├── points_in_polygons_npu.cpp │ │ │ │ │ ├── psa_mask_npu.cpp │ │ │ │ │ ├── roi_align_npu.cpp │ │ │ │ │ ├── roi_pool_npu.cpp │ │ │ │ │ └── voxelization_npu.cpp │ │ │ │ │ ├── pixel_group.cpp │ │ │ │ │ ├── points_in_boxes.cpp │ │ │ │ │ ├── points_in_polygons.cpp │ │ │ │ │ ├── prroi_pool.cpp │ │ │ │ │ ├── psamask.cpp │ │ │ │ │ ├── pybind.cpp │ │ │ │ │ ├── riroi_align_rotated.cpp │ │ │ │ │ ├── roi_align.cpp │ │ │ │ │ ├── roi_align_rotated.cpp │ │ │ │ │ ├── roi_pool.cpp │ │ │ │ │ ├── roiaware_pool3d.cpp │ │ │ │ │ ├── roipoint_pool3d.cpp │ │ │ │ │ ├── rotated_feature_align.cpp │ │ │ │ │ ├── scatter_points.cpp │ │ │ │ │ ├── sparse_pool_ops.cpp │ │ │ │ │ ├── spconv_ops.cpp │ │ │ │ │ ├── spconv_utils.h │ │ │ │ │ ├── sync_bn.cpp │ │ │ │ │ ├── three_interpolate.cpp │ │ │ │ │ ├── three_nn.cpp │ │ │ │ │ ├── tin_shift.cpp │ │ │ │ │ ├── upfirdn2d.cpp │ │ │ │ │ └── voxelization.cpp │ │ │ ├── deform_conv.py │ │ │ ├── deform_roi_pool.py │ │ │ ├── deprecated_wrappers.py │ │ │ ├── diff_iou_rotated.py │ │ │ ├── filtered_lrelu.py │ │ │ ├── focal_loss.py │ │ │ ├── furthest_point_sample.py │ │ │ ├── fused_bias_leakyrelu.py │ │ │ ├── gather_points.py │ │ │ ├── group_points.py │ │ │ ├── info.py │ │ │ ├── iou3d.py │ │ │ ├── knn.py │ │ │ ├── masked_conv.py │ │ │ ├── merge_cells.py │ │ │ ├── min_area_polygons.py │ │ │ ├── modulated_deform_conv.py │ │ │ ├── multi_scale_deform_attn.py │ │ │ ├── nms.py │ │ │ ├── pixel_group.py │ │ │ ├── point_sample.py │ │ │ ├── points_in_boxes.py │ │ │ ├── points_in_polygons.py │ │ │ ├── points_sampler.py │ │ │ ├── prroi_pool.py │ │ │ ├── psa_mask.py │ │ │ ├── riroi_align_rotated.py │ │ │ ├── roi_align.py │ │ │ ├── roi_align_rotated.py │ │ │ ├── roi_pool.py │ │ │ ├── roiaware_pool3d.py │ │ │ ├── roipoint_pool3d.py │ │ │ ├── rotated_feature_align.py │ │ │ ├── saconv.py │ │ │ ├── scatter_points.py │ │ │ ├── sparse_conv.py │ │ │ ├── sparse_functional.py │ │ │ ├── sparse_modules.py │ │ │ ├── sparse_ops.py │ │ │ ├── sparse_pool.py │ │ │ ├── sparse_structure.py │ │ │ ├── sync_bn.py │ │ │ ├── three_interpolate.py │ │ │ ├── three_nn.py │ │ │ ├── tin_shift.py │ │ │ ├── upfirdn2d.py │ │ │ └── voxelize.py │ │ ├── transforms │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── builder.py │ │ │ ├── formatting.py │ │ │ ├── loading.py │ │ │ ├── processing.py │ │ │ ├── utils.py │ │ │ └── wrappers.py │ │ ├── utils │ │ │ ├── __init__.py │ │ │ ├── device_type.py │ │ │ ├── env.py │ │ │ ├── ext_loader.py │ │ │ └── parrots_jit.py │ │ ├── version.py │ │ ├── video │ │ │ ├── __init__.py │ │ │ ├── io.py │ │ │ ├── optflow.py │ │ │ └── processing.py │ │ └── visualization │ │ │ ├── __init__.py │ │ │ ├── color.py │ │ │ ├── image.py │ │ │ └── optflow.py │ ├── requirements.txt │ ├── requirements │ │ ├── build.txt │ │ ├── docs.txt │ │ ├── optional.txt │ │ ├── runtime.txt │ │ └── test.txt │ ├── setup.cfg │ └── setup.py ├── det │ ├── .gitignore │ ├── MANIFEST.in │ ├── configs │ │ ├── _base_ │ │ │ ├── datasets │ │ │ │ ├── ade20k_instance.py │ │ │ │ ├── ade20k_panoptic.py │ │ │ │ ├── ade20k_semantic.py │ │ │ │ ├── cityscapes_detection.py │ │ │ │ ├── cityscapes_instance.py │ │ │ │ ├── coco_caption.py │ │ │ │ ├── coco_detection.py │ │ │ │ ├── coco_instance.py │ │ │ │ ├── coco_instance_semantic.py │ │ │ │ ├── coco_panoptic.py │ │ │ │ ├── coco_semantic.py │ │ │ │ ├── deepfashion.py │ │ │ │ ├── dsdl.py │ │ │ │ ├── isaid_instance.py │ │ │ │ ├── lvis_v0.5_instance.py │ │ │ │ ├── lvis_v1_instance.py │ │ │ │ ├── mot_challenge.py │ │ │ │ ├── mot_challenge_det.py │ │ │ │ ├── mot_challenge_reid.py │ │ │ │ ├── objects365v1_detection.py │ │ │ │ ├── objects365v2_detection.py │ │ │ │ ├── openimages_detection.py │ │ │ │ ├── refcoco+.py │ │ │ │ ├── refcoco.py │ │ │ │ ├── refcocog.py │ │ │ │ ├── semi_coco_detection.py │ │ │ │ ├── v3det.py │ │ │ │ ├── voc0712.py │ │ │ │ ├── wider_face.py │ │ │ │ └── youtube_vis.py │ │ │ ├── default_runtime.py │ │ │ ├── models │ │ │ │ ├── cascade-mask-rcnn_r50_fpn.py │ │ │ │ ├── cascade-rcnn_r50_fpn.py │ │ │ │ ├── fast-rcnn_r50_fpn.py │ │ │ │ ├── faster-rcnn_r50-caffe-c4.py │ │ │ │ ├── faster-rcnn_r50-caffe-dc5.py │ │ │ │ ├── faster-rcnn_r50_fpn.py │ │ │ │ ├── mask-rcnn_r50-caffe-c4.py │ │ │ │ ├── mask-rcnn_r50_fpn.py │ │ │ │ ├── retinanet_r50_fpn.py │ │ │ │ ├── rpn_r50-caffe-c4.py │ │ │ │ ├── rpn_r50_fpn.py │ │ │ │ └── ssd300.py │ │ │ └── schedules │ │ │ │ ├── schedule_1x.py │ │ │ │ ├── schedule_20e.py │ │ │ │ └── schedule_2x.py │ │ └── rtmdet │ │ │ ├── README.md │ │ │ ├── classification │ │ │ ├── README.md │ │ │ ├── cspnext-l_8xb256-rsb-a1-600e_in1k.py │ │ │ ├── cspnext-m_8xb256-rsb-a1-600e_in1k.py │ │ │ ├── cspnext-s_8xb256-rsb-a1-600e_in1k.py │ │ │ ├── cspnext-tiny_8xb256-rsb-a1-600e_in1k.py │ │ │ └── cspnext-x_8xb256-rsb-a1-600e_in1k.py │ │ │ ├── metafile.yml │ │ │ ├── rtmdet-ins_l_8xb32-300e_coco.py │ │ │ ├── rtmdet-ins_m_8xb32-300e_coco.py │ │ │ ├── rtmdet-ins_s_8xb32-300e_coco.py │ │ │ ├── rtmdet-ins_tiny_8xb32-300e_coco.py │ │ │ ├── rtmdet-ins_x_8xb16-300e_coco.py │ │ │ ├── rtmdet_l_8xb32-300e_coco.py │ │ │ ├── rtmdet_m_8xb32-300e_coco.py │ │ │ ├── rtmdet_s_8xb32-300e_coco.py │ │ │ ├── rtmdet_tiny_8xb32-300e_coco.py │ │ │ ├── rtmdet_tta.py │ │ │ ├── rtmdet_x_8xb32-300e_coco.py │ │ │ └── rtmdet_x_p6_4xb8-300e_coco.py │ ├── mmdet │ │ ├── __init__.py │ │ ├── apis │ │ │ ├── __init__.py │ │ │ ├── det_inferencer.py │ │ │ └── inference.py │ │ ├── configs │ │ │ ├── _base_ │ │ │ │ ├── datasets │ │ │ │ │ ├── coco_detection.py │ │ │ │ │ ├── coco_instance.py │ │ │ │ │ ├── coco_instance_semantic.py │ │ │ │ │ ├── coco_panoptic.py │ │ │ │ │ └── mot_challenge.py │ │ │ │ ├── default_runtime.py │ │ │ │ ├── models │ │ │ │ │ ├── cascade_mask_rcnn_r50_fpn.py │ │ │ │ │ ├── cascade_rcnn_r50_fpn.py │ │ │ │ │ ├── faster_rcnn_r50_fpn.py │ │ │ │ │ ├── mask_rcnn_r50_caffe_c4.py │ │ │ │ │ ├── mask_rcnn_r50_fpn.py │ │ │ │ │ └── retinanet_r50_fpn.py │ │ │ │ └── schedules │ │ │ │ │ ├── schedule_1x.py │ │ │ │ │ └── schedule_2x.py │ │ │ ├── cascade_rcnn │ │ │ │ ├── cascade_mask_rcnn_r50_fpn_1x_coco.py │ │ │ │ └── cascade_rcnn_r50_fpn_1x_coco.py │ │ │ ├── common │ │ │ │ ├── lsj_100e_coco_detection.py │ │ │ │ ├── lsj_100e_coco_instance.py │ │ │ │ ├── lsj_200e_coco_detection.py │ │ │ │ ├── lsj_200e_coco_instance.py │ │ │ │ ├── ms_3x_coco.py │ │ │ │ ├── ms_3x_coco_instance.py │ │ │ │ ├── ms_90k_coco.py │ │ │ │ ├── ms_poly_3x_coco_instance.py │ │ │ │ ├── ms_poly_90k_coco_instance.py │ │ │ │ ├── ssj_270_coco_instance.py │ │ │ │ └── ssj_scp_270k_coco_instance.py │ │ │ ├── deformable_detr │ │ │ │ ├── deformable_detr_r50_16xb2_50e_coco.py │ │ │ │ ├── deformable_detr_refine_r50_16xb2_50e_coco.py │ │ │ │ └── deformable_detr_refine_twostage_r50_16xb2_50e_coco.py │ │ │ ├── detr │ │ │ │ ├── detr_r101_8xb2_500e_coco.py │ │ │ │ ├── detr_r18_8xb2_500e_coco.py │ │ │ │ ├── detr_r50_8xb2_150e_coco.py │ │ │ │ └── detr_r50_8xb2_500e_coco.py │ │ │ ├── dino │ │ │ │ ├── dino_4scale_r50_8xb2_12e_coco.py │ │ │ │ ├── dino_4scale_r50_8xb2_24e_coco.py │ │ │ │ ├── dino_4scale_r50_8xb2_36e_coco.py │ │ │ │ ├── dino_4scale_r50_improved_8xb2_12e_coco.py │ │ │ │ ├── dino_5scale_swin_l_8xb2_12e_coco.py │ │ │ │ └── dino_5scale_swin_l_8xb2_36e_coco.py │ │ │ ├── faster_rcnn │ │ │ │ └── faster_rcnn_r50_fpn_1x_coco.py │ │ │ ├── mask_rcnn │ │ │ │ ├── mask_rcnn_r101_caffe_fpn_1x_coco.py │ │ │ │ ├── mask_rcnn_r101_caffe_fpn_ms_poly_3x_coco.py │ │ │ │ ├── mask_rcnn_r101_fpn_1x_coco.py │ │ │ │ ├── mask_rcnn_r101_fpn_2x_coco.py │ │ │ │ ├── mask_rcnn_r101_fpn_8xb8_amp_lsj_200e_coco.py │ │ │ │ ├── mask_rcnn_r101_fpn_ms_poly_3x_coco.py │ │ │ │ ├── mask_rcnn_r18_fpn_8xb8_amp_lsj_200e_coco.py │ │ │ │ ├── mask_rcnn_r50_caffe_c4_1x_coco.py │ │ │ │ ├── mask_rcnn_r50_caffe_fpn_1x_coco.py │ │ │ │ ├── mask_rcnn_r50_caffe_fpn_ms_1x_coco.py │ │ │ │ ├── mask_rcnn_r50_caffe_fpn_ms_poly_1x_coco.py │ │ │ │ ├── mask_rcnn_r50_caffe_fpn_ms_poly_2x_coco.py │ │ │ │ ├── mask_rcnn_r50_caffe_fpn_ms_poly_3x_coco.py │ │ │ │ ├── mask_rcnn_r50_caffe_fpn_poly_1x_coco_v1.py │ │ │ │ ├── mask_rcnn_r50_fpn_1x_coco.py │ │ │ │ ├── mask_rcnn_r50_fpn_1x_wandb_coco.py │ │ │ │ ├── mask_rcnn_r50_fpn_2x_coco.py │ │ │ │ ├── mask_rcnn_r50_fpn_8xb8_amp_lsj_200e_coco.py │ │ │ │ ├── mask_rcnn_r50_fpn_amp_1x_coco.py │ │ │ │ ├── mask_rcnn_r50_fpn_ms_poly_-3x_coco.py │ │ │ │ ├── mask_rcnn_r50_fpn_poly_1x_coco.py │ │ │ │ ├── mask_rcnn_x101_32x4d_fpn_1x_coco.py │ │ │ │ ├── mask_rcnn_x101_32x4d_fpn_2x_coco.py │ │ │ │ ├── mask_rcnn_x101_32x4d_fpn_ms_poly_3x_coco.py │ │ │ │ ├── mask_rcnn_x101_32x8d_fpn_1x_coco.py │ │ │ │ ├── mask_rcnn_x101_32x8d_fpn_ms_poly_1x_coco.py │ │ │ │ ├── mask_rcnn_x101_32x8d_fpn_ms_poly_3x_coco.py │ │ │ │ ├── mask_rcnn_x101_64_4d_fpn_1x_coco.py │ │ │ │ ├── mask_rcnn_x101_64x4d_fpn_2x_coco.py │ │ │ │ └── mask_rcnn_x101_64x4d_fpn_ms_poly_3x_coco.py │ │ │ ├── maskformer │ │ │ │ ├── maskformer_r50_ms_16xb1_75e_coco.py │ │ │ │ └── maskformer_swin_l_p4_w12_64xb1_ms_300e_coco.py │ │ │ ├── panoptic_fpn │ │ │ │ └── panoptic_fpn_r50_fpn_1x_coco.py │ │ │ ├── qdtrack │ │ │ │ ├── qdtrack_faster_rcnn_r50_fpn_4e_base.py │ │ │ │ └── qdtrack_faster_rcnn_r50_fpn_8xb2-4e_mot17halftrain_test-mot17halfval.py │ │ │ ├── retinanet │ │ │ │ ├── retinanet_r50_fpn_1x_coco.py │ │ │ │ └── retinanet_tta.py │ │ │ └── rtmdet │ │ │ │ ├── rtmdet_ins_l_8xb32_300e_coco.py │ │ │ │ ├── rtmdet_ins_m_8xb32_300e_coco.py │ │ │ │ ├── rtmdet_ins_s_8xb32_300e_coco.py │ │ │ │ ├── rtmdet_ins_tiny_8xb32_300e_coco.py │ │ │ │ ├── rtmdet_ins_x_8xb16_300e_coco.py │ │ │ │ ├── rtmdet_l_8xb32_300e_coco.py │ │ │ │ ├── rtmdet_m_8xb32_300e_coco.py │ │ │ │ ├── rtmdet_s_8xb32_300e_coco.py │ │ │ │ ├── rtmdet_tiny_8xb32_300e_coco.py │ │ │ │ ├── rtmdet_tta.py │ │ │ │ └── rtmdet_x_8xb32_300e_coco.py │ │ ├── datasets │ │ │ ├── __init__.py │ │ │ ├── ade20k.py │ │ │ ├── api_wrappers │ │ │ │ ├── __init__.py │ │ │ │ ├── coco_api.py │ │ │ │ └── cocoeval_mp.py │ │ │ ├── base_det_dataset.py │ │ │ ├── base_semseg_dataset.py │ │ │ ├── base_video_dataset.py │ │ │ ├── cityscapes.py │ │ │ ├── coco.py │ │ │ ├── coco_caption.py │ │ │ ├── coco_panoptic.py │ │ │ ├── coco_semantic.py │ │ │ ├── crowdhuman.py │ │ │ ├── dataset_wrappers.py │ │ │ ├── deepfashion.py │ │ │ ├── dsdl.py │ │ │ ├── isaid.py │ │ │ ├── lvis.py │ │ │ ├── mot_challenge_dataset.py │ │ │ ├── objects365.py │ │ │ ├── openimages.py │ │ │ ├── refcoco.py │ │ │ ├── reid_dataset.py │ │ │ ├── samplers │ │ │ │ ├── __init__.py │ │ │ │ ├── batch_sampler.py │ │ │ │ ├── class_aware_sampler.py │ │ │ │ ├── multi_data_sampler.py │ │ │ │ ├── multi_source_sampler.py │ │ │ │ └── track_img_sampler.py │ │ │ ├── transforms │ │ │ │ ├── __init__.py │ │ │ │ ├── augment_wrappers.py │ │ │ │ ├── colorspace.py │ │ │ │ ├── formatting.py │ │ │ │ ├── frame_sampling.py │ │ │ │ ├── geometric.py │ │ │ │ ├── instaboost.py │ │ │ │ ├── loading.py │ │ │ │ ├── transformers_glip.py │ │ │ │ ├── transforms.py │ │ │ │ └── wrappers.py │ │ │ ├── utils.py │ │ │ ├── v3det.py │ │ │ ├── voc.py │ │ │ ├── wider_face.py │ │ │ ├── xml_style.py │ │ │ └── youtube_vis_dataset.py │ │ ├── engine │ │ │ ├── __init__.py │ │ │ ├── hooks │ │ │ │ ├── __init__.py │ │ │ │ ├── checkloss_hook.py │ │ │ │ ├── mean_teacher_hook.py │ │ │ │ ├── memory_profiler_hook.py │ │ │ │ ├── num_class_check_hook.py │ │ │ │ ├── pipeline_switch_hook.py │ │ │ │ ├── set_epoch_info_hook.py │ │ │ │ ├── sync_norm_hook.py │ │ │ │ ├── utils.py │ │ │ │ ├── visualization_hook.py │ │ │ │ └── yolox_mode_switch_hook.py │ │ │ ├── optimizers │ │ │ │ ├── __init__.py │ │ │ │ └── layer_decay_optimizer_constructor.py │ │ │ ├── runner │ │ │ │ ├── __init__.py │ │ │ │ └── loops.py │ │ │ └── schedulers │ │ │ │ ├── __init__.py │ │ │ │ └── quadratic_warmup.py │ │ ├── evaluation │ │ │ ├── __init__.py │ │ │ ├── functional │ │ │ │ ├── __init__.py │ │ │ │ ├── bbox_overlaps.py │ │ │ │ ├── cityscapes_utils.py │ │ │ │ ├── class_names.py │ │ │ │ ├── mean_ap.py │ │ │ │ ├── panoptic_utils.py │ │ │ │ ├── recall.py │ │ │ │ ├── ytvis.py │ │ │ │ └── ytviseval.py │ │ │ └── metrics │ │ │ │ ├── __init__.py │ │ │ │ ├── base_video_metric.py │ │ │ │ ├── cityscapes_metric.py │ │ │ │ ├── coco_caption_metric.py │ │ │ │ ├── coco_metric.py │ │ │ │ ├── coco_occluded_metric.py │ │ │ │ ├── coco_panoptic_metric.py │ │ │ │ ├── coco_video_metric.py │ │ │ │ ├── crowdhuman_metric.py │ │ │ │ ├── dump_det_results.py │ │ │ │ ├── dump_proposals_metric.py │ │ │ │ ├── lvis_metric.py │ │ │ │ ├── mot_challenge_metric.py │ │ │ │ ├── openimages_metric.py │ │ │ │ ├── refseg_metric.py │ │ │ │ ├── reid_metric.py │ │ │ │ ├── semseg_metric.py │ │ │ │ ├── voc_metric.py │ │ │ │ └── youtube_vis_metric.py │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── backbones │ │ │ │ ├── __init__.py │ │ │ │ ├── csp_darknet.py │ │ │ │ ├── cspnext.py │ │ │ │ ├── darknet.py │ │ │ │ ├── detectors_resnet.py │ │ │ │ ├── detectors_resnext.py │ │ │ │ ├── efficientnet.py │ │ │ │ ├── hourglass.py │ │ │ │ ├── hrnet.py │ │ │ │ ├── mobilenet_v2.py │ │ │ │ ├── pvt.py │ │ │ │ ├── regnet.py │ │ │ │ ├── res2net.py │ │ │ │ ├── resnest.py │ │ │ │ ├── resnet.py │ │ │ │ ├── resnext.py │ │ │ │ ├── ssd_vgg.py │ │ │ │ ├── swin.py │ │ │ │ └── trident_resnet.py │ │ │ ├── data_preprocessors │ │ │ │ ├── __init__.py │ │ │ │ ├── data_preprocessor.py │ │ │ │ ├── reid_data_preprocessor.py │ │ │ │ └── track_data_preprocessor.py │ │ │ ├── dense_heads │ │ │ │ ├── __init__.py │ │ │ │ ├── anchor_free_head.py │ │ │ │ ├── anchor_head.py │ │ │ │ ├── atss_head.py │ │ │ │ ├── atss_vlfusion_head.py │ │ │ │ ├── autoassign_head.py │ │ │ │ ├── base_dense_head.py │ │ │ │ ├── base_mask_head.py │ │ │ │ ├── boxinst_head.py │ │ │ │ ├── cascade_rpn_head.py │ │ │ │ ├── centernet_head.py │ │ │ │ ├── centernet_update_head.py │ │ │ │ ├── centripetal_head.py │ │ │ │ ├── condinst_head.py │ │ │ │ ├── conditional_detr_head.py │ │ │ │ ├── corner_head.py │ │ │ │ ├── dab_detr_head.py │ │ │ │ ├── ddod_head.py │ │ │ │ ├── ddq_detr_head.py │ │ │ │ ├── deformable_detr_head.py │ │ │ │ ├── dense_test_mixins.py │ │ │ │ ├── detr_head.py │ │ │ │ ├── dino_head.py │ │ │ │ ├── embedding_rpn_head.py │ │ │ │ ├── fcos_head.py │ │ │ │ ├── fovea_head.py │ │ │ │ ├── free_anchor_retina_head.py │ │ │ │ ├── fsaf_head.py │ │ │ │ ├── ga_retina_head.py │ │ │ │ ├── ga_rpn_head.py │ │ │ │ ├── gfl_head.py │ │ │ │ ├── grounding_dino_head.py │ │ │ │ ├── guided_anchor_head.py │ │ │ │ ├── lad_head.py │ │ │ │ ├── ld_head.py │ │ │ │ ├── mask2former_head.py │ │ │ │ ├── maskformer_head.py │ │ │ │ ├── nasfcos_head.py │ │ │ │ ├── paa_head.py │ │ │ │ ├── pisa_retinanet_head.py │ │ │ │ ├── pisa_ssd_head.py │ │ │ │ ├── reppoints_head.py │ │ │ │ ├── retina_head.py │ │ │ │ ├── retina_sepbn_head.py │ │ │ │ ├── rpn_head.py │ │ │ │ ├── rtmdet_head.py │ │ │ │ ├── rtmdet_ins_head.py │ │ │ │ ├── sabl_retina_head.py │ │ │ │ ├── solo_head.py │ │ │ │ ├── solov2_head.py │ │ │ │ ├── ssd_head.py │ │ │ │ ├── tood_head.py │ │ │ │ ├── vfnet_head.py │ │ │ │ ├── yolact_head.py │ │ │ │ ├── yolo_head.py │ │ │ │ ├── yolof_head.py │ │ │ │ └── yolox_head.py │ │ │ ├── detectors │ │ │ │ ├── __init__.py │ │ │ │ ├── atss.py │ │ │ │ ├── autoassign.py │ │ │ │ ├── base.py │ │ │ │ ├── base_detr.py │ │ │ │ ├── boxinst.py │ │ │ │ ├── cascade_rcnn.py │ │ │ │ ├── centernet.py │ │ │ │ ├── condinst.py │ │ │ │ ├── conditional_detr.py │ │ │ │ ├── cornernet.py │ │ │ │ ├── crowddet.py │ │ │ │ ├── d2_wrapper.py │ │ │ │ ├── dab_detr.py │ │ │ │ ├── ddod.py │ │ │ │ ├── ddq_detr.py │ │ │ │ ├── deformable_detr.py │ │ │ │ ├── detr.py │ │ │ │ ├── dino.py │ │ │ │ ├── fast_rcnn.py │ │ │ │ ├── faster_rcnn.py │ │ │ │ ├── fcos.py │ │ │ │ ├── fovea.py │ │ │ │ ├── fsaf.py │ │ │ │ ├── gfl.py │ │ │ │ ├── glip.py │ │ │ │ ├── grid_rcnn.py │ │ │ │ ├── grounding_dino.py │ │ │ │ ├── htc.py │ │ │ │ ├── kd_one_stage.py │ │ │ │ ├── lad.py │ │ │ │ ├── mask2former.py │ │ │ │ ├── mask_rcnn.py │ │ │ │ ├── mask_scoring_rcnn.py │ │ │ │ ├── maskformer.py │ │ │ │ ├── nasfcos.py │ │ │ │ ├── paa.py │ │ │ │ ├── panoptic_fpn.py │ │ │ │ ├── panoptic_two_stage_segmentor.py │ │ │ │ ├── point_rend.py │ │ │ │ ├── queryinst.py │ │ │ │ ├── reppoints_detector.py │ │ │ │ ├── retinanet.py │ │ │ │ ├── rpn.py │ │ │ │ ├── rtmdet.py │ │ │ │ ├── scnet.py │ │ │ │ ├── semi_base.py │ │ │ │ ├── single_stage.py │ │ │ │ ├── single_stage_instance_seg.py │ │ │ │ ├── soft_teacher.py │ │ │ │ ├── solo.py │ │ │ │ ├── solov2.py │ │ │ │ ├── sparse_rcnn.py │ │ │ │ ├── tood.py │ │ │ │ ├── trident_faster_rcnn.py │ │ │ │ ├── two_stage.py │ │ │ │ ├── vfnet.py │ │ │ │ ├── yolact.py │ │ │ │ ├── yolo.py │ │ │ │ ├── yolof.py │ │ │ │ └── yolox.py │ │ │ ├── language_models │ │ │ │ ├── __init__.py │ │ │ │ └── bert.py │ │ │ ├── layers │ │ │ │ ├── __init__.py │ │ │ │ ├── activations.py │ │ │ │ ├── bbox_nms.py │ │ │ │ ├── brick_wrappers.py │ │ │ │ ├── conv_upsample.py │ │ │ │ ├── csp_layer.py │ │ │ │ ├── dropblock.py │ │ │ │ ├── ema.py │ │ │ │ ├── inverted_residual.py │ │ │ │ ├── matrix_nms.py │ │ │ │ ├── msdeformattn_pixel_decoder.py │ │ │ │ ├── normed_predictor.py │ │ │ │ ├── pixel_decoder.py │ │ │ │ ├── positional_encoding.py │ │ │ │ ├── res_layer.py │ │ │ │ ├── se_layer.py │ │ │ │ └── transformer │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── conditional_detr_layers.py │ │ │ │ │ ├── dab_detr_layers.py │ │ │ │ │ ├── ddq_detr_layers.py │ │ │ │ │ ├── deformable_detr_layers.py │ │ │ │ │ ├── detr_layers.py │ │ │ │ │ ├── dino_layers.py │ │ │ │ │ ├── grounding_dino_layers.py │ │ │ │ │ ├── mask2former_layers.py │ │ │ │ │ └── utils.py │ │ │ ├── losses │ │ │ │ ├── __init__.py │ │ │ │ ├── accuracy.py │ │ │ │ ├── ae_loss.py │ │ │ │ ├── balanced_l1_loss.py │ │ │ │ ├── cross_entropy_loss.py │ │ │ │ ├── ddq_detr_aux_loss.py │ │ │ │ ├── dice_loss.py │ │ │ │ ├── eqlv2_loss.py │ │ │ │ ├── focal_loss.py │ │ │ │ ├── gaussian_focal_loss.py │ │ │ │ ├── gfocal_loss.py │ │ │ │ ├── ghm_loss.py │ │ │ │ ├── iou_loss.py │ │ │ │ ├── kd_loss.py │ │ │ │ ├── l2_loss.py │ │ │ │ ├── margin_loss.py │ │ │ │ ├── mse_loss.py │ │ │ │ ├── multipos_cross_entropy_loss.py │ │ │ │ ├── pisa_loss.py │ │ │ │ ├── seesaw_loss.py │ │ │ │ ├── smooth_l1_loss.py │ │ │ │ ├── triplet_loss.py │ │ │ │ ├── utils.py │ │ │ │ └── varifocal_loss.py │ │ │ ├── mot │ │ │ │ ├── __init__.py │ │ │ │ ├── base.py │ │ │ │ ├── bytetrack.py │ │ │ │ ├── deep_sort.py │ │ │ │ ├── ocsort.py │ │ │ │ ├── qdtrack.py │ │ │ │ └── strongsort.py │ │ │ ├── necks │ │ │ │ ├── __init__.py │ │ │ │ ├── bfp.py │ │ │ │ ├── channel_mapper.py │ │ │ │ ├── cspnext_pafpn.py │ │ │ │ ├── ct_resnet_neck.py │ │ │ │ ├── dilated_encoder.py │ │ │ │ ├── dyhead.py │ │ │ │ ├── fpg.py │ │ │ │ ├── fpn.py │ │ │ │ ├── fpn_carafe.py │ │ │ │ ├── fpn_dropblock.py │ │ │ │ ├── hrfpn.py │ │ │ │ ├── nas_fpn.py │ │ │ │ ├── nasfcos_fpn.py │ │ │ │ ├── pafpn.py │ │ │ │ ├── rfp.py │ │ │ │ ├── ssd_neck.py │ │ │ │ ├── ssh.py │ │ │ │ ├── yolo_neck.py │ │ │ │ └── yolox_pafpn.py │ │ │ ├── reid │ │ │ │ ├── __init__.py │ │ │ │ ├── base_reid.py │ │ │ │ ├── fc_module.py │ │ │ │ ├── gap.py │ │ │ │ └── linear_reid_head.py │ │ │ ├── roi_heads │ │ │ │ ├── __init__.py │ │ │ │ ├── base_roi_head.py │ │ │ │ ├── bbox_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── bbox_head.py │ │ │ │ │ ├── convfc_bbox_head.py │ │ │ │ │ ├── dii_head.py │ │ │ │ │ ├── double_bbox_head.py │ │ │ │ │ ├── multi_instance_bbox_head.py │ │ │ │ │ ├── sabl_head.py │ │ │ │ │ └── scnet_bbox_head.py │ │ │ │ ├── cascade_roi_head.py │ │ │ │ ├── double_roi_head.py │ │ │ │ ├── dynamic_roi_head.py │ │ │ │ ├── grid_roi_head.py │ │ │ │ ├── htc_roi_head.py │ │ │ │ ├── mask_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── coarse_mask_head.py │ │ │ │ │ ├── dynamic_mask_head.py │ │ │ │ │ ├── fcn_mask_head.py │ │ │ │ │ ├── feature_relay_head.py │ │ │ │ │ ├── fused_semantic_head.py │ │ │ │ │ ├── global_context_head.py │ │ │ │ │ ├── grid_head.py │ │ │ │ │ ├── htc_mask_head.py │ │ │ │ │ ├── mask_point_head.py │ │ │ │ │ ├── maskiou_head.py │ │ │ │ │ ├── scnet_mask_head.py │ │ │ │ │ └── scnet_semantic_head.py │ │ │ │ ├── mask_scoring_roi_head.py │ │ │ │ ├── multi_instance_roi_head.py │ │ │ │ ├── pisa_roi_head.py │ │ │ │ ├── point_rend_roi_head.py │ │ │ │ ├── roi_extractors │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_roi_extractor.py │ │ │ │ │ ├── generic_roi_extractor.py │ │ │ │ │ └── single_level_roi_extractor.py │ │ │ │ ├── scnet_roi_head.py │ │ │ │ ├── shared_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── res_layer.py │ │ │ │ ├── sparse_roi_head.py │ │ │ │ ├── standard_roi_head.py │ │ │ │ ├── test_mixins.py │ │ │ │ └── trident_roi_head.py │ │ │ ├── seg_heads │ │ │ │ ├── __init__.py │ │ │ │ ├── base_semantic_head.py │ │ │ │ ├── panoptic_fpn_head.py │ │ │ │ └── panoptic_fusion_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_panoptic_fusion_head.py │ │ │ │ │ ├── heuristic_fusion_head.py │ │ │ │ │ └── maskformer_fusion_head.py │ │ │ ├── task_modules │ │ │ │ ├── __init__.py │ │ │ │ ├── assigners │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── approx_max_iou_assigner.py │ │ │ │ │ ├── assign_result.py │ │ │ │ │ ├── atss_assigner.py │ │ │ │ │ ├── base_assigner.py │ │ │ │ │ ├── center_region_assigner.py │ │ │ │ │ ├── dynamic_soft_label_assigner.py │ │ │ │ │ ├── grid_assigner.py │ │ │ │ │ ├── hungarian_assigner.py │ │ │ │ │ ├── iou2d_calculator.py │ │ │ │ │ ├── match_cost.py │ │ │ │ │ ├── max_iou_assigner.py │ │ │ │ │ ├── multi_instance_assigner.py │ │ │ │ │ ├── point_assigner.py │ │ │ │ │ ├── region_assigner.py │ │ │ │ │ ├── sim_ota_assigner.py │ │ │ │ │ ├── task_aligned_assigner.py │ │ │ │ │ ├── topk_hungarian_assigner.py │ │ │ │ │ └── uniform_assigner.py │ │ │ │ ├── builder.py │ │ │ │ ├── coders │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_bbox_coder.py │ │ │ │ │ ├── bucketing_bbox_coder.py │ │ │ │ │ ├── delta_xywh_bbox_coder.py │ │ │ │ │ ├── distance_point_bbox_coder.py │ │ │ │ │ ├── legacy_delta_xywh_bbox_coder.py │ │ │ │ │ ├── pseudo_bbox_coder.py │ │ │ │ │ ├── tblr_bbox_coder.py │ │ │ │ │ └── yolo_bbox_coder.py │ │ │ │ ├── prior_generators │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── anchor_generator.py │ │ │ │ │ ├── point_generator.py │ │ │ │ │ └── utils.py │ │ │ │ ├── samplers │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_sampler.py │ │ │ │ │ ├── combined_sampler.py │ │ │ │ │ ├── instance_balanced_pos_sampler.py │ │ │ │ │ ├── iou_balanced_neg_sampler.py │ │ │ │ │ ├── mask_pseudo_sampler.py │ │ │ │ │ ├── mask_sampling_result.py │ │ │ │ │ ├── multi_instance_random_sampler.py │ │ │ │ │ ├── multi_instance_sampling_result.py │ │ │ │ │ ├── ohem_sampler.py │ │ │ │ │ ├── pseudo_sampler.py │ │ │ │ │ ├── random_sampler.py │ │ │ │ │ ├── sampling_result.py │ │ │ │ │ └── score_hlr_sampler.py │ │ │ │ └── tracking │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── aflink.py │ │ │ │ │ ├── camera_motion_compensation.py │ │ │ │ │ ├── interpolation.py │ │ │ │ │ ├── kalman_filter.py │ │ │ │ │ └── similarity.py │ │ │ ├── test_time_augs │ │ │ │ ├── __init__.py │ │ │ │ ├── det_tta.py │ │ │ │ └── merge_augs.py │ │ │ ├── trackers │ │ │ │ ├── __init__.py │ │ │ │ ├── base_tracker.py │ │ │ │ ├── byte_tracker.py │ │ │ │ ├── masktrack_rcnn_tracker.py │ │ │ │ ├── ocsort_tracker.py │ │ │ │ ├── quasi_dense_tracker.py │ │ │ │ ├── sort_tracker.py │ │ │ │ └── strongsort_tracker.py │ │ │ ├── tracking_heads │ │ │ │ ├── __init__.py │ │ │ │ ├── mask2former_track_head.py │ │ │ │ ├── quasi_dense_embed_head.py │ │ │ │ ├── quasi_dense_track_head.py │ │ │ │ ├── roi_embed_head.py │ │ │ │ └── roi_track_head.py │ │ │ ├── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── gaussian_target.py │ │ │ │ ├── image.py │ │ │ │ ├── make_divisible.py │ │ │ │ ├── misc.py │ │ │ │ ├── panoptic_gt_processing.py │ │ │ │ ├── point_sample.py │ │ │ │ ├── vlfuse_helper.py │ │ │ │ └── wbf.py │ │ │ └── vis │ │ │ │ ├── __init__.py │ │ │ │ ├── mask2former_vis.py │ │ │ │ └── masktrack_rcnn.py │ │ ├── registry.py │ │ ├── structures │ │ │ ├── __init__.py │ │ │ ├── bbox │ │ │ │ ├── __init__.py │ │ │ │ ├── base_boxes.py │ │ │ │ ├── bbox_overlaps.py │ │ │ │ ├── box_type.py │ │ │ │ ├── horizontal_boxes.py │ │ │ │ └── transforms.py │ │ │ ├── det_data_sample.py │ │ │ ├── mask │ │ │ │ ├── __init__.py │ │ │ │ ├── mask_target.py │ │ │ │ ├── structures.py │ │ │ │ └── utils.py │ │ │ ├── reid_data_sample.py │ │ │ └── track_data_sample.py │ │ ├── testing │ │ │ ├── __init__.py │ │ │ ├── _fast_stop_training_hook.py │ │ │ └── _utils.py │ │ ├── utils │ │ │ ├── __init__.py │ │ │ ├── benchmark.py │ │ │ ├── collect_env.py │ │ │ ├── compat_config.py │ │ │ ├── contextmanagers.py │ │ │ ├── dist_utils.py │ │ │ ├── large_image.py │ │ │ ├── logger.py │ │ │ ├── memory.py │ │ │ ├── misc.py │ │ │ ├── mot_error_visualize.py │ │ │ ├── profiling.py │ │ │ ├── replace_cfg_vals.py │ │ │ ├── setup_env.py │ │ │ ├── split_batch.py │ │ │ ├── typing_utils.py │ │ │ ├── util_mixins.py │ │ │ └── util_random.py │ │ ├── version.py │ │ └── visualization │ │ │ ├── __init__.py │ │ │ ├── local_visualizer.py │ │ │ └── palette.py │ ├── model-index.yml │ ├── pytest.ini │ ├── requirements.txt │ ├── requirements │ │ ├── albu.txt │ │ ├── build.txt │ │ ├── docs.txt │ │ ├── mminstall.txt │ │ ├── multimodal.txt │ │ ├── optional.txt │ │ ├── readthedocs.txt │ │ ├── runtime.txt │ │ ├── tests.txt │ │ └── tracking.txt │ ├── setup.cfg │ ├── setup.py │ └── tools │ │ ├── analysis_tools │ │ ├── analyze_logs.py │ │ ├── analyze_results.py │ │ ├── benchmark.py │ │ ├── browse_dataset.py │ │ ├── coco_error_analysis.py │ │ ├── coco_occluded_separated_recall.py │ │ ├── confusion_matrix.py │ │ ├── eval_metric.py │ │ ├── fuse_results.py │ │ ├── get_flops.py │ │ ├── mot │ │ │ ├── browse_dataset.py │ │ │ ├── dist_mot_search.sh │ │ │ ├── mot_error_visualize.py │ │ │ ├── mot_param_search.py │ │ │ └── slurm_mot_search.sh │ │ ├── optimize_anchors.py │ │ ├── robustness_eval.py │ │ └── test_robustness.py │ │ ├── dataset_converters │ │ ├── ade20k2coco.py │ │ ├── cityscapes.py │ │ ├── coco_stuff164k.py │ │ ├── crowdhuman2coco.py │ │ ├── images2coco.py │ │ ├── mot2coco.py │ │ ├── mot2reid.py │ │ ├── pascal_voc.py │ │ ├── prepare_coco_semantic_annos_from_panoptic_annos.py │ │ ├── scripts │ │ │ ├── preprocess_coco2017.sh │ │ │ ├── preprocess_voc2007.sh │ │ │ └── preprocess_voc2012.sh │ │ └── youtubevis2coco.py │ │ ├── deployment │ │ ├── mmdet2torchserve.py │ │ ├── mmdet_handler.py │ │ └── test_torchserver.py │ │ ├── dist_test.sh │ │ ├── dist_test_tracking.sh │ │ ├── dist_train.sh │ │ ├── misc │ │ ├── download_dataset.py │ │ ├── gen_coco_panoptic_test_info.py │ │ ├── get_crowdhuman_id_hw.py │ │ ├── get_image_metas.py │ │ ├── print_config.py │ │ └── split_coco.py │ │ ├── model_converters │ │ ├── detectron2_to_mmdet.py │ │ ├── detectron2pytorch.py │ │ ├── detic_to_mmdet.py │ │ ├── glip_to_mmdet.py │ │ ├── groundingdino_to_mmdet.py │ │ ├── publish_model.py │ │ ├── regnet2mmdet.py │ │ ├── selfsup2mmdet.py │ │ ├── swinv1_to_mmdet.py │ │ ├── upgrade_model_version.py │ │ └── upgrade_ssd_version.py │ │ ├── slurm_test.sh │ │ ├── slurm_test_tracking.sh │ │ ├── slurm_train.sh │ │ ├── test.py │ │ ├── test_tracking.py │ │ └── train.py ├── docs │ ├── DEPTH_README.md │ ├── NORMAL_README.md │ ├── POSE_README.md │ ├── PRETRAIN_README.md │ ├── SEG_README.md │ └── finetune │ │ └── NORMAL_README.md ├── engine │ ├── .gitignore │ ├── MANIFEST.in │ ├── mmengine │ │ ├── __init__.py │ │ ├── _strategy │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── colossalai.py │ │ │ ├── deepspeed.py │ │ │ ├── distributed.py │ │ │ ├── fsdp.py │ │ │ ├── single_device.py │ │ │ └── utils.py │ │ ├── analysis │ │ │ ├── __init__.py │ │ │ ├── complexity_analysis.py │ │ │ ├── jit_analysis.py │ │ │ ├── jit_handles.py │ │ │ └── print_helper.py │ │ ├── config │ │ │ ├── __init__.py │ │ │ ├── config.py │ │ │ ├── lazy.py │ │ │ └── utils.py │ │ ├── dataset │ │ │ ├── __init__.py │ │ │ ├── base_dataset.py │ │ │ ├── dataset_wrapper.py │ │ │ ├── sampler.py │ │ │ └── utils.py │ │ ├── device │ │ │ ├── __init__.py │ │ │ └── utils.py │ │ ├── dist │ │ │ ├── __init__.py │ │ │ ├── dist.py │ │ │ └── utils.py │ │ ├── evaluator │ │ │ ├── __init__.py │ │ │ ├── evaluator.py │ │ │ ├── metric.py │ │ │ └── utils.py │ │ ├── fileio │ │ │ ├── __init__.py │ │ │ ├── backends │ │ │ │ ├── __init__.py │ │ │ │ ├── base.py │ │ │ │ ├── http_backend.py │ │ │ │ ├── lmdb_backend.py │ │ │ │ ├── local_backend.py │ │ │ │ ├── memcached_backend.py │ │ │ │ ├── petrel_backend.py │ │ │ │ └── registry_utils.py │ │ │ ├── file_client.py │ │ │ ├── handlers │ │ │ │ ├── __init__.py │ │ │ │ ├── base.py │ │ │ │ ├── json_handler.py │ │ │ │ ├── pickle_handler.py │ │ │ │ ├── registry_utils.py │ │ │ │ └── yaml_handler.py │ │ │ ├── io.py │ │ │ └── parse.py │ │ ├── hooks │ │ │ ├── __init__.py │ │ │ ├── checkpoint_hook.py │ │ │ ├── early_stopping_hook.py │ │ │ ├── ema_hook.py │ │ │ ├── empty_cache_hook.py │ │ │ ├── hook.py │ │ │ ├── iter_timer_hook.py │ │ │ ├── logger_hook.py │ │ │ ├── naive_visualization_hook.py │ │ │ ├── param_scheduler_hook.py │ │ │ ├── profiler_hook.py │ │ │ ├── runtime_info_hook.py │ │ │ ├── sampler_seed_hook.py │ │ │ ├── sync_buffer_hook.py │ │ │ └── test_time_aug_hook.py │ │ ├── hub │ │ │ ├── __init__.py │ │ │ ├── deprecated.json │ │ │ ├── hub.py │ │ │ ├── mmcls.json │ │ │ ├── openmmlab.json │ │ │ └── torchvision_0.12.json │ │ ├── infer │ │ │ ├── __init__.py │ │ │ └── infer.py │ │ ├── logging │ │ │ ├── __init__.py │ │ │ ├── history_buffer.py │ │ │ ├── logger.py │ │ │ └── message_hub.py │ │ ├── model │ │ │ ├── __init__.py │ │ │ ├── averaged_model.py │ │ │ ├── base_model │ │ │ │ ├── __init__.py │ │ │ │ ├── base_model.py │ │ │ │ └── data_preprocessor.py │ │ │ ├── base_module.py │ │ │ ├── efficient_conv_bn_eval.py │ │ │ ├── test_time_aug.py │ │ │ ├── utils.py │ │ │ ├── weight_init.py │ │ │ └── wrappers │ │ │ │ ├── __init__.py │ │ │ │ ├── distributed.py │ │ │ │ ├── fully_sharded_distributed.py │ │ │ │ ├── seperate_distributed.py │ │ │ │ └── utils.py │ │ ├── optim │ │ │ ├── __init__.py │ │ │ ├── optimizer │ │ │ │ ├── __init__.py │ │ │ │ ├── amp_optimizer_wrapper.py │ │ │ │ ├── apex_optimizer_wrapper.py │ │ │ │ ├── base.py │ │ │ │ ├── builder.py │ │ │ │ ├── default_constructor.py │ │ │ │ ├── optimizer_wrapper.py │ │ │ │ ├── optimizer_wrapper_dict.py │ │ │ │ └── zero_optimizer.py │ │ │ └── scheduler │ │ │ │ ├── __init__.py │ │ │ │ ├── lr_scheduler.py │ │ │ │ ├── momentum_scheduler.py │ │ │ │ └── param_scheduler.py │ │ ├── registry │ │ │ ├── __init__.py │ │ │ ├── build_functions.py │ │ │ ├── default_scope.py │ │ │ ├── registry.py │ │ │ ├── root.py │ │ │ └── utils.py │ │ ├── runner │ │ │ ├── __init__.py │ │ │ ├── _flexible_runner.py │ │ │ ├── activation_checkpointing.py │ │ │ ├── amp.py │ │ │ ├── base_loop.py │ │ │ ├── checkpoint.py │ │ │ ├── log_processor.py │ │ │ ├── loops.py │ │ │ ├── priority.py │ │ │ ├── runner.py │ │ │ └── utils.py │ │ ├── structures │ │ │ ├── __init__.py │ │ │ ├── base_data_element.py │ │ │ ├── instance_data.py │ │ │ ├── label_data.py │ │ │ └── pixel_data.py │ │ ├── testing │ │ │ ├── __init__.py │ │ │ ├── _internal │ │ │ │ ├── __init__.py │ │ │ │ └── distributed.py │ │ │ ├── compare.py │ │ │ └── runner_test_case.py │ │ ├── utils │ │ │ ├── __init__.py │ │ │ ├── dl_utils │ │ │ │ ├── __init__.py │ │ │ │ ├── collect_env.py │ │ │ │ ├── hub.py │ │ │ │ ├── misc.py │ │ │ │ ├── parrots_wrapper.py │ │ │ │ ├── setup_env.py │ │ │ │ ├── time_counter.py │ │ │ │ ├── torch_ops.py │ │ │ │ ├── trace.py │ │ │ │ └── visualize.py │ │ │ ├── manager.py │ │ │ ├── misc.py │ │ │ ├── package_utils.py │ │ │ ├── path.py │ │ │ ├── progressbar.py │ │ │ ├── progressbar_rich.py │ │ │ ├── timer.py │ │ │ └── version_utils.py │ │ ├── version.py │ │ └── visualization │ │ │ ├── __init__.py │ │ │ ├── utils.py │ │ │ ├── vis_backend.py │ │ │ └── visualizer.py │ ├── pytest.ini │ ├── requirements.txt │ ├── requirements │ │ ├── docs.txt │ │ ├── runtime.txt │ │ └── tests.txt │ ├── setup.cfg │ └── setup.py ├── git_script.sh ├── lite │ ├── README.md │ ├── demo │ │ ├── adhoc_image_dataset.py │ │ ├── classes_and_palettes.py │ │ ├── detector_utils.py │ │ ├── extract_feature.py │ │ ├── pose_utils.py │ │ ├── vis_depth.py │ │ ├── vis_normal.py │ │ ├── vis_pose.py │ │ ├── vis_seg.py │ │ └── worker_pool.py │ ├── docs │ │ ├── DEPTH_README.md │ │ ├── NORMAL_README.md │ │ ├── POSE_README.md │ │ ├── PRETRAIN_README.md │ │ └── SEG_README.md │ └── scripts │ │ └── demo │ │ ├── bfloat16 │ │ ├── depth.sh │ │ ├── extract_feature.sh │ │ ├── normal.sh │ │ ├── pose_keypoints133.sh │ │ ├── pose_keypoints17.sh │ │ ├── pose_keypoints308.sh │ │ └── seg.sh │ │ ├── float16 │ │ ├── depth.sh │ │ ├── extract_feature.sh │ │ ├── normal.sh │ │ ├── pose_keypoints133.sh │ │ ├── pose_keypoints17.sh │ │ ├── pose_keypoints308.sh │ │ └── seg.sh │ │ └── torchscript │ │ ├── depth.sh │ │ ├── extract_feature.sh │ │ ├── normal.sh │ │ ├── pose_keypoints133.sh │ │ ├── pose_keypoints17.sh │ │ ├── pose_keypoints308.sh │ │ └── seg.sh ├── pose │ ├── .gitignore │ ├── MANIFEST.in │ ├── configs │ │ ├── _base_ │ │ │ ├── datasets │ │ │ │ ├── 300w.py │ │ │ │ ├── aflw.py │ │ │ │ ├── aic.py │ │ │ │ ├── aic2coco.py │ │ │ │ ├── aic2goliath.py │ │ │ │ ├── aic2goliath3d.py │ │ │ │ ├── ak.py │ │ │ │ ├── animalpose.py │ │ │ │ ├── ap10k.py │ │ │ │ ├── atrw.py │ │ │ │ ├── campus.py │ │ │ │ ├── coco.py │ │ │ │ ├── coco_openpose.py │ │ │ │ ├── coco_wholebody.py │ │ │ │ ├── coco_wholebody2goliath.py │ │ │ │ ├── coco_wholebody2goliath3d.py │ │ │ │ ├── coco_wholebody_face.py │ │ │ │ ├── coco_wholebody_hand.py │ │ │ │ ├── coco_wholebody_openpose.py │ │ │ │ ├── cofw.py │ │ │ │ ├── crowdpose.py │ │ │ │ ├── crowdpose2coco.py │ │ │ │ ├── crowdpose2goliath.py │ │ │ │ ├── crowdpose2goliath3d.py │ │ │ │ ├── deepfashion2.py │ │ │ │ ├── deepfashion_full.py │ │ │ │ ├── deepfashion_lower.py │ │ │ │ ├── deepfashion_upper.py │ │ │ │ ├── fly.py │ │ │ │ ├── freihand2d.py │ │ │ │ ├── goliath.py │ │ │ │ ├── goliath3d.py │ │ │ │ ├── h36m.py │ │ │ │ ├── halpe.py │ │ │ │ ├── halpe26.py │ │ │ │ ├── horse10.py │ │ │ │ ├── humanart.py │ │ │ │ ├── humanart_aic.py │ │ │ │ ├── interhand2d.py │ │ │ │ ├── interhand3d.py │ │ │ │ ├── jhmdb.py │ │ │ │ ├── lapa.py │ │ │ │ ├── locust.py │ │ │ │ ├── macaque.py │ │ │ │ ├── mhp.py │ │ │ │ ├── mpi_inf_3dhp.py │ │ │ │ ├── mpii.py │ │ │ │ ├── mpii2coco.py │ │ │ │ ├── mpii2goliath.py │ │ │ │ ├── mpii2goliath3d.py │ │ │ │ ├── mpii_trb.py │ │ │ │ ├── ochuman.py │ │ │ │ ├── onehand10k.py │ │ │ │ ├── panoptic_body3d.py │ │ │ │ ├── panoptic_hand2d.py │ │ │ │ ├── posetrack18.py │ │ │ │ ├── rhd2d.py │ │ │ │ ├── shelf.py │ │ │ │ ├── wflw.py │ │ │ │ └── zebra.py │ │ │ └── default_runtime.py │ │ └── sapiens_pose │ │ │ ├── coco_wholebody │ │ │ ├── sapiens_0.3b-210e_coco_wholebody-1024x768.py │ │ │ ├── sapiens_0.6b-210e_coco_wholebody-1024x768.py │ │ │ ├── sapiens_1b-210e_coco_wholebody-1024x768.py │ │ │ └── sapiens_2b-210e_coco_wholebody-1024x768.py │ │ │ └── goliath │ │ │ ├── sapiens_0.3b-210e_goliath-1024x768.py │ │ │ ├── sapiens_0.6b-210e_goliath-1024x768.py │ │ │ ├── sapiens_1b-210e_goliath-1024x768.py │ │ │ └── sapiens_2b-210e_goliath-1024x768.py │ ├── demo │ │ ├── body3d_pose_lifter_demo.py │ │ ├── bottomup_demo.py │ │ ├── demo_vis.py │ │ ├── docs │ │ │ ├── en │ │ │ │ ├── 2d_animal_demo.md │ │ │ │ ├── 2d_face_demo.md │ │ │ │ ├── 2d_hand_demo.md │ │ │ │ ├── 2d_human_pose_demo.md │ │ │ │ ├── 2d_wholebody_pose_demo.md │ │ │ │ ├── 3d_human_pose_demo.md │ │ │ │ ├── mmdet_modelzoo.md │ │ │ │ └── webcam_api_demo.md │ │ │ └── zh_cn │ │ │ │ ├── 2d_animal_demo.md │ │ │ │ ├── 2d_face_demo.md │ │ │ │ ├── 2d_hand_demo.md │ │ │ │ ├── 2d_human_pose_demo.md │ │ │ │ ├── 2d_wholebody_pose_demo.md │ │ │ │ ├── 3d_human_pose_demo.md │ │ │ │ ├── mmdet_modelzoo.md │ │ │ │ └── webcam_api_demo.md │ │ ├── image_demo.py │ │ ├── inferencer_demo.py │ │ ├── mmdetection_cfg │ │ │ ├── rtmdet_m_640-8xb32_coco-person.py │ │ │ └── rtmdet_m_640-8xb32_coco-person_no_nms.py │ │ ├── timer.py │ │ ├── topdown_demo_with_mmdet.py │ │ └── video_demo_optimized.py │ ├── mmpose │ │ ├── __init__.py │ │ ├── apis │ │ │ ├── __init__.py │ │ │ ├── inference.py │ │ │ ├── inference_3d.py │ │ │ ├── inference_tracking.py │ │ │ └── inferencers │ │ │ │ ├── __init__.py │ │ │ │ ├── base_mmpose_inferencer.py │ │ │ │ ├── mmpose_inferencer.py │ │ │ │ ├── pose2d_inferencer.py │ │ │ │ ├── pose3d_inferencer.py │ │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── default_det_models.py │ │ │ │ └── get_model_alias.py │ │ ├── codecs │ │ │ ├── __init__.py │ │ │ ├── associative_embedding.py │ │ │ ├── base.py │ │ │ ├── decoupled_heatmap.py │ │ │ ├── image_pose_lifting.py │ │ │ ├── integral_regression_label.py │ │ │ ├── megvii_heatmap.py │ │ │ ├── msra_heatmap.py │ │ │ ├── regression_label.py │ │ │ ├── simcc_label.py │ │ │ ├── spr.py │ │ │ ├── udp_heatmap.py │ │ │ ├── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── gaussian_heatmap.py │ │ │ │ ├── instance_property.py │ │ │ │ ├── offset_heatmap.py │ │ │ │ ├── post_processing.py │ │ │ │ └── refinement.py │ │ │ └── video_pose_lifting.py │ │ ├── configs │ │ │ ├── _base_ │ │ │ │ └── default_runtime.py │ │ │ └── body_2d_keypoint │ │ │ │ ├── rtmpose │ │ │ │ └── coco │ │ │ │ │ ├── rtmpose_m_8xb256-420e_coco-256x192.py │ │ │ │ │ └── rtmpose_s_8xb256_420e_aic_coco_256x192.py │ │ │ │ └── topdown_heatmap │ │ │ │ └── coco │ │ │ │ └── td-hm_hrnet-w48_udp-8xb32-210e_coco-256x192.py │ │ ├── datasets │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ ├── dataset_wrappers.py │ │ │ ├── datasets │ │ │ │ ├── __init__.py │ │ │ │ ├── animal │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── animalkingdom_dataset.py │ │ │ │ │ ├── animalpose_dataset.py │ │ │ │ │ ├── ap10k_dataset.py │ │ │ │ │ ├── atrw_dataset.py │ │ │ │ │ ├── fly_dataset.py │ │ │ │ │ ├── horse10_dataset.py │ │ │ │ │ ├── locust_dataset.py │ │ │ │ │ ├── macaque_dataset.py │ │ │ │ │ └── zebra_dataset.py │ │ │ │ ├── base │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_coco_style_dataset.py │ │ │ │ │ └── base_mocap_dataset.py │ │ │ │ ├── body │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── aic2coco_dataset.py │ │ │ │ │ ├── aic2goliath_dataset.py │ │ │ │ │ ├── aic_dataset.py │ │ │ │ │ ├── coco2goliath_dataset.py │ │ │ │ │ ├── coco_dataset.py │ │ │ │ │ ├── crowdpose2coco_dataset.py │ │ │ │ │ ├── crowdpose2goliath_dataset.py │ │ │ │ │ ├── crowdpose_dataset.py │ │ │ │ │ ├── goliath3d_eval_dataset.py │ │ │ │ │ ├── goliath_dataset.py │ │ │ │ │ ├── goliath_eval_dataset.py │ │ │ │ │ ├── humanart_dataset.py │ │ │ │ │ ├── jhmdb_dataset.py │ │ │ │ │ ├── mhp_dataset.py │ │ │ │ │ ├── mpii2coco_dataset.py │ │ │ │ │ ├── mpii2goliath_dataset.py │ │ │ │ │ ├── mpii_dataset.py │ │ │ │ │ ├── mpii_trb_dataset.py │ │ │ │ │ ├── ochuman_dataset.py │ │ │ │ │ ├── posetrack18_dataset.py │ │ │ │ │ └── posetrack18_video_dataset.py │ │ │ │ ├── body3d │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── aic2goliath3d_dataset.py │ │ │ │ │ ├── coco_wholebody2goliath3d_dataset.py │ │ │ │ │ ├── crowdpose2goliath3d_dataset.py │ │ │ │ │ ├── goliath3d_dataset.py │ │ │ │ │ ├── h36m_dataset.py │ │ │ │ │ └── mpii2goliath3d_dataset.py │ │ │ │ ├── face │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── aflw_dataset.py │ │ │ │ │ ├── coco_wholebody_face_dataset.py │ │ │ │ │ ├── cofw_dataset.py │ │ │ │ │ ├── face_300w_dataset.py │ │ │ │ │ ├── lapa_dataset.py │ │ │ │ │ └── wflw_dataset.py │ │ │ │ ├── fashion │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── deepfashion2_dataset.py │ │ │ │ │ └── deepfashion_dataset.py │ │ │ │ ├── hand │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── coco_wholebody_hand_dataset.py │ │ │ │ │ ├── freihand_dataset.py │ │ │ │ │ ├── onehand10k_dataset.py │ │ │ │ │ ├── panoptic_hand2d_dataset.py │ │ │ │ │ └── rhd2d_dataset.py │ │ │ │ ├── utils.py │ │ │ │ └── wholebody │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── coco_wholebody2goliath_dataset.py │ │ │ │ │ ├── coco_wholebody_dataset.py │ │ │ │ │ └── halpe_dataset.py │ │ │ ├── samplers.py │ │ │ └── transforms │ │ │ │ ├── __init__.py │ │ │ │ ├── bottomup_transforms.py │ │ │ │ ├── common_transforms.py │ │ │ │ ├── converting.py │ │ │ │ ├── formatting.py │ │ │ │ ├── loading.py │ │ │ │ ├── pose3d_transforms.py │ │ │ │ └── topdown_transforms.py │ │ ├── engine │ │ │ ├── __init__.py │ │ │ ├── hooks │ │ │ │ ├── __init__.py │ │ │ │ ├── custom_visualization_hook.py │ │ │ │ ├── ema_hook.py │ │ │ │ ├── general_visualization_hook.py │ │ │ │ ├── pose3d_visualization_hook.py │ │ │ │ └── visualization_hook.py │ │ │ └── optim_wrappers │ │ │ │ ├── __init__.py │ │ │ │ └── layer_decay_optim_wrapper.py │ │ ├── evaluation │ │ │ ├── __init__.py │ │ │ ├── functional │ │ │ │ ├── __init__.py │ │ │ │ ├── keypoint_eval.py │ │ │ │ ├── mesh_eval.py │ │ │ │ └── nms.py │ │ │ └── metrics │ │ │ │ ├── __init__.py │ │ │ │ ├── coco_metric.py │ │ │ │ ├── coco_wholebody_metric.py │ │ │ │ ├── goliath3d_coco_wholebody_metric.py │ │ │ │ ├── goliath_coco_wholebody_metric.py │ │ │ │ ├── goliath_metric.py │ │ │ │ ├── keypoint_2d_metrics.py │ │ │ │ ├── keypoint_3d_metrics.py │ │ │ │ ├── keypoint_partition_metric.py │ │ │ │ └── posetrack18_metric.py │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── backbones │ │ │ │ ├── __init__.py │ │ │ │ ├── alexnet.py │ │ │ │ ├── base_backbone.py │ │ │ │ ├── cpm.py │ │ │ │ ├── hourglass.py │ │ │ │ ├── hourglass_ae.py │ │ │ │ ├── hrformer.py │ │ │ │ ├── hrnet.py │ │ │ │ ├── litehrnet.py │ │ │ │ ├── mobilenet_v2.py │ │ │ │ ├── mobilenet_v3.py │ │ │ │ ├── mspn.py │ │ │ │ ├── pvt.py │ │ │ │ ├── regnet.py │ │ │ │ ├── resnest.py │ │ │ │ ├── resnet.py │ │ │ │ ├── resnext.py │ │ │ │ ├── rsn.py │ │ │ │ ├── scnet.py │ │ │ │ ├── seresnet.py │ │ │ │ ├── seresnext.py │ │ │ │ ├── shufflenet_v1.py │ │ │ │ ├── shufflenet_v2.py │ │ │ │ ├── swin.py │ │ │ │ ├── tcn.py │ │ │ │ ├── utils │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── channel_shuffle.py │ │ │ │ │ ├── ckpt_convert.py │ │ │ │ │ ├── inverted_residual.py │ │ │ │ │ ├── make_divisible.py │ │ │ │ │ ├── se_layer.py │ │ │ │ │ └── utils.py │ │ │ │ ├── v2v_net.py │ │ │ │ ├── vgg.py │ │ │ │ ├── vipnas_mbv3.py │ │ │ │ └── vipnas_resnet.py │ │ │ ├── builder.py │ │ │ ├── data_preprocessors │ │ │ │ ├── __init__.py │ │ │ │ └── data_preprocessor.py │ │ │ ├── heads │ │ │ │ ├── __init__.py │ │ │ │ ├── base_head.py │ │ │ │ ├── coord_cls_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── rtmcc_head.py │ │ │ │ │ └── simcc_head.py │ │ │ │ ├── heatmap_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── ae_head.py │ │ │ │ │ ├── cid_head.py │ │ │ │ │ ├── cpm_head.py │ │ │ │ │ ├── heatmap_head.py │ │ │ │ │ ├── mspn_head.py │ │ │ │ │ ├── pose3d_heatmap_head.py │ │ │ │ │ └── vipnas_head.py │ │ │ │ ├── hybrid_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── dekr_head.py │ │ │ │ │ └── vis_head.py │ │ │ │ └── regression_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── dsnt_head.py │ │ │ │ │ ├── integral_regression_head.py │ │ │ │ │ ├── regression_head.py │ │ │ │ │ ├── rle_head.py │ │ │ │ │ ├── temporal_regression_head.py │ │ │ │ │ └── trajectory_regression_head.py │ │ │ ├── losses │ │ │ │ ├── __init__.py │ │ │ │ ├── ae_loss.py │ │ │ │ ├── classification_loss.py │ │ │ │ ├── heatmap_loss.py │ │ │ │ ├── loss_wrappers.py │ │ │ │ ├── pose3d_loss.py │ │ │ │ └── regression_loss.py │ │ │ ├── necks │ │ │ │ ├── __init__.py │ │ │ │ ├── fmap_proc_neck.py │ │ │ │ ├── fpn.py │ │ │ │ ├── gap_neck.py │ │ │ │ └── posewarper_neck.py │ │ │ ├── pose_estimators │ │ │ │ ├── __init__.py │ │ │ │ ├── base.py │ │ │ │ ├── bottomup.py │ │ │ │ ├── pose_lifter.py │ │ │ │ ├── topdown.py │ │ │ │ └── topdown3d.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── check_and_update_config.py │ │ │ │ ├── ckpt_convert.py │ │ │ │ ├── geometry.py │ │ │ │ ├── ops.py │ │ │ │ ├── realnvp.py │ │ │ │ ├── regularizations.py │ │ │ │ ├── rtmcc_block.py │ │ │ │ ├── transformer.py │ │ │ │ └── tta.py │ │ ├── registry.py │ │ ├── structures │ │ │ ├── __init__.py │ │ │ ├── bbox │ │ │ │ ├── __init__.py │ │ │ │ └── transforms.py │ │ │ ├── keypoint │ │ │ │ ├── __init__.py │ │ │ │ └── transforms.py │ │ │ ├── multilevel_pixel_data.py │ │ │ ├── pose_data_sample.py │ │ │ └── utils.py │ │ ├── testing │ │ │ ├── __init__.py │ │ │ └── _utils.py │ │ ├── utils │ │ │ ├── __init__.py │ │ │ ├── camera.py │ │ │ ├── collect_env.py │ │ │ ├── config_utils.py │ │ │ ├── hooks.py │ │ │ ├── logger.py │ │ │ ├── setup_env.py │ │ │ ├── tensor_utils.py │ │ │ ├── timer.py │ │ │ └── typing.py │ │ ├── version.py │ │ └── visualization │ │ │ ├── __init__.py │ │ │ ├── fast_visualizer.py │ │ │ ├── local_visualizer.py │ │ │ ├── local_visualizer_3d.py │ │ │ ├── opencv_backend_visualizer.py │ │ │ └── simcc_vis.py │ ├── pytest.ini │ ├── requirements.txt │ ├── requirements │ │ ├── albu.txt │ │ ├── build.txt │ │ ├── docs.txt │ │ ├── mminstall.txt │ │ ├── optional.txt │ │ ├── poseval.txt │ │ ├── readthedocs.txt │ │ ├── runtime.txt │ │ └── tests.txt │ ├── scripts │ │ ├── demo │ │ │ └── local │ │ │ │ ├── keypoints133.sh │ │ │ │ ├── keypoints133_openpose.sh │ │ │ │ ├── keypoints17.sh │ │ │ │ ├── keypoints17_openpose.sh │ │ │ │ └── keypoints308.sh │ │ ├── optimize │ │ │ ├── bfloat16 │ │ │ │ ├── pose133_optimizer.sh │ │ │ │ ├── pose17_optimizer.sh │ │ │ │ └── pose308_optimizer.sh │ │ │ └── torchscript │ │ │ │ ├── pose133_optimizer.sh │ │ │ │ ├── pose17_optimizer.sh │ │ │ │ └── pose308_optimizer.sh │ │ └── train │ │ │ ├── coco_mpii_crowdpose_aic │ │ │ ├── sapiens_0.3b │ │ │ │ ├── node.sh │ │ │ │ └── slurm.sh │ │ │ ├── sapiens_0.6b │ │ │ │ ├── node.sh │ │ │ │ └── slurm.sh │ │ │ ├── sapiens_1b │ │ │ │ ├── node.sh │ │ │ │ └── slurm.sh │ │ │ └── sapiens_2b │ │ │ │ ├── node.sh │ │ │ │ └── slurm.sh │ │ │ ├── coco_wholebody │ │ │ ├── sapiens_0.3b │ │ │ │ ├── node.sh │ │ │ │ └── slurm.sh │ │ │ ├── sapiens_0.6b │ │ │ │ ├── node.sh │ │ │ │ └── slurm.sh │ │ │ ├── sapiens_1b │ │ │ │ ├── node.sh │ │ │ │ └── slurm.sh │ │ │ └── sapiens_2b │ │ │ │ ├── node.sh │ │ │ │ └── slurm.sh │ │ │ └── goliath │ │ │ ├── sapiens_0.3b │ │ │ ├── node.sh │ │ │ └── slurm.sh │ │ │ ├── sapiens_0.6b │ │ │ ├── node.sh │ │ │ └── slurm.sh │ │ │ ├── sapiens_1b │ │ │ ├── node.sh │ │ │ └── slurm.sh │ │ │ └── sapiens_2b │ │ │ ├── node.sh │ │ │ └── slurm.sh │ ├── setup.cfg │ ├── setup.py │ └── tools │ │ ├── analysis_tools │ │ ├── analyze_logs.py │ │ ├── get_flops.py │ │ ├── memory_error_utils.py │ │ ├── optimal_batch_size.py │ │ └── print_config.py │ │ ├── deployment │ │ ├── pytorch2torchscript.py │ │ └── torch_optimization.py │ │ ├── dist_test.sh │ │ ├── dist_train.sh │ │ ├── misc │ │ ├── browse_dataset.py │ │ ├── create_no_backbone_prefix_checkpoint.py │ │ ├── image_to_video.py │ │ ├── keypoints2coco_without_mmdet.py │ │ └── publish_model.py │ │ ├── process │ │ ├── check_valid_videos.py │ │ ├── convert_valid_videos.py │ │ ├── fix_corrupted_video.py │ │ ├── pose_process.py │ │ ├── pose_video_process.py │ │ └── vis_video_pose.py │ │ ├── slurm_test.sh │ │ ├── slurm_train.sh │ │ ├── test.py │ │ ├── torchserve │ │ ├── mmpose2torchserve.py │ │ ├── mmpose_handler.py │ │ └── test_torchserver.py │ │ ├── train.py │ │ └── vis_train.py ├── pretrain │ ├── .gitignore │ ├── MANIFEST.in │ ├── configs │ │ ├── _base_ │ │ │ ├── datasets │ │ │ │ ├── cifar100_bs16.py │ │ │ │ ├── cifar10_bs16.py │ │ │ │ ├── coco_caption.py │ │ │ │ ├── coco_okvqa.py │ │ │ │ ├── coco_retrieval.py │ │ │ │ ├── coco_vg_vqa.py │ │ │ │ ├── coco_vqa.py │ │ │ │ ├── cub_bs8_384.py │ │ │ │ ├── cub_bs8_448.py │ │ │ │ ├── flickr30k_caption.py │ │ │ │ ├── flickr30k_retrieval.py │ │ │ │ ├── gqa.py │ │ │ │ ├── imagenet21k_bs128.py │ │ │ │ ├── imagenet_bs128_mbv3.py │ │ │ │ ├── imagenet_bs128_poolformer_medium_224.py │ │ │ │ ├── imagenet_bs128_poolformer_small_224.py │ │ │ │ ├── imagenet_bs128_revvit_224.py │ │ │ │ ├── imagenet_bs128_riformer_medium_384.py │ │ │ │ ├── imagenet_bs128_riformer_small_384.py │ │ │ │ ├── imagenet_bs128_vig_224.py │ │ │ │ ├── imagenet_bs16_eva_196.py │ │ │ │ ├── imagenet_bs16_eva_336.py │ │ │ │ ├── imagenet_bs16_eva_448.py │ │ │ │ ├── imagenet_bs16_eva_560.py │ │ │ │ ├── imagenet_bs16_pil_bicubic_384.py │ │ │ │ ├── imagenet_bs256_beitv2.py │ │ │ │ ├── imagenet_bs256_davit_224.py │ │ │ │ ├── imagenet_bs256_itpn.py │ │ │ │ ├── imagenet_bs256_levit_224.py │ │ │ │ ├── imagenet_bs256_rsb_a12.py │ │ │ │ ├── imagenet_bs256_rsb_a3.py │ │ │ │ ├── imagenet_bs256_simmim_192.py │ │ │ │ ├── imagenet_bs256_swin_192.py │ │ │ │ ├── imagenet_bs32.py │ │ │ │ ├── imagenet_bs32_byol.py │ │ │ │ ├── imagenet_bs32_mocov2.py │ │ │ │ ├── imagenet_bs32_pil_bicubic.py │ │ │ │ ├── imagenet_bs32_pil_resize.py │ │ │ │ ├── imagenet_bs32_simclr.py │ │ │ │ ├── imagenet_bs512_mae.py │ │ │ │ ├── imagenet_bs512_mae_mammoth.py │ │ │ │ ├── imagenet_bs512_mocov3.py │ │ │ │ ├── imagenet_bs64.py │ │ │ │ ├── imagenet_bs64_autoaug.py │ │ │ │ ├── imagenet_bs64_clip_224.py │ │ │ │ ├── imagenet_bs64_clip_384.py │ │ │ │ ├── imagenet_bs64_clip_448.py │ │ │ │ ├── imagenet_bs64_convmixer_224.py │ │ │ │ ├── imagenet_bs64_deit3_224.py │ │ │ │ ├── imagenet_bs64_deit3_384.py │ │ │ │ ├── imagenet_bs64_edgenext_256.py │ │ │ │ ├── imagenet_bs64_hivit_224.py │ │ │ │ ├── imagenet_bs64_mixer_224.py │ │ │ │ ├── imagenet_bs64_pil_resize.py │ │ │ │ ├── imagenet_bs64_pil_resize_autoaug.py │ │ │ │ ├── imagenet_bs64_swin_224.py │ │ │ │ ├── imagenet_bs64_swin_256.py │ │ │ │ ├── imagenet_bs64_swin_384.py │ │ │ │ ├── imagenet_bs64_t2t_224.py │ │ │ │ ├── imagenet_bs8_pil_bicubic_320.py │ │ │ │ ├── inshop_bs32_448.py │ │ │ │ ├── nlvr2.py │ │ │ │ ├── nocaps.py │ │ │ │ ├── ocrvqa.py │ │ │ │ ├── pipelines │ │ │ │ │ ├── auto_aug.py │ │ │ │ │ └── rand_aug.py │ │ │ │ ├── refcoco.py │ │ │ │ ├── vizwiz.py │ │ │ │ ├── voc_bs16.py │ │ │ │ └── vsr.py │ │ │ ├── default_runtime.py │ │ │ ├── models │ │ │ │ ├── conformer │ │ │ │ │ ├── base-p16.py │ │ │ │ │ ├── small-p16.py │ │ │ │ │ ├── small-p32.py │ │ │ │ │ └── tiny-p16.py │ │ │ │ ├── convmixer │ │ │ │ │ ├── convmixer-1024-20.py │ │ │ │ │ ├── convmixer-1536-20.py │ │ │ │ │ └── convmixer-768-32.py │ │ │ │ ├── convnext │ │ │ │ │ ├── convnext-base.py │ │ │ │ │ ├── convnext-large.py │ │ │ │ │ ├── convnext-small.py │ │ │ │ │ ├── convnext-tiny.py │ │ │ │ │ └── convnext-xlarge.py │ │ │ │ ├── convnext_v2 │ │ │ │ │ ├── atto.py │ │ │ │ │ ├── base.py │ │ │ │ │ ├── femto.py │ │ │ │ │ ├── huge.py │ │ │ │ │ ├── large.py │ │ │ │ │ ├── nano.py │ │ │ │ │ ├── pico.py │ │ │ │ │ └── tiny.py │ │ │ │ ├── davit │ │ │ │ │ ├── davit-base.py │ │ │ │ │ ├── davit-small.py │ │ │ │ │ └── davit-tiny.py │ │ │ │ ├── deit3 │ │ │ │ │ ├── deit3-base-p16-224.py │ │ │ │ │ ├── deit3-base-p16-384.py │ │ │ │ │ ├── deit3-huge-p14-224.py │ │ │ │ │ ├── deit3-large-p16-224.py │ │ │ │ │ ├── deit3-large-p16-384.py │ │ │ │ │ ├── deit3-medium-p16-224.py │ │ │ │ │ ├── deit3-small-p16-224.py │ │ │ │ │ └── deit3-small-p16-384.py │ │ │ │ ├── densenet │ │ │ │ │ ├── densenet121.py │ │ │ │ │ ├── densenet161.py │ │ │ │ │ ├── densenet169.py │ │ │ │ │ └── densenet201.py │ │ │ │ ├── edgenext │ │ │ │ │ ├── edgenext-base.py │ │ │ │ │ ├── edgenext-small.py │ │ │ │ │ ├── edgenext-xsmall.py │ │ │ │ │ └── edgenext-xxsmall.py │ │ │ │ ├── efficientformer-l1.py │ │ │ │ ├── efficientnet_b0.py │ │ │ │ ├── efficientnet_b1.py │ │ │ │ ├── efficientnet_b2.py │ │ │ │ ├── efficientnet_b3.py │ │ │ │ ├── efficientnet_b4.py │ │ │ │ ├── efficientnet_b5.py │ │ │ │ ├── efficientnet_b6.py │ │ │ │ ├── efficientnet_b7.py │ │ │ │ ├── efficientnet_b8.py │ │ │ │ ├── efficientnet_em.py │ │ │ │ ├── efficientnet_es.py │ │ │ │ ├── efficientnet_l2.py │ │ │ │ ├── efficientnet_v2 │ │ │ │ │ ├── efficientnetv2_b0.py │ │ │ │ │ ├── efficientnetv2_b1.py │ │ │ │ │ ├── efficientnetv2_b2.py │ │ │ │ │ ├── efficientnetv2_b3.py │ │ │ │ │ ├── efficientnetv2_l.py │ │ │ │ │ ├── efficientnetv2_m.py │ │ │ │ │ ├── efficientnetv2_s.py │ │ │ │ │ └── efficientnetv2_xl.py │ │ │ │ ├── eva │ │ │ │ │ ├── eva-g.py │ │ │ │ │ └── eva-l.py │ │ │ │ ├── hivit │ │ │ │ │ ├── base_224.py │ │ │ │ │ ├── small_224.py │ │ │ │ │ └── tiny_224.py │ │ │ │ ├── hornet │ │ │ │ │ ├── hornet-base-gf.py │ │ │ │ │ ├── hornet-base.py │ │ │ │ │ ├── hornet-large-gf.py │ │ │ │ │ ├── hornet-large-gf384.py │ │ │ │ │ ├── hornet-large.py │ │ │ │ │ ├── hornet-small-gf.py │ │ │ │ │ ├── hornet-small.py │ │ │ │ │ ├── hornet-tiny-gf.py │ │ │ │ │ └── hornet-tiny.py │ │ │ │ ├── hrnet │ │ │ │ │ ├── hrnet-w18.py │ │ │ │ │ ├── hrnet-w30.py │ │ │ │ │ ├── hrnet-w32.py │ │ │ │ │ ├── hrnet-w40.py │ │ │ │ │ ├── hrnet-w44.py │ │ │ │ │ ├── hrnet-w48.py │ │ │ │ │ └── hrnet-w64.py │ │ │ │ ├── inception_v3.py │ │ │ │ ├── itpn_hivit-base-p16.py │ │ │ │ ├── levit-256-p16.py │ │ │ │ ├── mae_hivit-base-p16.py │ │ │ │ ├── mae_vit-base-p16.py │ │ │ │ ├── mae_vit-eva02-base-p16.py │ │ │ │ ├── mixmim │ │ │ │ │ └── mixmim_base.py │ │ │ │ ├── mlp_mixer_base_patch16.py │ │ │ │ ├── mlp_mixer_large_patch16.py │ │ │ │ ├── mobilenet_v2_1x.py │ │ │ │ ├── mobilenet_v3 │ │ │ │ │ ├── mobilenet_v3_large_imagenet.py │ │ │ │ │ ├── mobilenet_v3_small_050_imagenet.py │ │ │ │ │ ├── mobilenet_v3_small_075_imagenet.py │ │ │ │ │ ├── mobilenet_v3_small_cifar.py │ │ │ │ │ └── mobilenet_v3_small_imagenet.py │ │ │ │ ├── mobileone │ │ │ │ │ ├── mobileone_s0.py │ │ │ │ │ ├── mobileone_s1.py │ │ │ │ │ ├── mobileone_s2.py │ │ │ │ │ ├── mobileone_s3.py │ │ │ │ │ └── mobileone_s4.py │ │ │ │ ├── mobilevit │ │ │ │ │ ├── mobilevit_s.py │ │ │ │ │ ├── mobilevit_xs.py │ │ │ │ │ └── mobilevit_xxs.py │ │ │ │ ├── mvit │ │ │ │ │ ├── mvitv2-base.py │ │ │ │ │ ├── mvitv2-large.py │ │ │ │ │ ├── mvitv2-small.py │ │ │ │ │ └── mvitv2-tiny.py │ │ │ │ ├── poolformer │ │ │ │ │ ├── poolformer_m36.py │ │ │ │ │ ├── poolformer_m48.py │ │ │ │ │ ├── poolformer_s12.py │ │ │ │ │ ├── poolformer_s24.py │ │ │ │ │ └── poolformer_s36.py │ │ │ │ ├── regnet │ │ │ │ │ ├── regnetx_1.6gf.py │ │ │ │ │ ├── regnetx_12gf.py │ │ │ │ │ ├── regnetx_3.2gf.py │ │ │ │ │ ├── regnetx_4.0gf.py │ │ │ │ │ ├── regnetx_400mf.py │ │ │ │ │ ├── regnetx_6.4gf.py │ │ │ │ │ ├── regnetx_8.0gf.py │ │ │ │ │ └── regnetx_800mf.py │ │ │ │ ├── replknet-31B_in1k.py │ │ │ │ ├── replknet-31L_in1k.py │ │ │ │ ├── replknet-XL_in1k.py │ │ │ │ ├── repmlp-base_224.py │ │ │ │ ├── repvgg-A0_in1k.py │ │ │ │ ├── repvgg-B3_lbs-mixup_in1k.py │ │ │ │ ├── res2net101-w26-s4.py │ │ │ │ ├── res2net50-w14-s8.py │ │ │ │ ├── res2net50-w26-s4.py │ │ │ │ ├── res2net50-w26-s6.py │ │ │ │ ├── res2net50-w26-s8.py │ │ │ │ ├── res2net50-w48-s2.py │ │ │ │ ├── resnest101.py │ │ │ │ ├── resnest200.py │ │ │ │ ├── resnest269.py │ │ │ │ ├── resnest50.py │ │ │ │ ├── resnet101.py │ │ │ │ ├── resnet101_cifar.py │ │ │ │ ├── resnet152.py │ │ │ │ ├── resnet152_cifar.py │ │ │ │ ├── resnet18.py │ │ │ │ ├── resnet18_cifar.py │ │ │ │ ├── resnet34.py │ │ │ │ ├── resnet34_cifar.py │ │ │ │ ├── resnet34_gem.py │ │ │ │ ├── resnet50.py │ │ │ │ ├── resnet50_cifar.py │ │ │ │ ├── resnet50_cifar_cutmix.py │ │ │ │ ├── resnet50_cifar_mixup.py │ │ │ │ ├── resnet50_cutmix.py │ │ │ │ ├── resnet50_label_smooth.py │ │ │ │ ├── resnet50_mixup.py │ │ │ │ ├── resnetv1c50.py │ │ │ │ ├── resnetv1d101.py │ │ │ │ ├── resnetv1d152.py │ │ │ │ ├── resnetv1d50.py │ │ │ │ ├── resnext101_32x4d.py │ │ │ │ ├── resnext101_32x8d.py │ │ │ │ ├── resnext152_32x4d.py │ │ │ │ ├── resnext50_32x4d.py │ │ │ │ ├── revvit │ │ │ │ │ ├── revvit-base.py │ │ │ │ │ └── revvit-small.py │ │ │ │ ├── seresnet101.py │ │ │ │ ├── seresnet50.py │ │ │ │ ├── seresnext101_32x4d.py │ │ │ │ ├── seresnext50_32x4d.py │ │ │ │ ├── shufflenet_v1_1x.py │ │ │ │ ├── shufflenet_v2_1x.py │ │ │ │ ├── swin_transformer │ │ │ │ │ ├── base_224.py │ │ │ │ │ ├── base_384.py │ │ │ │ │ ├── large_224.py │ │ │ │ │ ├── large_384.py │ │ │ │ │ ├── small_224.py │ │ │ │ │ └── tiny_224.py │ │ │ │ ├── swin_transformer_v2 │ │ │ │ │ ├── base_256.py │ │ │ │ │ ├── base_384.py │ │ │ │ │ ├── large_256.py │ │ │ │ │ ├── large_384.py │ │ │ │ │ ├── small_256.py │ │ │ │ │ └── tiny_256.py │ │ │ │ ├── t2t-vit-t-14.py │ │ │ │ ├── t2t-vit-t-19.py │ │ │ │ ├── t2t-vit-t-24.py │ │ │ │ ├── tinyvit │ │ │ │ │ ├── tinyvit-11m.py │ │ │ │ │ ├── tinyvit-21m.py │ │ │ │ │ └── tinyvit-5m.py │ │ │ │ ├── tnt_s_patch16_224.py │ │ │ │ ├── twins_pcpvt_base.py │ │ │ │ ├── twins_svt_base.py │ │ │ │ ├── van │ │ │ │ │ ├── van_base.py │ │ │ │ │ ├── van_large.py │ │ │ │ │ ├── van_small.py │ │ │ │ │ └── van_tiny.py │ │ │ │ ├── vgg11.py │ │ │ │ ├── vgg11bn.py │ │ │ │ ├── vgg13.py │ │ │ │ ├── vgg13bn.py │ │ │ │ ├── vgg16.py │ │ │ │ ├── vgg16bn.py │ │ │ │ ├── vgg19.py │ │ │ │ ├── vgg19bn.py │ │ │ │ ├── vig │ │ │ │ │ ├── pyramid_vig_base.py │ │ │ │ │ ├── pyramid_vig_medium.py │ │ │ │ │ ├── pyramid_vig_small.py │ │ │ │ │ ├── pyramid_vig_tiny.py │ │ │ │ │ ├── vig_base.py │ │ │ │ │ ├── vig_small.py │ │ │ │ │ └── vig_tiny.py │ │ │ │ ├── vit-base-p16.py │ │ │ │ ├── vit-base-p32.py │ │ │ │ ├── vit-large-p16.py │ │ │ │ ├── vit-large-p32.py │ │ │ │ └── wide-resnet50.py │ │ │ └── schedules │ │ │ │ ├── cifar10_bs128.py │ │ │ │ ├── cub_bs64.py │ │ │ │ ├── imagenet_bs1024_adamw_conformer.py │ │ │ │ ├── imagenet_bs1024_adamw_hivit.py │ │ │ │ ├── imagenet_bs1024_adamw_revvit.py │ │ │ │ ├── imagenet_bs1024_adamw_swin.py │ │ │ │ ├── imagenet_bs1024_coslr.py │ │ │ │ ├── imagenet_bs1024_linearlr_bn_nowd.py │ │ │ │ ├── imagenet_bs2048.py │ │ │ │ ├── imagenet_bs2048_AdamW.py │ │ │ │ ├── imagenet_bs2048_adamw_levit.py │ │ │ │ ├── imagenet_bs2048_coslr.py │ │ │ │ ├── imagenet_bs2048_rsb.py │ │ │ │ ├── imagenet_bs256.py │ │ │ │ ├── imagenet_bs256_140e.py │ │ │ │ ├── imagenet_bs256_200e_coslr_warmup.py │ │ │ │ ├── imagenet_bs256_coslr.py │ │ │ │ ├── imagenet_bs256_coslr_coswd_300e.py │ │ │ │ ├── imagenet_bs256_epochstep.py │ │ │ │ ├── imagenet_bs4096_AdamW.py │ │ │ │ ├── imagenet_lars_coslr_200e.py │ │ │ │ ├── imagenet_lars_coslr_90e.py │ │ │ │ ├── imagenet_sgd_coslr_100e.py │ │ │ │ ├── imagenet_sgd_coslr_200e.py │ │ │ │ └── imagenet_sgd_steplr_100e.py │ │ └── sapiens_mae │ │ │ ├── humans_300m_test │ │ │ ├── mae_sapiens_0.3b-p16_8xb512-coslr-1600e_humans_300m_test.py │ │ │ ├── mae_sapiens_0.6b-p16_8xb512-coslr-1600e_humans_300m_test.py │ │ │ ├── mae_sapiens_1b-p16_8xb512-coslr-1600e_humans_300m_test.py │ │ │ ├── mae_sapiens_2b-p16_8xb512-coslr-1600e_humans_300m_test.py │ │ │ └── vis_mae_sapiens_1b-p16_8xb512-coslr-1600e_humans_300m_test.py │ │ │ └── imagenet │ │ │ └── mae_sapiens_0.3b-p16_8xb512-coslr-1600e_imagenet.py │ ├── demo │ │ ├── extract_attention_map.py │ │ ├── extract_feature.py │ │ ├── image_demo.py │ │ └── mae_demo.py │ ├── mmpretrain │ │ ├── __init__.py │ │ ├── apis │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── feature_extractor.py │ │ │ ├── image_caption.py │ │ │ ├── image_classification.py │ │ │ ├── image_retrieval.py │ │ │ ├── mae_inferencer.py │ │ │ ├── model.py │ │ │ ├── multimodal_retrieval.py │ │ │ ├── nlvr.py │ │ │ ├── utils.py │ │ │ ├── visual_grounding.py │ │ │ └── visual_question_answering.py │ │ ├── configs │ │ │ ├── _base_ │ │ │ │ ├── datasets │ │ │ │ │ ├── cifar10_bs16.py │ │ │ │ │ ├── imagenet21k_bs128.py │ │ │ │ │ ├── imagenet_bs128_mbv3.py │ │ │ │ │ ├── imagenet_bs256_beitv2.py │ │ │ │ │ ├── imagenet_bs32.py │ │ │ │ │ ├── imagenet_bs32_pil_resize.py │ │ │ │ │ ├── imagenet_bs32_simclr.py │ │ │ │ │ ├── imagenet_bs512_mae.py │ │ │ │ │ ├── imagenet_bs64_pil_resize.py │ │ │ │ │ ├── imagenet_bs64_pil_resize_autoaug.py │ │ │ │ │ ├── imagenet_bs64_swin_224.py │ │ │ │ │ └── imagenet_bs64_swin_384.py │ │ │ │ ├── default_runtime.py │ │ │ │ ├── models │ │ │ │ │ ├── convnext_base.py │ │ │ │ │ ├── mae_hivit_base_p16.py │ │ │ │ │ ├── mae_vit_base_p16.py │ │ │ │ │ ├── mobilenet_v2_1x.py │ │ │ │ │ ├── mobilenet_v3_small.py │ │ │ │ │ ├── resnet18.py │ │ │ │ │ └── vit_base_p16.py │ │ │ │ └── schedules │ │ │ │ │ ├── cifar10_bs128.py │ │ │ │ │ ├── imagenet_bs1024_adamw_swin.py │ │ │ │ │ ├── imagenet_bs256.py │ │ │ │ │ ├── imagenet_bs256_epochstep.py │ │ │ │ │ ├── imagenet_bs4096_adamw.py │ │ │ │ │ └── imagenet_lars_coslr_200e.py │ │ │ ├── beit │ │ │ │ ├── beit_beit_base_p16_8xb256_amp_coslr_300e_in1k.py │ │ │ │ └── benchmarks │ │ │ │ │ ├── beit-base-p16_8xb128-coslr-100e_in1k.py │ │ │ │ │ └── beit-base-p16_8xb64_in1k.py │ │ │ ├── beitv2 │ │ │ │ ├── beitv2_beit-base-p16_8xb256-amp-coslr-1600e_in1k.py │ │ │ │ ├── beitv2_beit-base-p16_8xb256-amp-coslr-300e_in1k.py │ │ │ │ └── benchmarks │ │ │ │ │ ├── beit-base-p16_8xb128-coslr-100e_in1k.py │ │ │ │ │ └── beit-base-p16_8xb64_in1k.py │ │ │ ├── convnext │ │ │ │ ├── convnext-base_32xb128_in1k.py │ │ │ │ ├── convnext-base_32xb128_in21k.py │ │ │ │ ├── convnext-large_64xb64_in1k-384px.py │ │ │ │ ├── convnext-large_64xb64_in1k.py │ │ │ │ ├── convnext-large_64xb64_in21k.py │ │ │ │ ├── convnext-small_32xb128_in1k-384px.py │ │ │ │ ├── convnext-small_32xb128_in1k.py │ │ │ │ ├── convnext-tiny_32xb128_in1k-384px.py │ │ │ │ ├── convnext-tiny_32xb128_in1k.py │ │ │ │ ├── convnext-xlarge_64xb64_in1k-384px.py │ │ │ │ ├── convnext-xlarge_64xb64_in1k.py │ │ │ │ ├── convnext-xlarge_64xb64_in21k.py │ │ │ │ └── convnext_base_32xb128_in1k_384px.py │ │ │ ├── eva │ │ │ │ └── eva_mae_style_vit_base_p16_16xb256_coslr_400e_in1k.py │ │ │ ├── mae │ │ │ │ ├── mae_hivit_base_p16_8xb512_amp_coslr_1600e_in1k.py │ │ │ │ ├── mae_hivit_base_p16_8xb512_amp_coslr_400e_in1k.py │ │ │ │ ├── mae_hivit_base_p16_8xb512_amp_coslr_800e_in1k.py │ │ │ │ ├── mae_hivit_large_p16_8xb512_amp_coslr_1600e_in1k.py │ │ │ │ ├── mae_hivit_large_p16_8xb512_amp_coslr_400e_in1k.py │ │ │ │ ├── mae_hivit_large_p16_8xb512_amp_coslr_800e_in1k.py │ │ │ │ ├── mae_vit_base_p16_8xb512_amp_coslr_1600e_in1k.py │ │ │ │ ├── mae_vit_base_p16_8xb512_amp_coslr_300e_in1k.py │ │ │ │ ├── mae_vit_base_p16_8xb512_amp_coslr_400e_in1k.py │ │ │ │ ├── mae_vit_base_p16_8xb512_amp_coslr_800e_in1k.py │ │ │ │ ├── mae_vit_huge_p14_8xb512_amp_coslr_1600e_in1k.py │ │ │ │ ├── mae_vit_large_p16_8xb512_amp_coslr_1600e_in1k.py │ │ │ │ ├── mae_vit_large_p16_8xb512_amp_coslr_300e_in1k.py │ │ │ │ ├── mae_vit_large_p16_8xb512_amp_coslr_400e_in1k.py │ │ │ │ └── mae_vit_large_p16_8xb512_amp_coslr_800e_in1k.py │ │ │ ├── mobilenet_v2 │ │ │ │ └── mobilenet_v2_8xb32_in1k.py │ │ │ ├── mobilenet_v3 │ │ │ │ ├── mobilenet_v3_large_8xb128_in1k.py │ │ │ │ ├── mobilenet_v3_small_050_8xb128_in1k.py │ │ │ │ ├── mobilenet_v3_small_075_8xb128_in1k.py │ │ │ │ ├── mobilenet_v3_small_8xb128_in1k.py │ │ │ │ └── mobilenet_v3_small_8xb16_cifar10.py │ │ │ ├── resnet │ │ │ │ └── resnet18_8xb32_in1k.py │ │ │ ├── simclr │ │ │ │ └── simclr_resnet50_16xb256_coslr_200e_in1k.py │ │ │ └── vision_transformer │ │ │ │ ├── vit_base_p16_32xb128_mae_in1k.py │ │ │ │ ├── vit_base_p16_64xb64_in1k.py │ │ │ │ ├── vit_base_p16_64xb64_in1k_384px.py │ │ │ │ ├── vit_base_p32_64xb64_in1k.py │ │ │ │ ├── vit_base_p32_64xb64_in1k_384px.py │ │ │ │ ├── vit_large_p16_64xb64_in1k.py │ │ │ │ ├── vit_large_p16_64xb64_in1k_384px.py │ │ │ │ ├── vit_large_p32_64xb64_in1k.py │ │ │ │ └── vit_large_p32_64xb64_in1k_384px.py │ │ ├── datasets │ │ │ ├── __init__.py │ │ │ ├── _airstore_base_dataset.py │ │ │ ├── airstore_base_dataset.py │ │ │ ├── base_dataset.py │ │ │ ├── builder.py │ │ │ ├── caltech101.py │ │ │ ├── categories.py │ │ │ ├── cifar.py │ │ │ ├── coco_caption.py │ │ │ ├── coco_retrieval.py │ │ │ ├── coco_vqa.py │ │ │ ├── cub.py │ │ │ ├── custom.py │ │ │ ├── dataset_wrappers.py │ │ │ ├── dtd.py │ │ │ ├── fgvcaircraft.py │ │ │ ├── flamingo.py │ │ │ ├── flickr30k_caption.py │ │ │ ├── flickr30k_retrieval.py │ │ │ ├── flowers102.py │ │ │ ├── food101.py │ │ │ ├── gqa_dataset.py │ │ │ ├── humans_300m.py │ │ │ ├── iconqa.py │ │ │ ├── imagenet.py │ │ │ ├── infographic_vqa.py │ │ │ ├── inshop.py │ │ │ ├── mnist.py │ │ │ ├── multi_label.py │ │ │ ├── multi_task.py │ │ │ ├── nlvr2.py │ │ │ ├── nocaps.py │ │ │ ├── ocr_vqa.py │ │ │ ├── oxfordiiitpet.py │ │ │ ├── places205.py │ │ │ ├── refcoco.py │ │ │ ├── samplers │ │ │ │ ├── __init__.py │ │ │ │ ├── repeat_aug.py │ │ │ │ └── sequential.py │ │ │ ├── scienceqa.py │ │ │ ├── stanfordcars.py │ │ │ ├── sun397.py │ │ │ ├── textvqa.py │ │ │ ├── transforms │ │ │ │ ├── __init__.py │ │ │ │ ├── auto_augment.py │ │ │ │ ├── formatting.py │ │ │ │ ├── processing.py │ │ │ │ ├── utils.py │ │ │ │ └── wrappers.py │ │ │ ├── utils.py │ │ │ ├── vg_vqa.py │ │ │ ├── visual_genome.py │ │ │ ├── vizwiz.py │ │ │ ├── voc.py │ │ │ └── vsr.py │ │ ├── engine │ │ │ ├── __init__.py │ │ │ ├── hooks │ │ │ │ ├── __init__.py │ │ │ │ ├── class_num_check_hook.py │ │ │ │ ├── densecl_hook.py │ │ │ │ ├── ema_hook.py │ │ │ │ ├── margin_head_hooks.py │ │ │ │ ├── precise_bn_hook.py │ │ │ │ ├── pretrain2_visualization_hook.py │ │ │ │ ├── pretrain_visualization_hook.py │ │ │ │ ├── retriever_hooks.py │ │ │ │ ├── simsiam_hook.py │ │ │ │ ├── swav_hook.py │ │ │ │ ├── switch_recipe_hook.py │ │ │ │ ├── visualization_hook.py │ │ │ │ └── warmup_param_hook.py │ │ │ ├── optimizers │ │ │ │ ├── __init__.py │ │ │ │ ├── adan_t.py │ │ │ │ ├── lamb.py │ │ │ │ ├── lars.py │ │ │ │ └── layer_decay_optim_wrapper_constructor.py │ │ │ ├── runners │ │ │ │ ├── __init__.py │ │ │ │ └── retrieval_loop.py │ │ │ └── schedulers │ │ │ │ ├── __init__.py │ │ │ │ └── weight_decay_scheduler.py │ │ ├── evaluation │ │ │ ├── __init__.py │ │ │ ├── functional │ │ │ │ └── __init__.py │ │ │ └── metrics │ │ │ │ ├── ANLS.py │ │ │ │ ├── __init__.py │ │ │ │ ├── caption.py │ │ │ │ ├── gqa.py │ │ │ │ ├── multi_label.py │ │ │ │ ├── multi_task.py │ │ │ │ ├── nocaps.py │ │ │ │ ├── retrieval.py │ │ │ │ ├── scienceqa.py │ │ │ │ ├── shape_bias_label.py │ │ │ │ ├── single_label.py │ │ │ │ ├── visual_grounding_eval.py │ │ │ │ ├── voc_multi_label.py │ │ │ │ └── vqa.py │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── backbones │ │ │ │ ├── __init__.py │ │ │ │ ├── alexnet.py │ │ │ │ ├── base_backbone.py │ │ │ │ ├── beit.py │ │ │ │ ├── conformer.py │ │ │ │ ├── convmixer.py │ │ │ │ ├── convnext.py │ │ │ │ ├── cspnet.py │ │ │ │ ├── davit.py │ │ │ │ ├── deit.py │ │ │ │ ├── deit3.py │ │ │ │ ├── densenet.py │ │ │ │ ├── edgenext.py │ │ │ │ ├── efficientformer.py │ │ │ │ ├── efficientnet.py │ │ │ │ ├── efficientnet_v2.py │ │ │ │ ├── hivit.py │ │ │ │ ├── hornet.py │ │ │ │ ├── hrnet.py │ │ │ │ ├── inception_v3.py │ │ │ │ ├── lenet.py │ │ │ │ ├── levit.py │ │ │ │ ├── mixmim.py │ │ │ │ ├── mlp_mixer.py │ │ │ │ ├── mobilenet_v2.py │ │ │ │ ├── mobilenet_v3.py │ │ │ │ ├── mobileone.py │ │ │ │ ├── mobilevit.py │ │ │ │ ├── mvit.py │ │ │ │ ├── poolformer.py │ │ │ │ ├── regnet.py │ │ │ │ ├── replknet.py │ │ │ │ ├── repmlp.py │ │ │ │ ├── repvgg.py │ │ │ │ ├── res2net.py │ │ │ │ ├── resnest.py │ │ │ │ ├── resnet.py │ │ │ │ ├── resnet_cifar.py │ │ │ │ ├── resnext.py │ │ │ │ ├── revvit.py │ │ │ │ ├── riformer.py │ │ │ │ ├── seresnet.py │ │ │ │ ├── seresnext.py │ │ │ │ ├── shufflenet_v1.py │ │ │ │ ├── shufflenet_v2.py │ │ │ │ ├── sparse_convnext.py │ │ │ │ ├── sparse_resnet.py │ │ │ │ ├── swin_transformer.py │ │ │ │ ├── swin_transformer_v2.py │ │ │ │ ├── t2t_vit.py │ │ │ │ ├── timm_backbone.py │ │ │ │ ├── tinyvit.py │ │ │ │ ├── tnt.py │ │ │ │ ├── twins.py │ │ │ │ ├── van.py │ │ │ │ ├── vgg.py │ │ │ │ ├── vig.py │ │ │ │ ├── vision_transformer.py │ │ │ │ ├── vision_transformer2.py │ │ │ │ ├── vit_eva02.py │ │ │ │ ├── vit_sam.py │ │ │ │ └── xcit.py │ │ │ ├── builder.py │ │ │ ├── classifiers │ │ │ │ ├── __init__.py │ │ │ │ ├── base.py │ │ │ │ ├── hugging_face.py │ │ │ │ ├── image.py │ │ │ │ └── timm.py │ │ │ ├── heads │ │ │ │ ├── __init__.py │ │ │ │ ├── beitv1_head.py │ │ │ │ ├── beitv2_head.py │ │ │ │ ├── cae_head.py │ │ │ │ ├── cls_head.py │ │ │ │ ├── conformer_head.py │ │ │ │ ├── contrastive_head.py │ │ │ │ ├── deit_head.py │ │ │ │ ├── efficientformer_head.py │ │ │ │ ├── grounding_head.py │ │ │ │ ├── itc_head.py │ │ │ │ ├── itm_head.py │ │ │ │ ├── itpn_clip_head.py │ │ │ │ ├── latent_heads.py │ │ │ │ ├── levit_head.py │ │ │ │ ├── linear_head.py │ │ │ │ ├── mae_head.py │ │ │ │ ├── mae_head2.py │ │ │ │ ├── margin_head.py │ │ │ │ ├── mim_head.py │ │ │ │ ├── mixmim_head.py │ │ │ │ ├── mocov3_head.py │ │ │ │ ├── multi_label_cls_head.py │ │ │ │ ├── multi_label_csra_head.py │ │ │ │ ├── multi_label_linear_head.py │ │ │ │ ├── multi_task_head.py │ │ │ │ ├── seq_gen_head.py │ │ │ │ ├── simmim_head.py │ │ │ │ ├── spark_head.py │ │ │ │ ├── stacked_head.py │ │ │ │ ├── swav_head.py │ │ │ │ ├── vig_head.py │ │ │ │ ├── vision_transformer_head.py │ │ │ │ └── vqa_head.py │ │ │ ├── losses │ │ │ │ ├── __init__.py │ │ │ │ ├── asymmetric_loss.py │ │ │ │ ├── cae_loss.py │ │ │ │ ├── cosine_similarity_loss.py │ │ │ │ ├── cross_correlation_loss.py │ │ │ │ ├── cross_entropy_loss.py │ │ │ │ ├── focal_loss.py │ │ │ │ ├── label_smooth_loss.py │ │ │ │ ├── reconstruction_loss.py │ │ │ │ ├── seesaw_loss.py │ │ │ │ ├── swav_loss.py │ │ │ │ └── utils.py │ │ │ ├── multimodal │ │ │ │ ├── __init__.py │ │ │ │ ├── blip │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── blip_caption.py │ │ │ │ │ ├── blip_grounding.py │ │ │ │ │ ├── blip_nlvr.py │ │ │ │ │ ├── blip_retrieval.py │ │ │ │ │ ├── blip_vqa.py │ │ │ │ │ └── language_model.py │ │ │ │ ├── blip2 │ │ │ │ │ ├── Qformer.py │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── blip2_caption.py │ │ │ │ │ ├── blip2_opt_vqa.py │ │ │ │ │ ├── blip2_retriever.py │ │ │ │ │ └── modeling_opt.py │ │ │ │ ├── chinese_clip │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── bert.py │ │ │ │ │ ├── chinese_clip.py │ │ │ │ │ └── utils.py │ │ │ │ ├── clip │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── clip.py │ │ │ │ │ ├── clip_transformer.py │ │ │ │ │ └── utils.py │ │ │ │ ├── flamingo │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── adapter.py │ │ │ │ │ ├── flamingo.py │ │ │ │ │ ├── modules.py │ │ │ │ │ └── utils.py │ │ │ │ ├── llava │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── llava.py │ │ │ │ │ └── modules.py │ │ │ │ ├── minigpt4 │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── minigpt4.py │ │ │ │ ├── ofa │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── ofa.py │ │ │ │ │ └── ofa_modules.py │ │ │ │ └── otter │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── otter.py │ │ │ ├── necks │ │ │ │ ├── __init__.py │ │ │ │ ├── beitv2_neck.py │ │ │ │ ├── cae_neck.py │ │ │ │ ├── densecl_neck.py │ │ │ │ ├── gap.py │ │ │ │ ├── gem.py │ │ │ │ ├── hr_fuse.py │ │ │ │ ├── itpn_neck.py │ │ │ │ ├── linear_neck.py │ │ │ │ ├── mae_neck.py │ │ │ │ ├── mae_neck2.py │ │ │ │ ├── milan_neck.py │ │ │ │ ├── mixmim_neck.py │ │ │ │ ├── mocov2_neck.py │ │ │ │ ├── nonlinear_neck.py │ │ │ │ ├── simmim_neck.py │ │ │ │ ├── spark_neck.py │ │ │ │ └── swav_neck.py │ │ │ ├── peft │ │ │ │ ├── __init__.py │ │ │ │ └── lora.py │ │ │ ├── retrievers │ │ │ │ ├── __init__.py │ │ │ │ ├── base.py │ │ │ │ └── image2image.py │ │ │ ├── selfsup │ │ │ │ ├── __init__.py │ │ │ │ ├── barlowtwins.py │ │ │ │ ├── base.py │ │ │ │ ├── beit.py │ │ │ │ ├── byol.py │ │ │ │ ├── cae.py │ │ │ │ ├── densecl.py │ │ │ │ ├── eva.py │ │ │ │ ├── itpn.py │ │ │ │ ├── mae.py │ │ │ │ ├── mae_eva02.py │ │ │ │ ├── mae_sapiens2.py │ │ │ │ ├── maskfeat.py │ │ │ │ ├── mff.py │ │ │ │ ├── milan.py │ │ │ │ ├── mixmim.py │ │ │ │ ├── moco.py │ │ │ │ ├── mocov3.py │ │ │ │ ├── simclr.py │ │ │ │ ├── simmim.py │ │ │ │ ├── simsiam.py │ │ │ │ ├── spark.py │ │ │ │ └── swav.py │ │ │ ├── tta │ │ │ │ ├── __init__.py │ │ │ │ └── score_tta.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── attention.py │ │ │ │ ├── batch_augments │ │ │ │ ├── __init__.py │ │ │ │ ├── cutmix.py │ │ │ │ ├── mixup.py │ │ │ │ ├── resizemix.py │ │ │ │ └── wrapper.py │ │ │ │ ├── batch_shuffle.py │ │ │ │ ├── box_utils.py │ │ │ │ ├── channel_shuffle.py │ │ │ │ ├── clip_generator_helper.py │ │ │ │ ├── data_preprocessor.py │ │ │ │ ├── ema.py │ │ │ │ ├── embed.py │ │ │ │ ├── helpers.py │ │ │ │ ├── huggingface.py │ │ │ │ ├── inverted_residual.py │ │ │ │ ├── layer_scale.py │ │ │ │ ├── make_divisible.py │ │ │ │ ├── norm.py │ │ │ │ ├── position_encoding.py │ │ │ │ ├── res_layer_extra_norm.py │ │ │ │ ├── se_layer.py │ │ │ │ ├── sparse_modules.py │ │ │ │ ├── swiglu_ffn.py │ │ │ │ ├── tokenizer.py │ │ │ │ └── vector_quantizer.py │ │ ├── registry.py │ │ ├── structures │ │ │ ├── __init__.py │ │ │ ├── data_sample.py │ │ │ ├── multi_task_data_sample.py │ │ │ └── utils.py │ │ ├── utils │ │ │ ├── __init__.py │ │ │ ├── analyze.py │ │ │ ├── collect_env.py │ │ │ ├── dependency.py │ │ │ ├── misc.py │ │ │ ├── progress.py │ │ │ └── setup_env.py │ │ ├── version.py │ │ └── visualization │ │ │ ├── __init__.py │ │ │ ├── utils.py │ │ │ └── visualizer.py │ ├── requirements.txt │ ├── requirements │ │ ├── docs.txt │ │ ├── mminstall.txt │ │ ├── multimodal.txt │ │ ├── optional.txt │ │ ├── readthedocs.txt │ │ ├── runtime.txt │ │ └── tests.txt │ ├── scripts │ │ ├── _install │ │ │ ├── make_links_dgx.sh │ │ │ └── make_links_rsc.sh │ │ ├── demo │ │ │ └── local │ │ │ │ └── extract_feature.sh │ │ ├── misc │ │ │ ├── clean_checkpoints.sh │ │ │ ├── finetune_clean_checkpoints.sh │ │ │ └── pretrain_clean_checkpoints.sh │ │ └── optimize │ │ │ ├── bfloat16 │ │ │ └── feature_extracter_optimizer.sh │ │ │ └── torchscript │ │ │ └── feature_extracter_optimizer.sh │ ├── setup.cfg │ ├── setup.py │ └── tools │ │ ├── analysis_tools │ │ ├── analyze_logs.py │ │ ├── analyze_results.py │ │ ├── confusion_matrix.py │ │ ├── eval_metric.py │ │ ├── get_flops.py │ │ ├── shape_bias.py │ │ └── utils.py │ │ ├── deployment │ │ ├── pytorch2torchscript.py │ │ └── torch_optimization.py │ │ ├── dist_test.sh │ │ ├── dist_train.sh │ │ ├── misc │ │ ├── clean_checkpoints.py │ │ ├── finetune_clean_checkpoints.py │ │ ├── pca_feature.py │ │ ├── pretrain_clean_checkpoints.py │ │ ├── print_config.py │ │ └── verify_dataset.py │ │ ├── slurm_test.sh │ │ ├── slurm_train.sh │ │ ├── test.py │ │ ├── train.py │ │ └── visualization │ │ ├── browse_dataset.py │ │ ├── vis_cam.py │ │ ├── vis_scheduler.py │ │ └── vis_tsne.py └── seg │ ├── .gitignore │ ├── MANIFEST.in │ ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ ├── ade20k.py │ │ │ ├── ade20k_640x640.py │ │ │ ├── bdd100k.py │ │ │ ├── chase_db1.py │ │ │ ├── cityscapes.py │ │ │ ├── cityscapes_1024x1024.py │ │ │ ├── cityscapes_768x768.py │ │ │ ├── cityscapes_769x769.py │ │ │ ├── cityscapes_832x832.py │ │ │ ├── coco-stuff10k.py │ │ │ ├── coco-stuff164k.py │ │ │ ├── drive.py │ │ │ ├── goliath.py │ │ │ ├── goliath_1024x1024.py │ │ │ ├── hrf.py │ │ │ ├── isaid.py │ │ │ ├── levir_256x256.py │ │ │ ├── loveda.py │ │ │ ├── mapillary_v1.py │ │ │ ├── mapillary_v1_65.py │ │ │ ├── mapillary_v2.py │ │ │ ├── nyu.py │ │ │ ├── nyu_512x512.py │ │ │ ├── pascal_context.py │ │ │ ├── pascal_context_59.py │ │ │ ├── pascal_voc12.py │ │ │ ├── pascal_voc12_aug.py │ │ │ ├── potsdam.py │ │ │ ├── refuge.py │ │ │ ├── stare.py │ │ │ ├── synapse.py │ │ │ └── vaihingen.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ ├── ann_r50-d8.py │ │ │ ├── apcnet_r50-d8.py │ │ │ ├── bisenetv1_r18-d32.py │ │ │ ├── bisenetv2.py │ │ │ ├── ccnet_r50-d8.py │ │ │ ├── cgnet.py │ │ │ ├── danet_r50-d8.py │ │ │ ├── deeplabv3_r50-d8.py │ │ │ ├── deeplabv3_unet_s5-d16.py │ │ │ ├── deeplabv3plus_r50-d8.py │ │ │ ├── dmnet_r50-d8.py │ │ │ ├── dnl_r50-d8.py │ │ │ ├── dpt_vit-b16.py │ │ │ ├── emanet_r50-d8.py │ │ │ ├── encnet_r50-d8.py │ │ │ ├── erfnet_fcn.py │ │ │ ├── fast_scnn.py │ │ │ ├── fastfcn_r50-d32_jpu_psp.py │ │ │ ├── fcn_hr18.py │ │ │ ├── fcn_r50-d8.py │ │ │ ├── fcn_unet_s5-d16.py │ │ │ ├── fpn_poolformer_s12.py │ │ │ ├── fpn_r50.py │ │ │ ├── gcnet_r50-d8.py │ │ │ ├── icnet_r50-d8.py │ │ │ ├── isanet_r50-d8.py │ │ │ ├── lraspp_m-v3-d8.py │ │ │ ├── nonlocal_r50-d8.py │ │ │ ├── ocrnet_hr18.py │ │ │ ├── ocrnet_r50-d8.py │ │ │ ├── pointrend_r50.py │ │ │ ├── psanet_r50-d8.py │ │ │ ├── pspnet_r50-d8.py │ │ │ ├── pspnet_unet_s5-d16.py │ │ │ ├── san_vit-b16.py │ │ │ ├── segformer_mit-b0.py │ │ │ ├── segmenter_vit-b16_mask.py │ │ │ ├── setr_mla.py │ │ │ ├── setr_naive.py │ │ │ ├── setr_pup.py │ │ │ ├── stdc.py │ │ │ ├── twins_pcpvt-s_fpn.py │ │ │ ├── twins_pcpvt-s_upernet.py │ │ │ ├── upernet_beit.py │ │ │ ├── upernet_convnext.py │ │ │ ├── upernet_mae.py │ │ │ ├── upernet_r50.py │ │ │ ├── upernet_swin.py │ │ │ ├── upernet_vit-b16_ln_mln.py │ │ │ └── vpd_sd.py │ │ └── schedules │ │ │ ├── schedule_160k.py │ │ │ ├── schedule_160k_vit.py │ │ │ ├── schedule_20k.py │ │ │ ├── schedule_240k.py │ │ │ ├── schedule_25k.py │ │ │ ├── schedule_320k.py │ │ │ ├── schedule_40k.py │ │ │ └── schedule_80k.py │ ├── sapiens_depth │ │ └── render_people │ │ │ ├── sapiens_0.3b_render_people-1024x768.py │ │ │ ├── sapiens_0.6b_render_people-1024x768.py │ │ │ ├── sapiens_1b_render_people-1024x768.py │ │ │ └── sapiens_2b_render_people-1024x768.py │ ├── sapiens_normal │ │ ├── normal_general │ │ │ └── sapiens_1b_normal_general-1024x768.py │ │ └── normal_render_people │ │ │ ├── sapiens_0.3b_normal_render_people-1024x768.py │ │ │ ├── sapiens_0.6b_normal_render_people-1024x768.py │ │ │ ├── sapiens_1b_normal_render_people-1024x768.py │ │ │ └── sapiens_2b_normal_render_people-1024x768.py │ └── sapiens_seg │ │ ├── cihp │ │ ├── sapiens_0.3b_cihp-1024x768.py │ │ ├── sapiens_0.6b_cihp-1024x768.py │ │ ├── sapiens_1b_cihp-1024x768.py │ │ └── sapiens_2b_cihp-1024x768.py │ │ ├── goliath │ │ ├── sapiens_0.3b_goliath-1024x768.py │ │ ├── sapiens_0.6b_goliath-1024x768.py │ │ ├── sapiens_1b_goliath-1024x768.py │ │ └── sapiens_2b_goliath-1024x768.py │ │ ├── lip │ │ ├── sapiens_0.3b_lip-1024x768.py │ │ ├── sapiens_0.6b_lip-1024x768.py │ │ ├── sapiens_1b_lip-1024x768.py │ │ └── sapiens_2b_lip-1024x768.py │ │ └── seg_face │ │ └── sapiens_1b_seg_face-1024x768.py │ ├── demo │ ├── demo_depth_vis.py │ ├── demo_normal_vis.py │ ├── demo_seg_vis.py │ ├── image_demo.py │ └── image_demo_with_inferencer.py │ ├── mmseg │ ├── __init__.py │ ├── apis │ │ ├── __init__.py │ │ ├── inference.py │ │ ├── mmseg_inferencer.py │ │ ├── remote_sense_inferencer.py │ │ └── utils.py │ ├── datasets │ │ ├── __init__.py │ │ ├── ade.py │ │ ├── albedo.py │ │ ├── albedo_render_people.py │ │ ├── atr2goliath.py │ │ ├── basesegdataset.py │ │ ├── bdd100k.py │ │ ├── cihp.py │ │ ├── cihp2goliath.py │ │ ├── cityscapes.py │ │ ├── coco_stuff.py │ │ ├── dataset_wrappers.py │ │ ├── depth_dataset_wrappers.py │ │ ├── face.py │ │ ├── general_dataset_wrappers.py │ │ ├── goliath.py │ │ ├── hdri.py │ │ ├── lip.py │ │ ├── lip2goliath.py │ │ ├── metric_render_people.py │ │ ├── normal_dataset_wrappers.py │ │ ├── normal_general.py │ │ ├── normal_render_people.py │ │ ├── pascal2goliath.py │ │ ├── pascal_context.py │ │ ├── pointmap_dataset_wrappers.py │ │ ├── pointmap_render_people.py │ │ ├── render_people.py │ │ ├── stereo_correspondences_dataset_wrappers.py │ │ ├── stereo_correspondences_render_people.py │ │ ├── stereo_pointmap_dataset_wrappers.py │ │ ├── stereo_pointmap_render_people.py │ │ ├── transforms │ │ │ ├── __init__.py │ │ │ ├── albedo_transforms.py │ │ │ ├── depth_transforms.py │ │ │ ├── formatting.py │ │ │ ├── general_transforms.py │ │ │ ├── hdri_transforms.py │ │ │ ├── loading.py │ │ │ ├── normal_transforms.py │ │ │ ├── pointmap_transforms.py │ │ │ ├── stereo_correspondences_transforms.py │ │ │ ├── stereo_pointmap_transforms.py │ │ │ └── transforms.py │ │ └── voc.py │ ├── engine │ │ ├── __init__.py │ │ ├── hooks │ │ │ ├── __init__.py │ │ │ ├── depth_visualization_hook.py │ │ │ ├── general_seg_visualization_hook.py │ │ │ ├── general_visualization_hook.py │ │ │ ├── normal_visualization_hook.py │ │ │ └── visualization_hook.py │ │ ├── optimizers │ │ │ ├── __init__.py │ │ │ ├── force_default_constructor.py │ │ │ ├── layer_decay_optim_wrapper.py │ │ │ ├── layer_decay_optimizer_constructor.py │ │ │ └── stereo_pointmap_layer_decay_optim_wrapper.py │ │ └── schedulers │ │ │ ├── __init__.py │ │ │ └── poly_ratio_scheduler.py │ ├── evaluation │ │ ├── __init__.py │ │ └── metrics │ │ │ ├── __init__.py │ │ │ ├── citys_metric.py │ │ │ ├── depth_metric.py │ │ │ └── iou_metric.py │ ├── models │ │ ├── __init__.py │ │ ├── assigners │ │ │ ├── __init__.py │ │ │ ├── base_assigner.py │ │ │ ├── hungarian_assigner.py │ │ │ └── match_cost.py │ │ ├── backbones │ │ │ ├── __init__.py │ │ │ ├── beit.py │ │ │ ├── bisenetv1.py │ │ │ ├── bisenetv2.py │ │ │ ├── cgnet.py │ │ │ ├── ddrnet.py │ │ │ ├── erfnet.py │ │ │ ├── fast_scnn.py │ │ │ ├── hrnet.py │ │ │ ├── icnet.py │ │ │ ├── mae.py │ │ │ ├── mit.py │ │ │ ├── mobilenet_v2.py │ │ │ ├── mobilenet_v3.py │ │ │ ├── mscan.py │ │ │ ├── pidnet.py │ │ │ ├── resnest.py │ │ │ ├── resnet.py │ │ │ ├── resnext.py │ │ │ ├── stdc.py │ │ │ ├── swin.py │ │ │ ├── timm_backbone.py │ │ │ ├── twins.py │ │ │ ├── unet.py │ │ │ ├── vit.py │ │ │ └── vpd.py │ │ ├── builder.py │ │ ├── data_preprocessor.py │ │ ├── decode_heads │ │ │ ├── __init__.py │ │ │ ├── ann_head.py │ │ │ ├── apc_head.py │ │ │ ├── aspp_head.py │ │ │ ├── cascade_decode_head.py │ │ │ ├── cc_head.py │ │ │ ├── da_head.py │ │ │ ├── ddr_head.py │ │ │ ├── decode_head.py │ │ │ ├── dm_head.py │ │ │ ├── dnl_head.py │ │ │ ├── dpt_head.py │ │ │ ├── ema_head.py │ │ │ ├── enc_head.py │ │ │ ├── fcn_head.py │ │ │ ├── fpn_head.py │ │ │ ├── gc_head.py │ │ │ ├── ham_head.py │ │ │ ├── isa_head.py │ │ │ ├── knet_head.py │ │ │ ├── lraspp_head.py │ │ │ ├── mask2former_head.py │ │ │ ├── maskformer_head.py │ │ │ ├── nl_head.py │ │ │ ├── ocr_head.py │ │ │ ├── pid_head.py │ │ │ ├── point_head.py │ │ │ ├── psa_head.py │ │ │ ├── psp_head.py │ │ │ ├── san_head.py │ │ │ ├── segformer_head.py │ │ │ ├── segmenter_mask_head.py │ │ │ ├── sep_aspp_head.py │ │ │ ├── sep_fcn_head.py │ │ │ ├── setr_mla_head.py │ │ │ ├── setr_up_head.py │ │ │ ├── stdc_head.py │ │ │ ├── uper_head.py │ │ │ ├── vit_depth_head.py │ │ │ ├── vit_hdri_head.py │ │ │ ├── vit_head.py │ │ │ ├── vit_metric_depth_head.py │ │ │ ├── vit_normal_head.py │ │ │ ├── vit_normal_head2.py │ │ │ ├── vit_pointmap_head.py │ │ │ ├── vit_stereo_correspondences_head.py │ │ │ ├── vit_stereo_pointmap_head.py │ │ │ └── vpd_depth_head.py │ │ ├── losses │ │ │ ├── __init__.py │ │ │ ├── accuracy.py │ │ │ ├── boundary_loss.py │ │ │ ├── cosine_similarity_loss.py │ │ │ ├── cross_entropy_loss.py │ │ │ ├── dice_loss.py │ │ │ ├── edge_aware_loss.py │ │ │ ├── focal_loss.py │ │ │ ├── huasdorff_distance_loss.py │ │ │ ├── kldiv_loss.py │ │ │ ├── l1_loss.py │ │ │ ├── lovasz_loss.py │ │ │ ├── metric_silog_loss.py │ │ │ ├── ohem_cross_entropy_loss.py │ │ │ ├── pointmap_consistency_loss.py │ │ │ ├── pointmap_l1_loss.py │ │ │ ├── pointmap_silog_loss.py │ │ │ ├── silog_loss.py │ │ │ ├── stereo_correspondences_loss.py │ │ │ ├── stereo_pointmap_correspondence_loss.py │ │ │ ├── stereo_pointmap_l1_loss.py │ │ │ ├── tversky_loss.py │ │ │ ├── unit_norm_loss.py │ │ │ └── utils.py │ │ ├── necks │ │ │ ├── __init__.py │ │ │ ├── featurepyramid.py │ │ │ ├── fpn.py │ │ │ ├── ic_neck.py │ │ │ ├── jpu.py │ │ │ ├── mla_neck.py │ │ │ └── multilevel_neck.py │ │ ├── segmentors │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── cascade_encoder_decoder.py │ │ │ ├── depth_estimator.py │ │ │ ├── encoder_decoder.py │ │ │ ├── hdri_estimator.py │ │ │ ├── multimodal_encoder_decoder.py │ │ │ ├── pointmap_estimator.py │ │ │ ├── seg_tta.py │ │ │ ├── stereo_correspondences_estimator.py │ │ │ └── stereo_pointmap_estimator.py │ │ ├── text_encoder │ │ │ ├── __init__.py │ │ │ └── clip_text_encoder.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── basic_block.py │ │ │ ├── embed.py │ │ │ ├── encoding.py │ │ │ ├── inverted_residual.py │ │ │ ├── make_divisible.py │ │ │ ├── point_sample.py │ │ │ ├── ppm.py │ │ │ ├── res_layer.py │ │ │ ├── san_layers.py │ │ │ ├── se_layer.py │ │ │ ├── self_attention_block.py │ │ │ ├── shape_convert.py │ │ │ ├── up_conv_block.py │ │ │ └── wrappers.py │ ├── registry │ │ ├── __init__.py │ │ └── registry.py │ ├── structures │ │ ├── __init__.py │ │ ├── sampler │ │ │ ├── __init__.py │ │ │ ├── base_pixel_sampler.py │ │ │ ├── builder.py │ │ │ └── ohem_pixel_sampler.py │ │ └── seg_data_sample.py │ ├── utils │ │ ├── __init__.py │ │ ├── bpe_simple_vocab_16e6.txt.gz │ │ ├── class_names.py │ │ ├── collect_env.py │ │ ├── get_templates.py │ │ ├── io.py │ │ ├── mask_classification.py │ │ ├── misc.py │ │ ├── nearest_neighbours.py │ │ ├── set_env.py │ │ ├── tokenizer.py │ │ └── typing_utils.py │ ├── version.py │ └── visualization │ │ ├── __init__.py │ │ └── local_visualizer.py │ ├── requirements.txt │ ├── requirements │ ├── albu.txt │ ├── docs.txt │ ├── mminstall.txt │ ├── multimodal.txt │ ├── optional.txt │ ├── readthedocs.txt │ ├── runtime.txt │ └── tests.txt │ ├── scripts │ ├── demo │ │ └── local │ │ │ ├── depth.sh │ │ │ ├── normal.sh │ │ │ ├── seg.sh │ │ │ └── seg_face.sh │ ├── finetune │ │ └── normal_general │ │ │ └── sapiens_1b │ │ │ ├── node.sh │ │ │ └── slurm.sh │ └── optimize │ │ ├── bfloat16 │ │ ├── depth_optimizer.sh │ │ ├── normal_optimizer.sh │ │ └── seg_optimizer.sh │ │ └── torchscript │ │ ├── depth_optimizer.sh │ │ ├── normal_optimizer.sh │ │ └── seg_optimizer.sh │ ├── setup.cfg │ ├── setup.py │ └── tools │ ├── analysis_tools │ ├── analyze_logs.py │ ├── benchmark.py │ ├── browse_dataset.py │ ├── confusion_matrix.py │ ├── get_flops.py │ ├── memory_error_utils.py │ ├── optimal_batch_size.py │ ├── profiler.py │ └── visualization_cam.py │ ├── deployment │ ├── pytorch2torchscript.py │ ├── sparsification.py │ └── torch_optimization.py │ ├── dist_test.sh │ ├── dist_train.sh │ ├── slurm_test.sh │ ├── slurm_train.sh │ ├── test.py │ └── train.py ├── scripts ├── configs │ └── ssdlite_mobilenetv2_scratch_600e_coco.py ├── demo.py ├── demo_video.py ├── load_eval_res.py ├── test.py └── train.py ├── setup.py ├── smplx_repo ├── .gitignore ├── LICENSE ├── README.md ├── config_files │ ├── smpl2smplh.yaml │ ├── smpl2smplx.yaml │ ├── smplh2smpl.yaml │ ├── smplh2smplx.yaml │ ├── smplh2smplx_as.yaml │ ├── smplh2smplx_onepose.yaml │ ├── smplx2smpl.yaml │ └── smplx2smplh.yaml ├── examples │ ├── demo.py │ ├── demo_layers.py │ ├── vis_flame_vertices.py │ └── vis_mano_vertices.py ├── optional-requirements.txt ├── requirements.txt ├── setup.py ├── smplx │ ├── __init__.py │ ├── body_models.py │ ├── joint_names.py │ ├── lbs.py │ ├── utils.py │ ├── vertex_ids.py │ └── vertex_joint_selector.py ├── tools │ ├── README.md │ ├── __init__.py │ ├── clean_ch.py │ └── merge_smplh_mano.py └── transfer_model │ ├── README.md │ ├── __init__.py │ ├── __main__.py │ ├── config │ ├── __init__.py │ ├── body_model_defaults.py │ ├── cmd_parser.py │ ├── dataset_defaults.py │ ├── defaults.py │ ├── loss_defaults.py │ ├── optim_defaults.py │ └── utils_cfg.py │ ├── docs │ └── transfer.md │ ├── losses │ ├── __init__.py │ ├── losses.py │ └── utils.py │ ├── merge_output.py │ ├── optimizers │ ├── __init__.py │ ├── minimize.py │ └── optim_factory.py │ ├── requirements.txt │ ├── transfer_model.py │ ├── utils │ ├── __init__.py │ ├── def_transfer.py │ ├── mesh_utils.py │ ├── metrics.py │ ├── np_utils.py │ ├── o3d_utils.py │ ├── pose_utils.py │ ├── timer.py │ ├── torch_utils.py │ └── typing.py │ ├── view_pkl.py │ └── write_obj.py ├── test_bash_slurm.sh ├── torch-trust-ncg ├── .gitignore ├── LICENSE ├── README.md ├── examples │ ├── SoftMaxRegression │ │ └── main.py │ └── ToyExamples │ │ └── main.py ├── requirements.txt ├── setup.py └── torchtrustncg │ ├── __init__.py │ ├── trust_region.py │ └── utils.py └── train_bash_slurm.sh /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/.dockerignore -------------------------------------------------------------------------------- /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/.gitignore -------------------------------------------------------------------------------- /DAv2/DA-2K.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/DA-2K.md -------------------------------------------------------------------------------- /DAv2/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/LICENSE -------------------------------------------------------------------------------- /DAv2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/README.md -------------------------------------------------------------------------------- /DAv2/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/app.py -------------------------------------------------------------------------------- /DAv2/assets/examples/demo01.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo01.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo02.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo02.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo03.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo03.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo04.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo04.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo05.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo05.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo06.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo06.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo07.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo07.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo08.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo08.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo09.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo09.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo10.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo10.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo11.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo11.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo12.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo12.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo13.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo13.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo14.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo14.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo15.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo15.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo16.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo16.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo17.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo17.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo18.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo18.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo19.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo19.jpg -------------------------------------------------------------------------------- /DAv2/assets/examples/demo20.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/assets/examples/demo20.jpg -------------------------------------------------------------------------------- /DAv2/depth_anything_v2/dinov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/depth_anything_v2/dinov2.py -------------------------------------------------------------------------------- /DAv2/depth_anything_v2/dpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/depth_anything_v2/dpt.py -------------------------------------------------------------------------------- /DAv2/depth_anything_v2/util/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/depth_anything_v2/util/blocks.py -------------------------------------------------------------------------------- /DAv2/metric_depth/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/metric_depth/README.md -------------------------------------------------------------------------------- /DAv2/metric_depth/dataset/hypersim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/metric_depth/dataset/hypersim.py -------------------------------------------------------------------------------- /DAv2/metric_depth/dataset/kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/metric_depth/dataset/kitti.py -------------------------------------------------------------------------------- /DAv2/metric_depth/dataset/vkitti2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/metric_depth/dataset/vkitti2.py -------------------------------------------------------------------------------- /DAv2/metric_depth/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/metric_depth/dist_train.sh -------------------------------------------------------------------------------- /DAv2/metric_depth/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/metric_depth/requirements.txt -------------------------------------------------------------------------------- /DAv2/metric_depth/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/metric_depth/run.py -------------------------------------------------------------------------------- /DAv2/metric_depth/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/metric_depth/train.py -------------------------------------------------------------------------------- /DAv2/metric_depth/util/dist_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/metric_depth/util/dist_helper.py -------------------------------------------------------------------------------- /DAv2/metric_depth/util/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/metric_depth/util/loss.py -------------------------------------------------------------------------------- /DAv2/metric_depth/util/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/metric_depth/util/metric.py -------------------------------------------------------------------------------- /DAv2/metric_depth/util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/metric_depth/util/utils.py -------------------------------------------------------------------------------- /DAv2/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/requirements.txt -------------------------------------------------------------------------------- /DAv2/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/run.py -------------------------------------------------------------------------------- /DAv2/run_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/DAv2/run_video.py -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/README.md -------------------------------------------------------------------------------- /aios_repo/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/.gitignore -------------------------------------------------------------------------------- /aios_repo/.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/.pre-commit-config.yaml -------------------------------------------------------------------------------- /aios_repo/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/LICENSE -------------------------------------------------------------------------------- /aios_repo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/README.md -------------------------------------------------------------------------------- /aios_repo/build_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/build_helper.py -------------------------------------------------------------------------------- /aios_repo/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/config/__init__.py -------------------------------------------------------------------------------- /aios_repo/config/aios_smplx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/config/aios_smplx.py -------------------------------------------------------------------------------- /aios_repo/config/aios_smplx_bedlam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/config/aios_smplx_bedlam.py -------------------------------------------------------------------------------- /aios_repo/config/aios_smplx_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/config/aios_smplx_demo.py -------------------------------------------------------------------------------- /aios_repo/config/aios_smplx_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/config/aios_smplx_train.py -------------------------------------------------------------------------------- /aios_repo/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/config/config.py -------------------------------------------------------------------------------- /aios_repo/config/test_agora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/config/test_agora.py -------------------------------------------------------------------------------- /aios_repo/config/test_arctic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/config/test_arctic.py -------------------------------------------------------------------------------- /aios_repo/config/test_egobody.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/config/test_egobody.py -------------------------------------------------------------------------------- /aios_repo/config/test_ehf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/config/test_ehf.py -------------------------------------------------------------------------------- /aios_repo/config/test_ubody.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/config/test_ubody.py -------------------------------------------------------------------------------- /aios_repo/datasets/AGORA_MM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/datasets/AGORA_MM.py -------------------------------------------------------------------------------- /aios_repo/datasets/ARCTIC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/datasets/ARCTIC.py -------------------------------------------------------------------------------- /aios_repo/datasets/BEDLAM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/datasets/BEDLAM.py -------------------------------------------------------------------------------- /aios_repo/datasets/COCO_NA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/datasets/COCO_NA.py -------------------------------------------------------------------------------- /aios_repo/datasets/EHF.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/datasets/EHF.py -------------------------------------------------------------------------------- /aios_repo/datasets/INFERENCE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/datasets/INFERENCE.py -------------------------------------------------------------------------------- /aios_repo/datasets/INFERENCE_AGORA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/datasets/INFERENCE_AGORA.py -------------------------------------------------------------------------------- /aios_repo/datasets/INFERENCE_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/datasets/INFERENCE_demo.py -------------------------------------------------------------------------------- /aios_repo/datasets/UBody_MM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/datasets/UBody_MM.py -------------------------------------------------------------------------------- /aios_repo/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/datasets/__init__.py -------------------------------------------------------------------------------- /aios_repo/datasets/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/datasets/dataset.py -------------------------------------------------------------------------------- /aios_repo/datasets/humandata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/datasets/humandata.py -------------------------------------------------------------------------------- /aios_repo/detrsmpl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/detrsmpl/__init__.py -------------------------------------------------------------------------------- /aios_repo/detrsmpl/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/detrsmpl/apis/__init__.py -------------------------------------------------------------------------------- /aios_repo/detrsmpl/apis/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/detrsmpl/apis/inference.py -------------------------------------------------------------------------------- /aios_repo/detrsmpl/apis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/detrsmpl/apis/test.py -------------------------------------------------------------------------------- /aios_repo/detrsmpl/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/detrsmpl/apis/train.py -------------------------------------------------------------------------------- /aios_repo/detrsmpl/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/detrsmpl/core/__init__.py -------------------------------------------------------------------------------- /aios_repo/detrsmpl/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/detrsmpl/data/__init__.py -------------------------------------------------------------------------------- /aios_repo/detrsmpl/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/detrsmpl/models/__init__.py -------------------------------------------------------------------------------- /aios_repo/detrsmpl/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/detrsmpl/utils/__init__.py -------------------------------------------------------------------------------- /aios_repo/detrsmpl/utils/geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/detrsmpl/utils/geometry.py -------------------------------------------------------------------------------- /aios_repo/detrsmpl/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/detrsmpl/utils/logger.py -------------------------------------------------------------------------------- /aios_repo/detrsmpl/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/detrsmpl/utils/misc.py -------------------------------------------------------------------------------- /aios_repo/detrsmpl/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/detrsmpl/version.py -------------------------------------------------------------------------------- /aios_repo/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/engine.py -------------------------------------------------------------------------------- /aios_repo/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/main.py -------------------------------------------------------------------------------- /aios_repo/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/models/__init__.py -------------------------------------------------------------------------------- /aios_repo/models/aios/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/models/aios/__init__.py -------------------------------------------------------------------------------- /aios_repo/models/aios/aios_smplx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/models/aios/aios_smplx.py -------------------------------------------------------------------------------- /aios_repo/models/aios/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/models/aios/matcher.py -------------------------------------------------------------------------------- /aios_repo/models/aios/ops/make.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/models/aios/ops/make.sh -------------------------------------------------------------------------------- /aios_repo/models/aios/ops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/models/aios/ops/setup.py -------------------------------------------------------------------------------- /aios_repo/models/aios/ops/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/models/aios/ops/test.py -------------------------------------------------------------------------------- /aios_repo/models/aios/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/models/aios/transformer.py -------------------------------------------------------------------------------- /aios_repo/models/aios/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/models/aios/utils.py -------------------------------------------------------------------------------- /aios_repo/models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/models/registry.py -------------------------------------------------------------------------------- /aios_repo/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/models/utils.py -------------------------------------------------------------------------------- /aios_repo/pyrightconfig.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/pyrightconfig.json -------------------------------------------------------------------------------- /aios_repo/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/requirements.txt -------------------------------------------------------------------------------- /aios_repo/scripts/inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/scripts/inference.sh -------------------------------------------------------------------------------- /aios_repo/scripts/test_agora.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/scripts/test_agora.sh -------------------------------------------------------------------------------- /aios_repo/scripts/test_agora_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/scripts/test_agora_val.sh -------------------------------------------------------------------------------- /aios_repo/scripts/test_bedlam.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/scripts/test_bedlam.sh -------------------------------------------------------------------------------- /aios_repo/scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/scripts/train.sh -------------------------------------------------------------------------------- /aios_repo/util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/__init__.py -------------------------------------------------------------------------------- /aios_repo/util/box_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/box_loss.py -------------------------------------------------------------------------------- /aios_repo/util/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/box_ops.py -------------------------------------------------------------------------------- /aios_repo/util/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/config.py -------------------------------------------------------------------------------- /aios_repo/util/dir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/dir.py -------------------------------------------------------------------------------- /aios_repo/util/distribute_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/distribute_utils.py -------------------------------------------------------------------------------- /aios_repo/util/formatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/formatting.py -------------------------------------------------------------------------------- /aios_repo/util/get_param_dicts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/get_param_dicts.py -------------------------------------------------------------------------------- /aios_repo/util/human_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/human_models.py -------------------------------------------------------------------------------- /aios_repo/util/keypoint_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/keypoint_ops.py -------------------------------------------------------------------------------- /aios_repo/util/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/logger.py -------------------------------------------------------------------------------- /aios_repo/util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/misc.py -------------------------------------------------------------------------------- /aios_repo/util/plot_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/plot_utils.py -------------------------------------------------------------------------------- /aios_repo/util/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/preprocessing.py -------------------------------------------------------------------------------- /aios_repo/util/smplx/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/smplx/README.md -------------------------------------------------------------------------------- /aios_repo/util/smplx/examples/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/smplx/examples/demo.py -------------------------------------------------------------------------------- /aios_repo/util/smplx/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/smplx/setup.py -------------------------------------------------------------------------------- /aios_repo/util/smplx/smplx/lbs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/smplx/smplx/lbs.py -------------------------------------------------------------------------------- /aios_repo/util/smplx/smplx/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/smplx/smplx/utils.py -------------------------------------------------------------------------------- /aios_repo/util/smplx/tools/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/smplx/tools/README.md -------------------------------------------------------------------------------- /aios_repo/util/time_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/time_counter.py -------------------------------------------------------------------------------- /aios_repo/util/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/transforms.py -------------------------------------------------------------------------------- /aios_repo/util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/utils.py -------------------------------------------------------------------------------- /aios_repo/util/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/aios_repo/util/vis.py -------------------------------------------------------------------------------- /api/BLADE_API.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/api/BLADE_API.py -------------------------------------------------------------------------------- /api/test_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/api/test_api.py -------------------------------------------------------------------------------- /assets/demo_results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/assets/demo_results.png -------------------------------------------------------------------------------- /assets/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/assets/pipeline.png -------------------------------------------------------------------------------- /assets/teaser_blade.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/assets/teaser_blade.png -------------------------------------------------------------------------------- /blade/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/__init__.py -------------------------------------------------------------------------------- /blade/cameras/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/cameras/__init__.py -------------------------------------------------------------------------------- /blade/cameras/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/cameras/builder.py -------------------------------------------------------------------------------- /blade/cameras/convert_convention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/cameras/convert_convention.py -------------------------------------------------------------------------------- /blade/cameras/convert_projection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/cameras/convert_projection.py -------------------------------------------------------------------------------- /blade/cameras/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/cameras/utils.py -------------------------------------------------------------------------------- /blade/configs/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/configs/base.py -------------------------------------------------------------------------------- /blade/configs/blade_depthnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/configs/blade_depthnet.py -------------------------------------------------------------------------------- /blade/configs/blade_inthewild.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/configs/blade_inthewild.py -------------------------------------------------------------------------------- /blade/configs/blade_posenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/configs/blade_posenet.py -------------------------------------------------------------------------------- /blade/custom_hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/custom_hooks/__init__.py -------------------------------------------------------------------------------- /blade/custom_hooks/wandb_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/custom_hooks/wandb_hook.py -------------------------------------------------------------------------------- /blade/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/datasets/__init__.py -------------------------------------------------------------------------------- /blade/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/datasets/builder.py -------------------------------------------------------------------------------- /blade/datasets/gender_lookup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/datasets/gender_lookup.py -------------------------------------------------------------------------------- /blade/datasets/human_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/datasets/human_image_dataset.py -------------------------------------------------------------------------------- /blade/datasets/image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/datasets/image_dataset.py -------------------------------------------------------------------------------- /blade/datasets/itw_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/datasets/itw_dataset.py -------------------------------------------------------------------------------- /blade/datasets/our_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/datasets/our_dataset.py -------------------------------------------------------------------------------- /blade/datasets/our_dataset_smplx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/datasets/our_dataset_smplx.py -------------------------------------------------------------------------------- /blade/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /blade/datasets/pipelines/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/datasets/pipelines/builder.py -------------------------------------------------------------------------------- /blade/datasets/pipelines/compose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/datasets/pipelines/compose.py -------------------------------------------------------------------------------- /blade/datasets/pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/datasets/pipelines/loading.py -------------------------------------------------------------------------------- /blade/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/datasets/utils.py -------------------------------------------------------------------------------- /blade/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/__init__.py -------------------------------------------------------------------------------- /blade/models/architectures/blade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/architectures/blade.py -------------------------------------------------------------------------------- /blade/models/architectures/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/architectures/builder.py -------------------------------------------------------------------------------- /blade/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/backbones/__init__.py -------------------------------------------------------------------------------- /blade/models/backbones/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/backbones/builder.py -------------------------------------------------------------------------------- /blade/models/backbones/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/backbones/hrnet.py -------------------------------------------------------------------------------- /blade/models/backbones/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/backbones/mlp.py -------------------------------------------------------------------------------- /blade/models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/backbones/resnet.py -------------------------------------------------------------------------------- /blade/models/bert/graphormer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/bert/graphormer.py -------------------------------------------------------------------------------- /blade/models/bert/mlp_mixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/bert/mlp_mixer.py -------------------------------------------------------------------------------- /blade/models/body_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/body_models/__init__.py -------------------------------------------------------------------------------- /blade/models/body_models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/body_models/base.py -------------------------------------------------------------------------------- /blade/models/body_models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/body_models/builder.py -------------------------------------------------------------------------------- /blade/models/body_models/flame.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/body_models/flame.py -------------------------------------------------------------------------------- /blade/models/body_models/mano.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/body_models/mano.py -------------------------------------------------------------------------------- /blade/models/body_models/smpl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/body_models/smpl.py -------------------------------------------------------------------------------- /blade/models/body_models/smpl_d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/body_models/smpl_d.py -------------------------------------------------------------------------------- /blade/models/body_models/smpl_x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/body_models/smpl_x.py -------------------------------------------------------------------------------- /blade/models/body_models/smplh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/body_models/smplh.py -------------------------------------------------------------------------------- /blade/models/body_models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/body_models/utils.py -------------------------------------------------------------------------------- /blade/models/heads/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/heads/builder.py -------------------------------------------------------------------------------- /blade/models/heads/depthonly_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/heads/depthonly_head.py -------------------------------------------------------------------------------- /blade/models/layers/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/layers/conv.py -------------------------------------------------------------------------------- /blade/models/layers/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/layers/fpn.py -------------------------------------------------------------------------------- /blade/models/layers/gcn_lift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/layers/gcn_lift.py -------------------------------------------------------------------------------- /blade/models/layers/graph_cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/layers/graph_cnn.py -------------------------------------------------------------------------------- /blade/models/layers/mlp_lift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/layers/mlp_lift.py -------------------------------------------------------------------------------- /blade/models/layers/realnvp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/layers/realnvp.py -------------------------------------------------------------------------------- /blade/models/layers/stgcn_lift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/layers/stgcn_lift.py -------------------------------------------------------------------------------- /blade/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/losses/__init__.py -------------------------------------------------------------------------------- /blade/models/losses/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/losses/builder.py -------------------------------------------------------------------------------- /blade/models/losses/mesh_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/losses/mesh_loss.py -------------------------------------------------------------------------------- /blade/models/losses/mse_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/losses/mse_loss.py -------------------------------------------------------------------------------- /blade/models/losses/norm_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/losses/norm_loss.py -------------------------------------------------------------------------------- /blade/models/losses/prior_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/losses/prior_loss.py -------------------------------------------------------------------------------- /blade/models/losses/render_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/losses/render_loss.py -------------------------------------------------------------------------------- /blade/models/losses/smooth_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/losses/smooth_loss.py -------------------------------------------------------------------------------- /blade/models/losses/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/losses/utils.py -------------------------------------------------------------------------------- /blade/models/necks/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/necks/builder.py -------------------------------------------------------------------------------- /blade/models/transformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/transformer/__init__.py -------------------------------------------------------------------------------- /blade/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/utils/__init__.py -------------------------------------------------------------------------------- /blade/models/utils/basic_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/utils/basic_block.py -------------------------------------------------------------------------------- /blade/models/utils/embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/utils/embed.py -------------------------------------------------------------------------------- /blade/models/utils/encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/utils/encoding.py -------------------------------------------------------------------------------- /blade/models/utils/make_divisible.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/utils/make_divisible.py -------------------------------------------------------------------------------- /blade/models/utils/point_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/utils/point_sample.py -------------------------------------------------------------------------------- /blade/models/utils/ppm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/utils/ppm.py -------------------------------------------------------------------------------- /blade/models/utils/res_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/utils/res_layer.py -------------------------------------------------------------------------------- /blade/models/utils/san_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/utils/san_layers.py -------------------------------------------------------------------------------- /blade/models/utils/se_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/utils/se_layer.py -------------------------------------------------------------------------------- /blade/models/utils/shape_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/utils/shape_convert.py -------------------------------------------------------------------------------- /blade/models/utils/up_conv_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/utils/up_conv_block.py -------------------------------------------------------------------------------- /blade/models/utils/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/utils/wrappers.py -------------------------------------------------------------------------------- /blade/models/visualizers/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/models/visualizers/builder.py -------------------------------------------------------------------------------- /blade/render/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/render/__init__.py -------------------------------------------------------------------------------- /blade/render/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/render/builder.py -------------------------------------------------------------------------------- /blade/render/explicit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/render/explicit/__init__.py -------------------------------------------------------------------------------- /blade/render/implicit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/render/implicit/__init__.py -------------------------------------------------------------------------------- /blade/structures/meshes/meshes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/structures/meshes/meshes.py -------------------------------------------------------------------------------- /blade/structures/meshes/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/structures/meshes/utils.py -------------------------------------------------------------------------------- /blade/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/transforms/__init__.py -------------------------------------------------------------------------------- /blade/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/utils/__init__.py -------------------------------------------------------------------------------- /blade/utils/bbox_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/utils/bbox_utils.py -------------------------------------------------------------------------------- /blade/utils/cylindar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/utils/cylindar.py -------------------------------------------------------------------------------- /blade/utils/demo_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/utils/demo_utils.py -------------------------------------------------------------------------------- /blade/utils/frame_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/utils/frame_utils.py -------------------------------------------------------------------------------- /blade/utils/ftxty_solve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/utils/ftxty_solve.py -------------------------------------------------------------------------------- /blade/utils/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/utils/helpers.py -------------------------------------------------------------------------------- /blade/utils/keypoint_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/utils/keypoint_utils.py -------------------------------------------------------------------------------- /blade/utils/mesh_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/utils/mesh_utils.py -------------------------------------------------------------------------------- /blade/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/utils/torch_utils.py -------------------------------------------------------------------------------- /blade/utils/visualize_keypoints2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/utils/visualize_keypoints2d.py -------------------------------------------------------------------------------- /blade/utils/visualize_smpl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/utils/visualize_smpl.py -------------------------------------------------------------------------------- /blade/utils/visualize_voxel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/blade/utils/visualize_voxel.py -------------------------------------------------------------------------------- /body_models/smpl/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /body_models/smplx/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cache/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /demo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/demo.sh -------------------------------------------------------------------------------- /demo_images/000012.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/demo_images/000012.jpg -------------------------------------------------------------------------------- /docs/BEDLAMCC_GENERATION.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/docs/BEDLAMCC_GENERATION.md -------------------------------------------------------------------------------- /docs/DATASETS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/docs/DATASETS.md -------------------------------------------------------------------------------- /docs/INSTALL_CUDA13.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/docs/INSTALL_CUDA13.md -------------------------------------------------------------------------------- /docs/MULTI_NODE_TRAINING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/docs/MULTI_NODE_TRAINING.md -------------------------------------------------------------------------------- /docs/bedlamcc_assets.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/docs/bedlamcc_assets.zip -------------------------------------------------------------------------------- /mmcv/.circleci/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/.circleci/config.yml -------------------------------------------------------------------------------- /mmcv/.dev_scripts/visualize_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/.dev_scripts/visualize_lr.py -------------------------------------------------------------------------------- /mmcv/.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/.dockerignore -------------------------------------------------------------------------------- /mmcv/.github/pull_request_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/.github/pull_request_template.md -------------------------------------------------------------------------------- /mmcv/.github/workflows/build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/.github/workflows/build.yml -------------------------------------------------------------------------------- /mmcv/.github/workflows/build_pat.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/.github/workflows/build_pat.yml -------------------------------------------------------------------------------- /mmcv/.github/workflows/lint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/.github/workflows/lint.yml -------------------------------------------------------------------------------- /mmcv/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/.gitignore -------------------------------------------------------------------------------- /mmcv/.owners.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/.owners.yml -------------------------------------------------------------------------------- /mmcv/.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/.pre-commit-config.yaml -------------------------------------------------------------------------------- /mmcv/.readthedocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/.readthedocs.yml -------------------------------------------------------------------------------- /mmcv/CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/CITATION.cff -------------------------------------------------------------------------------- /mmcv/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/CONTRIBUTING.md -------------------------------------------------------------------------------- /mmcv/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/Dockerfile -------------------------------------------------------------------------------- /mmcv/Jenkinsfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/Jenkinsfile -------------------------------------------------------------------------------- /mmcv/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/LICENSE -------------------------------------------------------------------------------- /mmcv/LICENSES.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/LICENSES.md -------------------------------------------------------------------------------- /mmcv/MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/MANIFEST.in -------------------------------------------------------------------------------- /mmcv/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/README.md -------------------------------------------------------------------------------- /mmcv/README_zh-CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/README_zh-CN.md -------------------------------------------------------------------------------- /mmcv/TERMINOLOGY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/TERMINOLOGY.md -------------------------------------------------------------------------------- /mmcv/docs/en/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/en/Makefile -------------------------------------------------------------------------------- /mmcv/docs/en/_static/zhihu_qrcode.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/en/_static/zhihu_qrcode.jpg -------------------------------------------------------------------------------- /mmcv/docs/en/api.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/en/api.rst -------------------------------------------------------------------------------- /mmcv/docs/en/community/pr.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/en/community/pr.md -------------------------------------------------------------------------------- /mmcv/docs/en/compatibility.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/en/compatibility.md -------------------------------------------------------------------------------- /mmcv/docs/en/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/en/conf.py -------------------------------------------------------------------------------- /mmcv/docs/en/deployment/onnx.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/en/deployment/onnx.md -------------------------------------------------------------------------------- /mmcv/docs/en/faq.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/en/faq.md -------------------------------------------------------------------------------- /mmcv/docs/en/get_started/build.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/en/get_started/build.md -------------------------------------------------------------------------------- /mmcv/docs/en/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/en/index.rst -------------------------------------------------------------------------------- /mmcv/docs/en/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/en/make.bat -------------------------------------------------------------------------------- /mmcv/docs/en/understand_mmcv/cnn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/en/understand_mmcv/cnn.md -------------------------------------------------------------------------------- /mmcv/docs/en/understand_mmcv/io.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/en/understand_mmcv/io.md -------------------------------------------------------------------------------- /mmcv/docs/en/understand_mmcv/ops.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/en/understand_mmcv/ops.md -------------------------------------------------------------------------------- /mmcv/docs/en/understand_mmcv/utils.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/en/understand_mmcv/utils.md -------------------------------------------------------------------------------- /mmcv/docs/zh_cn/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/zh_cn/Makefile -------------------------------------------------------------------------------- /mmcv/docs/zh_cn/api.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/zh_cn/api.rst -------------------------------------------------------------------------------- /mmcv/docs/zh_cn/community/pr.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/zh_cn/community/pr.md -------------------------------------------------------------------------------- /mmcv/docs/zh_cn/compatibility.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/zh_cn/compatibility.md -------------------------------------------------------------------------------- /mmcv/docs/zh_cn/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/zh_cn/conf.py -------------------------------------------------------------------------------- /mmcv/docs/zh_cn/deployment/onnx.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/zh_cn/deployment/onnx.md -------------------------------------------------------------------------------- /mmcv/docs/zh_cn/faq.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/zh_cn/faq.md -------------------------------------------------------------------------------- /mmcv/docs/zh_cn/get_started/build.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/zh_cn/get_started/build.md -------------------------------------------------------------------------------- /mmcv/docs/zh_cn/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/zh_cn/index.rst -------------------------------------------------------------------------------- /mmcv/docs/zh_cn/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/zh_cn/make.bat -------------------------------------------------------------------------------- /mmcv/docs/zh_cn/understand_mmcv/io.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/docs/zh_cn/understand_mmcv/io.md -------------------------------------------------------------------------------- /mmcv/examples/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/examples/train.py -------------------------------------------------------------------------------- /mmcv/mmcv/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/arraymisc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/arraymisc/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/arraymisc/quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/arraymisc/quantization.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/alexnet.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/activation.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/context_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/context_block.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/conv.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/conv_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/conv_module.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/conv_ws.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/conv_ws.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/drop.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/hsigmoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/hsigmoid.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/hswish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/hswish.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/non_local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/non_local.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/norm.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/padding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/padding.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/plugin.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/registry.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/scale.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/swish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/swish.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/transformer.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/upsample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/upsample.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/bricks/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/bricks/wrappers.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/builder.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/resnet.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/utils/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/utils/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/utils/flops_counter.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/utils/fuse_conv_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/utils/fuse_conv_bn.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/utils/sync_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/utils/sync_bn.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/utils/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/utils/weight_init.py -------------------------------------------------------------------------------- /mmcv/mmcv/cnn/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/cnn/vgg.py -------------------------------------------------------------------------------- /mmcv/mmcv/device/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/device/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/device/ipu/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/device/ipu/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/device/ipu/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/device/ipu/dataloader.py -------------------------------------------------------------------------------- /mmcv/mmcv/device/ipu/hook_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/device/ipu/hook_wrapper.py -------------------------------------------------------------------------------- /mmcv/mmcv/device/ipu/model_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/device/ipu/model_wrapper.py -------------------------------------------------------------------------------- /mmcv/mmcv/device/ipu/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/device/ipu/runner.py -------------------------------------------------------------------------------- /mmcv/mmcv/device/ipu/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/device/ipu/utils.py -------------------------------------------------------------------------------- /mmcv/mmcv/device/mlu/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/device/mlu/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/device/mlu/_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/device/mlu/_functions.py -------------------------------------------------------------------------------- /mmcv/mmcv/device/mlu/data_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/device/mlu/data_parallel.py -------------------------------------------------------------------------------- /mmcv/mmcv/device/mlu/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/device/mlu/distributed.py -------------------------------------------------------------------------------- /mmcv/mmcv/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/engine/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/engine/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/engine/test.py -------------------------------------------------------------------------------- /mmcv/mmcv/fileio/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/fileio/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/fileio/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/fileio/file_client.py -------------------------------------------------------------------------------- /mmcv/mmcv/fileio/handlers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/fileio/handlers/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/fileio/handlers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/fileio/handlers/base.py -------------------------------------------------------------------------------- /mmcv/mmcv/fileio/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/fileio/io.py -------------------------------------------------------------------------------- /mmcv/mmcv/fileio/parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/fileio/parse.py -------------------------------------------------------------------------------- /mmcv/mmcv/image/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/image/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/image/colorspace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/image/colorspace.py -------------------------------------------------------------------------------- /mmcv/mmcv/image/geometric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/image/geometric.py -------------------------------------------------------------------------------- /mmcv/mmcv/image/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/image/io.py -------------------------------------------------------------------------------- /mmcv/mmcv/image/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/image/misc.py -------------------------------------------------------------------------------- /mmcv/mmcv/image/photometric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/image/photometric.py -------------------------------------------------------------------------------- /mmcv/mmcv/model_zoo/deprecated.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/model_zoo/deprecated.json -------------------------------------------------------------------------------- /mmcv/mmcv/model_zoo/mmcls.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/model_zoo/mmcls.json -------------------------------------------------------------------------------- /mmcv/mmcv/model_zoo/open_mmlab.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/model_zoo/open_mmlab.json -------------------------------------------------------------------------------- /mmcv/mmcv/onnx/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/onnx/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/onnx/info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/onnx/info.py -------------------------------------------------------------------------------- /mmcv/mmcv/onnx/onnx_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/onnx/onnx_utils/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/onnx/symbolic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/onnx/symbolic.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/assign_score_withk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/assign_score_withk.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/ball_query.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/ball_query.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/bbox.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/border_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/border_align.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/box_iou_rotated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/box_iou_rotated.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/carafe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/carafe.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/cc_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/cc_attention.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/chamfer_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/chamfer_distance.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/contour_expand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/contour_expand.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/convex_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/convex_iou.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/corner_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/corner_pool.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/correlation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/correlation.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/csrc/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/csrc/README.md -------------------------------------------------------------------------------- /mmcv/mmcv/ops/csrc/onnxruntime/nms.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/csrc/onnxruntime/nms.h -------------------------------------------------------------------------------- /mmcv/mmcv/ops/csrc/parrots/carafe.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/csrc/parrots/carafe.cpp -------------------------------------------------------------------------------- /mmcv/mmcv/ops/csrc/parrots/info.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/csrc/parrots/info.cpp -------------------------------------------------------------------------------- /mmcv/mmcv/ops/csrc/parrots/iou3d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/csrc/parrots/iou3d.cpp -------------------------------------------------------------------------------- /mmcv/mmcv/ops/csrc/parrots/knn.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/csrc/parrots/knn.cpp -------------------------------------------------------------------------------- /mmcv/mmcv/ops/csrc/parrots/nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/csrc/parrots/nms.cpp -------------------------------------------------------------------------------- /mmcv/mmcv/ops/csrc/pytorch/carafe.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/csrc/pytorch/carafe.cpp -------------------------------------------------------------------------------- /mmcv/mmcv/ops/csrc/pytorch/info.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/csrc/pytorch/info.cpp -------------------------------------------------------------------------------- /mmcv/mmcv/ops/csrc/pytorch/iou3d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/csrc/pytorch/iou3d.cpp -------------------------------------------------------------------------------- /mmcv/mmcv/ops/csrc/pytorch/knn.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/csrc/pytorch/knn.cpp -------------------------------------------------------------------------------- /mmcv/mmcv/ops/csrc/pytorch/nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/csrc/pytorch/nms.cpp -------------------------------------------------------------------------------- /mmcv/mmcv/ops/csrc/pytorch/pybind.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/csrc/pytorch/pybind.cpp -------------------------------------------------------------------------------- /mmcv/mmcv/ops/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/deform_conv.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/deform_roi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/deform_roi_pool.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/deprecated_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/deprecated_wrappers.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/diff_iou_rotated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/diff_iou_rotated.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/focal_loss.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/fused_bias_leakyrelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/fused_bias_leakyrelu.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/gather_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/gather_points.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/group_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/group_points.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/info.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/iou3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/iou3d.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/knn.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/masked_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/masked_conv.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/merge_cells.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/merge_cells.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/min_area_polygons.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/min_area_polygons.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/nms.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/pixel_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/pixel_group.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/point_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/point_sample.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/points_in_boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/points_in_boxes.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/points_in_polygons.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/points_in_polygons.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/points_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/points_sampler.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/psa_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/psa_mask.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/riroi_align_rotated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/riroi_align_rotated.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/roi_align.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/roi_align_rotated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/roi_align_rotated.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/roi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/roi_pool.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/roiaware_pool3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/roiaware_pool3d.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/roipoint_pool3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/roipoint_pool3d.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/saconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/saconv.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/scatter_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/scatter_points.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/sparse_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/sparse_conv.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/sparse_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/sparse_functional.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/sparse_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/sparse_modules.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/sparse_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/sparse_ops.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/sparse_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/sparse_pool.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/sparse_structure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/sparse_structure.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/sync_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/sync_bn.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/three_interpolate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/three_interpolate.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/three_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/three_nn.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/tin_shift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/tin_shift.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/upfirdn2d.py -------------------------------------------------------------------------------- /mmcv/mmcv/ops/voxelize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/ops/voxelize.py -------------------------------------------------------------------------------- /mmcv/mmcv/parallel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/parallel/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/parallel/_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/parallel/_functions.py -------------------------------------------------------------------------------- /mmcv/mmcv/parallel/collate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/parallel/collate.py -------------------------------------------------------------------------------- /mmcv/mmcv/parallel/data_container.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/parallel/data_container.py -------------------------------------------------------------------------------- /mmcv/mmcv/parallel/data_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/parallel/data_parallel.py -------------------------------------------------------------------------------- /mmcv/mmcv/parallel/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/parallel/distributed.py -------------------------------------------------------------------------------- /mmcv/mmcv/parallel/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/parallel/registry.py -------------------------------------------------------------------------------- /mmcv/mmcv/parallel/scatter_gather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/parallel/scatter_gather.py -------------------------------------------------------------------------------- /mmcv/mmcv/parallel/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/parallel/utils.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/base_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/base_module.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/base_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/base_runner.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/builder.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/checkpoint.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/dist_utils.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/fp16_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/fp16_utils.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/hooks/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/hooks/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/hooks/checkpoint.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/hooks/closure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/hooks/closure.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/hooks/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/hooks/ema.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/hooks/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/hooks/evaluation.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/hooks/hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/hooks/hook.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/hooks/iter_timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/hooks/iter_timer.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/hooks/logger/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/hooks/logger/base.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/hooks/logger/pavi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/hooks/logger/pavi.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/hooks/logger/text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/hooks/logger/text.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/hooks/lr_updater.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/hooks/lr_updater.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/hooks/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/hooks/memory.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/hooks/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/hooks/optimizer.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/hooks/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/hooks/profiler.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/hooks/sync_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/hooks/sync_buffer.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/iter_based_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/iter_based_runner.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/log_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/log_buffer.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/optimizer/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/optimizer/builder.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/priority.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/priority.py -------------------------------------------------------------------------------- /mmcv/mmcv/runner/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/runner/utils.py -------------------------------------------------------------------------------- /mmcv/mmcv/tensorrt/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/tensorrt/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/tensorrt/init_plugins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/tensorrt/init_plugins.py -------------------------------------------------------------------------------- /mmcv/mmcv/tensorrt/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/tensorrt/preprocess.py -------------------------------------------------------------------------------- /mmcv/mmcv/tensorrt/tensorrt_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/tensorrt/tensorrt_utils.py -------------------------------------------------------------------------------- /mmcv/mmcv/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/transforms/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/transforms/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/transforms/base.py -------------------------------------------------------------------------------- /mmcv/mmcv/transforms/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/transforms/builder.py -------------------------------------------------------------------------------- /mmcv/mmcv/transforms/formatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/transforms/formatting.py -------------------------------------------------------------------------------- /mmcv/mmcv/transforms/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/transforms/loading.py -------------------------------------------------------------------------------- /mmcv/mmcv/transforms/processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/transforms/processing.py -------------------------------------------------------------------------------- /mmcv/mmcv/transforms/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/transforms/utils.py -------------------------------------------------------------------------------- /mmcv/mmcv/transforms/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/transforms/wrappers.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/config.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/device_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/device_type.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/env.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/ext_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/ext_loader.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/hub.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/logging.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/misc.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/parrots_jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/parrots_jit.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/parrots_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/parrots_wrapper.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/path.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/progressbar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/progressbar.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/registry.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/seed.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/testing.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/timer.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/trace.py -------------------------------------------------------------------------------- /mmcv/mmcv/utils/version_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/utils/version_utils.py -------------------------------------------------------------------------------- /mmcv/mmcv/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/version.py -------------------------------------------------------------------------------- /mmcv/mmcv/video/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/video/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/video/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/video/io.py -------------------------------------------------------------------------------- /mmcv/mmcv/video/optflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/video/optflow.py -------------------------------------------------------------------------------- /mmcv/mmcv/video/processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/video/processing.py -------------------------------------------------------------------------------- /mmcv/mmcv/visualization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/visualization/__init__.py -------------------------------------------------------------------------------- /mmcv/mmcv/visualization/color.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/visualization/color.py -------------------------------------------------------------------------------- /mmcv/mmcv/visualization/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/visualization/image.py -------------------------------------------------------------------------------- /mmcv/mmcv/visualization/optflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/mmcv/visualization/optflow.py -------------------------------------------------------------------------------- /mmcv/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/requirements.txt -------------------------------------------------------------------------------- /mmcv/requirements/build.txt: -------------------------------------------------------------------------------- 1 | pytest-runner 2 | -------------------------------------------------------------------------------- /mmcv/requirements/docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/requirements/docs.txt -------------------------------------------------------------------------------- /mmcv/requirements/optional.txt: -------------------------------------------------------------------------------- 1 | ninja 2 | -------------------------------------------------------------------------------- /mmcv/requirements/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/requirements/test.txt -------------------------------------------------------------------------------- /mmcv/setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/setup.cfg -------------------------------------------------------------------------------- /mmcv/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/setup.py -------------------------------------------------------------------------------- /mmcv/tests/test_arraymisc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_arraymisc.py -------------------------------------------------------------------------------- /mmcv/tests/test_cnn/test_hsigmoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_cnn/test_hsigmoid.py -------------------------------------------------------------------------------- /mmcv/tests/test_cnn/test_hswish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_cnn/test_hswish.py -------------------------------------------------------------------------------- /mmcv/tests/test_cnn/test_non_local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_cnn/test_non_local.py -------------------------------------------------------------------------------- /mmcv/tests/test_cnn/test_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_cnn/test_scale.py -------------------------------------------------------------------------------- /mmcv/tests/test_cnn/test_swish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_cnn/test_swish.py -------------------------------------------------------------------------------- /mmcv/tests/test_cnn/test_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_cnn/test_wrappers.py -------------------------------------------------------------------------------- /mmcv/tests/test_fileclient.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_fileclient.py -------------------------------------------------------------------------------- /mmcv/tests/test_fileio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_fileio.py -------------------------------------------------------------------------------- /mmcv/tests/test_image/test_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_image/test_io.py -------------------------------------------------------------------------------- /mmcv/tests/test_load_model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_load_model_zoo.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_bbox.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_carafe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_carafe.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_info.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_iou3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_iou3d.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_knn.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_nms.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_onnx.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_psa_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_psa_mask.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_roi_align.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_roi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_roi_pool.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_saconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_saconv.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_spconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_spconv.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_syncbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_syncbn.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_tensorrt.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_three_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_three_nn.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_tin_shift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_tin_shift.py -------------------------------------------------------------------------------- /mmcv/tests/test_ops/test_upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_ops/test_upfirdn2d.py -------------------------------------------------------------------------------- /mmcv/tests/test_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_parallel.py -------------------------------------------------------------------------------- /mmcv/tests/test_runner/test_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_runner/test_fp16.py -------------------------------------------------------------------------------- /mmcv/tests/test_runner/test_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_runner/test_hooks.py -------------------------------------------------------------------------------- /mmcv/tests/test_runner/test_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_runner/test_runner.py -------------------------------------------------------------------------------- /mmcv/tests/test_runner/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_runner/test_utils.py -------------------------------------------------------------------------------- /mmcv/tests/test_utils/test_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_utils/test_config.py -------------------------------------------------------------------------------- /mmcv/tests/test_utils/test_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_utils/test_env.py -------------------------------------------------------------------------------- /mmcv/tests/test_utils/test_hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_utils/test_hub.py -------------------------------------------------------------------------------- /mmcv/tests/test_utils/test_logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_utils/test_logging.py -------------------------------------------------------------------------------- /mmcv/tests/test_utils/test_misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_utils/test_misc.py -------------------------------------------------------------------------------- /mmcv/tests/test_utils/test_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_utils/test_path.py -------------------------------------------------------------------------------- /mmcv/tests/test_utils/test_testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_utils/test_testing.py -------------------------------------------------------------------------------- /mmcv/tests/test_utils/test_timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_utils/test_timer.py -------------------------------------------------------------------------------- /mmcv/tests/test_utils/test_trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_utils/test_trace.py -------------------------------------------------------------------------------- /mmcv/tests/test_video/test_optflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_video/test_optflow.py -------------------------------------------------------------------------------- /mmcv/tests/test_video/test_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_video/test_reader.py -------------------------------------------------------------------------------- /mmcv/tests/test_visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmcv/tests/test_visualization.py -------------------------------------------------------------------------------- /mmhuman3d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/__init__.py -------------------------------------------------------------------------------- /mmhuman3d/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/apis/__init__.py -------------------------------------------------------------------------------- /mmhuman3d/apis/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/apis/inference.py -------------------------------------------------------------------------------- /mmhuman3d/apis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/apis/test.py -------------------------------------------------------------------------------- /mmhuman3d/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/apis/train.py -------------------------------------------------------------------------------- /mmhuman3d/configs/expose/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/expose/README.md -------------------------------------------------------------------------------- /mmhuman3d/configs/expose/expose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/expose/expose.py -------------------------------------------------------------------------------- /mmhuman3d/configs/expose/metafile.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/expose/metafile.yml -------------------------------------------------------------------------------- /mmhuman3d/configs/gta_human/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/gta_human/README.md -------------------------------------------------------------------------------- /mmhuman3d/configs/hmr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/hmr/README.md -------------------------------------------------------------------------------- /mmhuman3d/configs/hmr/metafile.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/hmr/metafile.yml -------------------------------------------------------------------------------- /mmhuman3d/configs/hybrik/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/hybrik/README.md -------------------------------------------------------------------------------- /mmhuman3d/configs/hybrik/metafile.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/hybrik/metafile.yml -------------------------------------------------------------------------------- /mmhuman3d/configs/pare/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/pare/README.md -------------------------------------------------------------------------------- /mmhuman3d/configs/pare/metafile.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/pare/metafile.yml -------------------------------------------------------------------------------- /mmhuman3d/configs/render/smpl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/render/smpl.py -------------------------------------------------------------------------------- /mmhuman3d/configs/smplify/smplify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/smplify/smplify.py -------------------------------------------------------------------------------- /mmhuman3d/configs/smplify/smplifyx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/smplify/smplifyx.py -------------------------------------------------------------------------------- /mmhuman3d/configs/spin/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/spin/README.md -------------------------------------------------------------------------------- /mmhuman3d/configs/spin/metafile.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/spin/metafile.yml -------------------------------------------------------------------------------- /mmhuman3d/configs/vibe/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/vibe/README.md -------------------------------------------------------------------------------- /mmhuman3d/configs/vibe/metafile.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/configs/vibe/metafile.yml -------------------------------------------------------------------------------- /mmhuman3d/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/core/__init__.py -------------------------------------------------------------------------------- /mmhuman3d/core/cameras/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/core/cameras/__init__.py -------------------------------------------------------------------------------- /mmhuman3d/core/cameras/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/core/cameras/builder.py -------------------------------------------------------------------------------- /mmhuman3d/core/cameras/cameras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/core/cameras/cameras.py -------------------------------------------------------------------------------- /mmhuman3d/core/distributed_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/core/distributed_wrapper.py -------------------------------------------------------------------------------- /mmhuman3d/core/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/core/evaluation/__init__.py -------------------------------------------------------------------------------- /mmhuman3d/core/optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/core/optimizer/__init__.py -------------------------------------------------------------------------------- /mmhuman3d/core/optimizer/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/core/optimizer/builder.py -------------------------------------------------------------------------------- /mmhuman3d/core/renderer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/core/renderer/__init__.py -------------------------------------------------------------------------------- /mmhuman3d/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/data/__init__.py -------------------------------------------------------------------------------- /mmhuman3d/data/data_converters/eft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/data/data_converters/eft.py -------------------------------------------------------------------------------- /mmhuman3d/data/data_converters/ehf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/data/data_converters/ehf.py -------------------------------------------------------------------------------- /mmhuman3d/data/data_converters/lsp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/data/data_converters/lsp.py -------------------------------------------------------------------------------- /mmhuman3d/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/data/datasets/__init__.py -------------------------------------------------------------------------------- /mmhuman3d/data/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/data/datasets/builder.py -------------------------------------------------------------------------------- /mmhuman3d/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/__init__.py -------------------------------------------------------------------------------- /mmhuman3d/models/backbones/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/backbones/builder.py -------------------------------------------------------------------------------- /mmhuman3d/models/backbones/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/backbones/hrnet.py -------------------------------------------------------------------------------- /mmhuman3d/models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/backbones/resnet.py -------------------------------------------------------------------------------- /mmhuman3d/models/body_models/flame.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/body_models/flame.py -------------------------------------------------------------------------------- /mmhuman3d/models/body_models/mano.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/body_models/mano.py -------------------------------------------------------------------------------- /mmhuman3d/models/body_models/smpl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/body_models/smpl.py -------------------------------------------------------------------------------- /mmhuman3d/models/body_models/smplx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/body_models/smplx.py -------------------------------------------------------------------------------- /mmhuman3d/models/body_models/star.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/body_models/star.py -------------------------------------------------------------------------------- /mmhuman3d/models/body_models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/body_models/utils.py -------------------------------------------------------------------------------- /mmhuman3d/models/heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/heads/__init__.py -------------------------------------------------------------------------------- /mmhuman3d/models/heads/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/heads/builder.py -------------------------------------------------------------------------------- /mmhuman3d/models/heads/expose_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/heads/expose_head.py -------------------------------------------------------------------------------- /mmhuman3d/models/heads/hmr_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/heads/hmr_head.py -------------------------------------------------------------------------------- /mmhuman3d/models/heads/hybrik_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/heads/hybrik_head.py -------------------------------------------------------------------------------- /mmhuman3d/models/heads/pare_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/heads/pare_head.py -------------------------------------------------------------------------------- /mmhuman3d/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/losses/__init__.py -------------------------------------------------------------------------------- /mmhuman3d/models/losses/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/losses/builder.py -------------------------------------------------------------------------------- /mmhuman3d/models/losses/gan_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/losses/gan_loss.py -------------------------------------------------------------------------------- /mmhuman3d/models/losses/mse_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/losses/mse_loss.py -------------------------------------------------------------------------------- /mmhuman3d/models/losses/prior_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/losses/prior_loss.py -------------------------------------------------------------------------------- /mmhuman3d/models/losses/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/losses/utils.py -------------------------------------------------------------------------------- /mmhuman3d/models/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/necks/__init__.py -------------------------------------------------------------------------------- /mmhuman3d/models/necks/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/necks/builder.py -------------------------------------------------------------------------------- /mmhuman3d/models/utils/SMPLX.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/utils/SMPLX.py -------------------------------------------------------------------------------- /mmhuman3d/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/utils/__init__.py -------------------------------------------------------------------------------- /mmhuman3d/models/utils/fits_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/utils/fits_dict.py -------------------------------------------------------------------------------- /mmhuman3d/models/utils/res_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/models/utils/res_layer.py -------------------------------------------------------------------------------- /mmhuman3d/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/utils/__init__.py -------------------------------------------------------------------------------- /mmhuman3d/utils/camera_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/utils/camera_utils.py -------------------------------------------------------------------------------- /mmhuman3d/utils/collect_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/utils/collect_env.py -------------------------------------------------------------------------------- /mmhuman3d/utils/demo_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/utils/demo_utils.py -------------------------------------------------------------------------------- /mmhuman3d/utils/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/utils/dist_utils.py -------------------------------------------------------------------------------- /mmhuman3d/utils/ffmpeg_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/utils/ffmpeg_utils.py -------------------------------------------------------------------------------- /mmhuman3d/utils/geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/utils/geometry.py -------------------------------------------------------------------------------- /mmhuman3d/utils/keypoint_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/utils/keypoint_utils.py -------------------------------------------------------------------------------- /mmhuman3d/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/utils/logger.py -------------------------------------------------------------------------------- /mmhuman3d/utils/mesh_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/utils/mesh_utils.py -------------------------------------------------------------------------------- /mmhuman3d/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/utils/misc.py -------------------------------------------------------------------------------- /mmhuman3d/utils/path_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/utils/path_utils.py -------------------------------------------------------------------------------- /mmhuman3d/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/utils/transforms.py -------------------------------------------------------------------------------- /mmhuman3d/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/mmhuman3d/version.py -------------------------------------------------------------------------------- /mmhuman_data/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretrained/model_init_weights/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretrained/pose/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretrained/rtmpose/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretrained/transfer_data/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sapiens/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/.gitignore -------------------------------------------------------------------------------- /sapiens/CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /sapiens/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/CONTRIBUTING.md -------------------------------------------------------------------------------- /sapiens/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/LICENSE -------------------------------------------------------------------------------- /sapiens/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/README.md -------------------------------------------------------------------------------- /sapiens/_install/conda.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/_install/conda.sh -------------------------------------------------------------------------------- /sapiens/_install/make_links.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/_install/make_links.sh -------------------------------------------------------------------------------- /sapiens/cv/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/.gitignore -------------------------------------------------------------------------------- /sapiens/cv/MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/MANIFEST.in -------------------------------------------------------------------------------- /sapiens/cv/mmcv/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/__init__.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/arraymisc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/arraymisc/__init__.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/cnn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/cnn/__init__.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/cnn/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/cnn/alexnet.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/cnn/bricks/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/cnn/bricks/conv.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/cnn/bricks/conv_ws.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/cnn/bricks/conv_ws.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/cnn/bricks/drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/cnn/bricks/drop.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/cnn/bricks/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/cnn/bricks/norm.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/cnn/bricks/scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/cnn/bricks/scale.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/cnn/bricks/swish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/cnn/bricks/swish.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/cnn/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/cnn/resnet.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/cnn/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/cnn/vgg.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/image/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/image/__init__.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/image/colorspace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/image/colorspace.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/image/geometric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/image/geometric.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/image/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/image/io.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/image/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/image/misc.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/__init__.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/ball_query.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/ball_query.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/bbox.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/bezier_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/bezier_align.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/bias_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/bias_act.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/border_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/border_align.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/carafe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/carafe.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/cc_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/cc_attention.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/convex_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/convex_iou.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/corner_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/corner_pool.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/correlation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/correlation.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/csrc/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/csrc/README.md -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/deform_conv.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/focal_loss.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/group_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/group_points.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/info.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/iou3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/iou3d.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/knn.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/masked_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/masked_conv.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/merge_cells.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/merge_cells.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/nms.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/pixel_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/pixel_group.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/point_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/point_sample.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/prroi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/prroi_pool.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/psa_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/psa_mask.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/roi_align.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/roi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/roi_pool.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/saconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/saconv.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/sparse_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/sparse_conv.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/sparse_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/sparse_ops.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/sparse_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/sparse_pool.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/sync_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/sync_bn.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/three_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/three_nn.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/tin_shift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/tin_shift.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/upfirdn2d.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/ops/voxelize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/ops/voxelize.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/transforms/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/transforms/base.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/transforms/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/transforms/utils.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/utils/__init__.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/utils/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/utils/env.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/utils/ext_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/utils/ext_loader.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/version.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/video/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/video/__init__.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/video/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/video/io.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/video/optflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/video/optflow.py -------------------------------------------------------------------------------- /sapiens/cv/mmcv/video/processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/mmcv/video/processing.py -------------------------------------------------------------------------------- /sapiens/cv/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/requirements.txt -------------------------------------------------------------------------------- /sapiens/cv/requirements/build.txt: -------------------------------------------------------------------------------- 1 | pytest-runner 2 | -------------------------------------------------------------------------------- /sapiens/cv/requirements/docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/requirements/docs.txt -------------------------------------------------------------------------------- /sapiens/cv/requirements/optional.txt: -------------------------------------------------------------------------------- 1 | ninja 2 | psutil 3 | -------------------------------------------------------------------------------- /sapiens/cv/requirements/runtime.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/requirements/runtime.txt -------------------------------------------------------------------------------- /sapiens/cv/requirements/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/requirements/test.txt -------------------------------------------------------------------------------- /sapiens/cv/setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/setup.cfg -------------------------------------------------------------------------------- /sapiens/cv/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/cv/setup.py -------------------------------------------------------------------------------- /sapiens/det/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/.gitignore -------------------------------------------------------------------------------- /sapiens/det/MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/MANIFEST.in -------------------------------------------------------------------------------- /sapiens/det/mmdet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/__init__.py -------------------------------------------------------------------------------- /sapiens/det/mmdet/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/apis/__init__.py -------------------------------------------------------------------------------- /sapiens/det/mmdet/apis/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/apis/inference.py -------------------------------------------------------------------------------- /sapiens/det/mmdet/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/datasets/coco.py -------------------------------------------------------------------------------- /sapiens/det/mmdet/datasets/dsdl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/datasets/dsdl.py -------------------------------------------------------------------------------- /sapiens/det/mmdet/datasets/isaid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/datasets/isaid.py -------------------------------------------------------------------------------- /sapiens/det/mmdet/datasets/lvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/datasets/lvis.py -------------------------------------------------------------------------------- /sapiens/det/mmdet/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/datasets/utils.py -------------------------------------------------------------------------------- /sapiens/det/mmdet/datasets/v3det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/datasets/v3det.py -------------------------------------------------------------------------------- /sapiens/det/mmdet/datasets/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/datasets/voc.py -------------------------------------------------------------------------------- /sapiens/det/mmdet/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/registry.py -------------------------------------------------------------------------------- /sapiens/det/mmdet/testing/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/testing/_utils.py -------------------------------------------------------------------------------- /sapiens/det/mmdet/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/utils/__init__.py -------------------------------------------------------------------------------- /sapiens/det/mmdet/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/utils/logger.py -------------------------------------------------------------------------------- /sapiens/det/mmdet/utils/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/utils/memory.py -------------------------------------------------------------------------------- /sapiens/det/mmdet/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/utils/misc.py -------------------------------------------------------------------------------- /sapiens/det/mmdet/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/mmdet/version.py -------------------------------------------------------------------------------- /sapiens/det/model-index.yml: -------------------------------------------------------------------------------- 1 | Import: 2 | - configs/rtmdet/metafile.yml 3 | -------------------------------------------------------------------------------- /sapiens/det/pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/pytest.ini -------------------------------------------------------------------------------- /sapiens/det/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/requirements.txt -------------------------------------------------------------------------------- /sapiens/det/requirements/albu.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/requirements/albu.txt -------------------------------------------------------------------------------- /sapiens/det/requirements/build.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/requirements/build.txt -------------------------------------------------------------------------------- /sapiens/det/requirements/docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/requirements/docs.txt -------------------------------------------------------------------------------- /sapiens/det/requirements/multimodal.txt: -------------------------------------------------------------------------------- 1 | fairscale 2 | nltk 3 | pycocoevalcap 4 | transformers 5 | -------------------------------------------------------------------------------- /sapiens/det/requirements/tests.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/requirements/tests.txt -------------------------------------------------------------------------------- /sapiens/det/setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/setup.cfg -------------------------------------------------------------------------------- /sapiens/det/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/setup.py -------------------------------------------------------------------------------- /sapiens/det/tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/tools/dist_test.sh -------------------------------------------------------------------------------- /sapiens/det/tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/tools/dist_train.sh -------------------------------------------------------------------------------- /sapiens/det/tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/tools/slurm_test.sh -------------------------------------------------------------------------------- /sapiens/det/tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/tools/slurm_train.sh -------------------------------------------------------------------------------- /sapiens/det/tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/tools/test.py -------------------------------------------------------------------------------- /sapiens/det/tools/test_tracking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/tools/test_tracking.py -------------------------------------------------------------------------------- /sapiens/det/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/det/tools/train.py -------------------------------------------------------------------------------- /sapiens/docs/DEPTH_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/docs/DEPTH_README.md -------------------------------------------------------------------------------- /sapiens/docs/NORMAL_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/docs/NORMAL_README.md -------------------------------------------------------------------------------- /sapiens/docs/POSE_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/docs/POSE_README.md -------------------------------------------------------------------------------- /sapiens/docs/PRETRAIN_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/docs/PRETRAIN_README.md -------------------------------------------------------------------------------- /sapiens/docs/SEG_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/docs/SEG_README.md -------------------------------------------------------------------------------- /sapiens/engine/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/engine/.gitignore -------------------------------------------------------------------------------- /sapiens/engine/MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/engine/MANIFEST.in -------------------------------------------------------------------------------- /sapiens/engine/mmengine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/engine/mmengine/__init__.py -------------------------------------------------------------------------------- /sapiens/engine/mmengine/hub/hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/engine/mmengine/hub/hub.py -------------------------------------------------------------------------------- /sapiens/engine/mmengine/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/engine/mmengine/version.py -------------------------------------------------------------------------------- /sapiens/engine/pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/engine/pytest.ini -------------------------------------------------------------------------------- /sapiens/engine/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/engine/requirements.txt -------------------------------------------------------------------------------- /sapiens/engine/setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/engine/setup.cfg -------------------------------------------------------------------------------- /sapiens/engine/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/engine/setup.py -------------------------------------------------------------------------------- /sapiens/git_script.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/git_script.sh -------------------------------------------------------------------------------- /sapiens/lite/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/lite/README.md -------------------------------------------------------------------------------- /sapiens/lite/demo/detector_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/lite/demo/detector_utils.py -------------------------------------------------------------------------------- /sapiens/lite/demo/pose_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/lite/demo/pose_utils.py -------------------------------------------------------------------------------- /sapiens/lite/demo/vis_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/lite/demo/vis_depth.py -------------------------------------------------------------------------------- /sapiens/lite/demo/vis_normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/lite/demo/vis_normal.py -------------------------------------------------------------------------------- /sapiens/lite/demo/vis_pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/lite/demo/vis_pose.py -------------------------------------------------------------------------------- /sapiens/lite/demo/vis_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/lite/demo/vis_seg.py -------------------------------------------------------------------------------- /sapiens/lite/demo/worker_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/lite/demo/worker_pool.py -------------------------------------------------------------------------------- /sapiens/lite/docs/DEPTH_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/lite/docs/DEPTH_README.md -------------------------------------------------------------------------------- /sapiens/lite/docs/NORMAL_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/lite/docs/NORMAL_README.md -------------------------------------------------------------------------------- /sapiens/lite/docs/POSE_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/lite/docs/POSE_README.md -------------------------------------------------------------------------------- /sapiens/lite/docs/SEG_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/lite/docs/SEG_README.md -------------------------------------------------------------------------------- /sapiens/pose/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/.gitignore -------------------------------------------------------------------------------- /sapiens/pose/MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/MANIFEST.in -------------------------------------------------------------------------------- /sapiens/pose/demo/bottomup_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/demo/bottomup_demo.py -------------------------------------------------------------------------------- /sapiens/pose/demo/demo_vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/demo/demo_vis.py -------------------------------------------------------------------------------- /sapiens/pose/demo/docs/zh_cn/3d_human_pose_demo.md: -------------------------------------------------------------------------------- 1 | coming soon 2 | -------------------------------------------------------------------------------- /sapiens/pose/demo/image_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/demo/image_demo.py -------------------------------------------------------------------------------- /sapiens/pose/demo/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/demo/timer.py -------------------------------------------------------------------------------- /sapiens/pose/mmpose/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/mmpose/__init__.py -------------------------------------------------------------------------------- /sapiens/pose/mmpose/codecs/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/mmpose/codecs/base.py -------------------------------------------------------------------------------- /sapiens/pose/mmpose/codecs/spr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/mmpose/codecs/spr.py -------------------------------------------------------------------------------- /sapiens/pose/mmpose/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/mmpose/registry.py -------------------------------------------------------------------------------- /sapiens/pose/mmpose/utils/camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/mmpose/utils/camera.py -------------------------------------------------------------------------------- /sapiens/pose/mmpose/utils/hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/mmpose/utils/hooks.py -------------------------------------------------------------------------------- /sapiens/pose/mmpose/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/mmpose/utils/logger.py -------------------------------------------------------------------------------- /sapiens/pose/mmpose/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/mmpose/utils/timer.py -------------------------------------------------------------------------------- /sapiens/pose/mmpose/utils/typing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/mmpose/utils/typing.py -------------------------------------------------------------------------------- /sapiens/pose/mmpose/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/mmpose/version.py -------------------------------------------------------------------------------- /sapiens/pose/pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/pytest.ini -------------------------------------------------------------------------------- /sapiens/pose/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/requirements.txt -------------------------------------------------------------------------------- /sapiens/pose/requirements/albu.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/requirements/albu.txt -------------------------------------------------------------------------------- /sapiens/pose/requirements/build.txt: -------------------------------------------------------------------------------- 1 | # These must be installed before building mmpose 2 | numpy 3 | torch>=1.8 4 | -------------------------------------------------------------------------------- /sapiens/pose/requirements/docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/requirements/docs.txt -------------------------------------------------------------------------------- /sapiens/pose/requirements/optional.txt: -------------------------------------------------------------------------------- 1 | requests 2 | -------------------------------------------------------------------------------- /sapiens/pose/requirements/tests.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/requirements/tests.txt -------------------------------------------------------------------------------- /sapiens/pose/setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/setup.cfg -------------------------------------------------------------------------------- /sapiens/pose/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/setup.py -------------------------------------------------------------------------------- /sapiens/pose/tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/tools/dist_test.sh -------------------------------------------------------------------------------- /sapiens/pose/tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/tools/dist_train.sh -------------------------------------------------------------------------------- /sapiens/pose/tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/tools/slurm_test.sh -------------------------------------------------------------------------------- /sapiens/pose/tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/tools/slurm_train.sh -------------------------------------------------------------------------------- /sapiens/pose/tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/tools/test.py -------------------------------------------------------------------------------- /sapiens/pose/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/tools/train.py -------------------------------------------------------------------------------- /sapiens/pose/tools/vis_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pose/tools/vis_train.py -------------------------------------------------------------------------------- /sapiens/pretrain/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pretrain/.gitignore -------------------------------------------------------------------------------- /sapiens/pretrain/MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pretrain/MANIFEST.in -------------------------------------------------------------------------------- /sapiens/pretrain/demo/image_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pretrain/demo/image_demo.py -------------------------------------------------------------------------------- /sapiens/pretrain/demo/mae_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pretrain/demo/mae_demo.py -------------------------------------------------------------------------------- /sapiens/pretrain/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pretrain/requirements.txt -------------------------------------------------------------------------------- /sapiens/pretrain/requirements/mminstall.txt: -------------------------------------------------------------------------------- 1 | mmcv>=2.0.0,<2.1.0 2 | mmengine>=0.8.3,<1.0.0 3 | -------------------------------------------------------------------------------- /sapiens/pretrain/requirements/multimodal.txt: -------------------------------------------------------------------------------- 1 | pycocotools 2 | transformers>=4.28.0 3 | -------------------------------------------------------------------------------- /sapiens/pretrain/requirements/tests.txt: -------------------------------------------------------------------------------- 1 | coverage 2 | interrogate 3 | pytest 4 | -------------------------------------------------------------------------------- /sapiens/pretrain/setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pretrain/setup.cfg -------------------------------------------------------------------------------- /sapiens/pretrain/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pretrain/setup.py -------------------------------------------------------------------------------- /sapiens/pretrain/tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pretrain/tools/dist_test.sh -------------------------------------------------------------------------------- /sapiens/pretrain/tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pretrain/tools/test.py -------------------------------------------------------------------------------- /sapiens/pretrain/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/pretrain/tools/train.py -------------------------------------------------------------------------------- /sapiens/seg/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/.gitignore -------------------------------------------------------------------------------- /sapiens/seg/MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/MANIFEST.in -------------------------------------------------------------------------------- /sapiens/seg/demo/demo_depth_vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/demo/demo_depth_vis.py -------------------------------------------------------------------------------- /sapiens/seg/demo/demo_normal_vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/demo/demo_normal_vis.py -------------------------------------------------------------------------------- /sapiens/seg/demo/demo_seg_vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/demo/demo_seg_vis.py -------------------------------------------------------------------------------- /sapiens/seg/demo/image_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/demo/image_demo.py -------------------------------------------------------------------------------- /sapiens/seg/mmseg/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/mmseg/__init__.py -------------------------------------------------------------------------------- /sapiens/seg/mmseg/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/mmseg/apis/__init__.py -------------------------------------------------------------------------------- /sapiens/seg/mmseg/apis/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/mmseg/apis/inference.py -------------------------------------------------------------------------------- /sapiens/seg/mmseg/apis/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/mmseg/apis/utils.py -------------------------------------------------------------------------------- /sapiens/seg/mmseg/datasets/ade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/mmseg/datasets/ade.py -------------------------------------------------------------------------------- /sapiens/seg/mmseg/datasets/cihp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/mmseg/datasets/cihp.py -------------------------------------------------------------------------------- /sapiens/seg/mmseg/datasets/face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/mmseg/datasets/face.py -------------------------------------------------------------------------------- /sapiens/seg/mmseg/datasets/hdri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/mmseg/datasets/hdri.py -------------------------------------------------------------------------------- /sapiens/seg/mmseg/datasets/lip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/mmseg/datasets/lip.py -------------------------------------------------------------------------------- /sapiens/seg/mmseg/datasets/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/mmseg/datasets/voc.py -------------------------------------------------------------------------------- /sapiens/seg/mmseg/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/mmseg/models/builder.py -------------------------------------------------------------------------------- /sapiens/seg/mmseg/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/mmseg/utils/__init__.py -------------------------------------------------------------------------------- /sapiens/seg/mmseg/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/mmseg/utils/io.py -------------------------------------------------------------------------------- /sapiens/seg/mmseg/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/mmseg/utils/misc.py -------------------------------------------------------------------------------- /sapiens/seg/mmseg/utils/set_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/mmseg/utils/set_env.py -------------------------------------------------------------------------------- /sapiens/seg/mmseg/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/mmseg/version.py -------------------------------------------------------------------------------- /sapiens/seg/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/requirements.txt -------------------------------------------------------------------------------- /sapiens/seg/requirements/albu.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/requirements/albu.txt -------------------------------------------------------------------------------- /sapiens/seg/requirements/docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/requirements/docs.txt -------------------------------------------------------------------------------- /sapiens/seg/requirements/multimodal.txt: -------------------------------------------------------------------------------- 1 | ftfy 2 | regex 3 | -------------------------------------------------------------------------------- /sapiens/seg/requirements/runtime.txt: -------------------------------------------------------------------------------- 1 | matplotlib 2 | numpy 3 | packaging 4 | prettytable 5 | scipy 6 | -------------------------------------------------------------------------------- /sapiens/seg/requirements/tests.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/requirements/tests.txt -------------------------------------------------------------------------------- /sapiens/seg/setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/setup.cfg -------------------------------------------------------------------------------- /sapiens/seg/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/setup.py -------------------------------------------------------------------------------- /sapiens/seg/tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/tools/dist_test.sh -------------------------------------------------------------------------------- /sapiens/seg/tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/tools/dist_train.sh -------------------------------------------------------------------------------- /sapiens/seg/tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/tools/slurm_test.sh -------------------------------------------------------------------------------- /sapiens/seg/tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/tools/slurm_train.sh -------------------------------------------------------------------------------- /sapiens/seg/tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/tools/test.py -------------------------------------------------------------------------------- /sapiens/seg/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/sapiens/seg/tools/train.py -------------------------------------------------------------------------------- /scripts/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/scripts/demo.py -------------------------------------------------------------------------------- /scripts/demo_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/scripts/demo_video.py -------------------------------------------------------------------------------- /scripts/load_eval_res.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/scripts/load_eval_res.py -------------------------------------------------------------------------------- /scripts/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/scripts/test.py -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/scripts/train.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/setup.py -------------------------------------------------------------------------------- /smplx_repo/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/.gitignore -------------------------------------------------------------------------------- /smplx_repo/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/LICENSE -------------------------------------------------------------------------------- /smplx_repo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/README.md -------------------------------------------------------------------------------- /smplx_repo/examples/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/examples/demo.py -------------------------------------------------------------------------------- /smplx_repo/examples/demo_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/examples/demo_layers.py -------------------------------------------------------------------------------- /smplx_repo/optional-requirements.txt: -------------------------------------------------------------------------------- 1 | pyrender>=0.1.23 2 | shapely 3 | trimesh>=2.37.6 4 | -------------------------------------------------------------------------------- /smplx_repo/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/requirements.txt -------------------------------------------------------------------------------- /smplx_repo/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/setup.py -------------------------------------------------------------------------------- /smplx_repo/smplx/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/smplx/__init__.py -------------------------------------------------------------------------------- /smplx_repo/smplx/body_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/smplx/body_models.py -------------------------------------------------------------------------------- /smplx_repo/smplx/joint_names.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/smplx/joint_names.py -------------------------------------------------------------------------------- /smplx_repo/smplx/lbs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/smplx/lbs.py -------------------------------------------------------------------------------- /smplx_repo/smplx/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/smplx/utils.py -------------------------------------------------------------------------------- /smplx_repo/smplx/vertex_ids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/smplx/vertex_ids.py -------------------------------------------------------------------------------- /smplx_repo/tools/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/tools/README.md -------------------------------------------------------------------------------- /smplx_repo/tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/tools/__init__.py -------------------------------------------------------------------------------- /smplx_repo/tools/clean_ch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/tools/clean_ch.py -------------------------------------------------------------------------------- /smplx_repo/transfer_model/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/smplx_repo/transfer_model/README.md -------------------------------------------------------------------------------- /test_bash_slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/test_bash_slurm.sh -------------------------------------------------------------------------------- /torch-trust-ncg/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/torch-trust-ncg/.gitignore -------------------------------------------------------------------------------- /torch-trust-ncg/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/torch-trust-ncg/LICENSE -------------------------------------------------------------------------------- /torch-trust-ncg/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/torch-trust-ncg/README.md -------------------------------------------------------------------------------- /torch-trust-ncg/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/torch-trust-ncg/requirements.txt -------------------------------------------------------------------------------- /torch-trust-ncg/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/torch-trust-ncg/setup.py -------------------------------------------------------------------------------- /train_bash_slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/blade/HEAD/train_bash_slurm.sh --------------------------------------------------------------------------------