├── .DS_Store ├── LICENSE ├── README.md ├── asset ├── .DS_Store ├── UI.jpg ├── badge-website.svg ├── intro.jpg ├── logo.png └── teaser.jpg ├── automatic_label.py ├── ckpt ├── controlnet │ └── config.json └── download.sh ├── data ├── .DS_Store ├── test1 │ ├── 1.jpg │ ├── 2.jpg │ ├── 3.jpg │ ├── masks │ │ ├── mask_1.png │ │ ├── mask_2.png │ │ └── mask_3.png │ └── sketch.jpg └── test2 │ ├── 1.jpg │ ├── 2.jpg │ ├── 3.jpg │ ├── 4.jpg │ ├── 5.jpg │ ├── masks │ ├── mask_1.png │ ├── mask_2.png │ ├── mask_3.png │ ├── mask_4.png │ └── mask_5.png │ └── sketch.jpg ├── dataloader ├── coco.py ├── file_io.py ├── image_multipair_loader_add.py ├── image_pair_loader.py ├── inference_loader.py ├── transforms.py ├── utils.py └── video_loader.py ├── inference ├── gradio_app.py ├── image_pair_edit_pipeline_multi_ref.py └── multi_ref_infer.py ├── requirements.txt ├── scripts ├── .DS_Store └── multi_ref_infer.sh ├── src ├── __init__.py ├── annotator │ ├── canny │ │ └── __init__.py │ ├── hed │ │ └── __init__.py │ ├── lineart │ │ ├── LICENSE │ │ ├── __init__.py │ │ └── __pycache__ │ │ │ └── __init__.cpython-310.pyc │ ├── lineart_anime │ │ ├── LICENSE │ │ ├── __init__.py │ │ └── __pycache__ │ │ │ ├── __init__.cpython-310.pyc │ │ │ └── __init__.cpython-38.pyc │ ├── midas │ │ ├── LICENSE │ │ ├── __init__.py │ │ ├── api.py │ │ ├── midas │ │ │ ├── __init__.py │ │ │ ├── base_model.py │ │ │ ├── blocks.py │ │ │ ├── dpt_depth.py │ │ │ ├── midas_net.py │ │ │ ├── midas_net_custom.py │ │ │ ├── transforms.py │ │ │ └── vit.py │ │ └── utils.py │ ├── mlsd │ │ ├── LICENSE │ │ ├── __init__.py │ │ ├── models │ │ │ ├── mbv2_mlsd_large.py │ │ │ └── mbv2_mlsd_tiny.py │ │ └── utils.py │ ├── normalbae │ │ ├── LICENSE │ │ ├── __init__.py │ │ ├── models │ │ │ ├── NNET.py │ │ │ ├── baseline.py │ │ │ └── submodules │ │ │ │ ├── decoder.py │ │ │ │ ├── efficientnet_repo │ │ │ │ ├── .gitignore │ │ │ │ ├── BENCHMARK.md │ │ │ │ ├── LICENSE │ │ │ │ ├── README.md │ │ │ │ ├── caffe2_benchmark.py │ │ │ │ ├── caffe2_validate.py │ │ │ │ ├── data │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── dataset.py │ │ │ │ │ ├── loader.py │ │ │ │ │ ├── tf_preprocessing.py │ │ │ │ │ └── transforms.py │ │ │ │ ├── geffnet │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── activations │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── activations.py │ │ │ │ │ │ ├── activations_jit.py │ │ │ │ │ │ └── activations_me.py │ │ │ │ │ ├── config.py │ │ │ │ │ ├── conv2d_layers.py │ │ │ │ │ ├── efficientnet_builder.py │ │ │ │ │ ├── gen_efficientnet.py │ │ │ │ │ ├── helpers.py │ │ │ │ │ ├── mobilenetv3.py │ │ │ │ │ ├── model_factory.py │ │ │ │ │ └── version.py │ │ │ │ ├── hubconf.py │ │ │ │ ├── onnx_export.py │ │ │ │ ├── onnx_optimize.py │ │ │ │ ├── onnx_to_caffe.py │ │ │ │ ├── onnx_validate.py │ │ │ │ ├── requirements.txt │ │ │ │ ├── setup.py │ │ │ │ ├── utils.py │ │ │ │ └── validate.py │ │ │ │ ├── encoder.py │ │ │ │ └── submodules.py │ │ └── utils │ │ │ ├── losses.py │ │ │ └── utils.py │ ├── oneformer │ │ ├── LICENSE │ │ ├── __init__.py │ │ ├── api.py │ │ ├── configs │ │ │ ├── ade20k │ │ │ │ ├── Base-ADE20K-UnifiedSegmentation.yaml │ │ │ │ ├── oneformer_R50_bs16_160k.yaml │ │ │ │ └── oneformer_swin_large_IN21k_384_bs16_160k.yaml │ │ │ └── coco │ │ │ │ ├── Base-COCO-UnifiedSegmentation.yaml │ │ │ │ ├── oneformer_R50_bs16_50ep.yaml │ │ │ │ └── oneformer_swin_large_IN21k_384_bs16_100ep.yaml │ │ ├── detectron2 │ │ │ ├── __init__.py │ │ │ ├── checkpoint │ │ │ │ ├── __init__.py │ │ │ │ ├── c2_model_loading.py │ │ │ │ ├── catalog.py │ │ │ │ └── detection_checkpoint.py │ │ │ ├── config │ │ │ │ ├── __init__.py │ │ │ │ ├── compat.py │ │ │ │ ├── config.py │ │ │ │ ├── defaults.py │ │ │ │ ├── instantiate.py │ │ │ │ └── lazy.py │ │ │ ├── data │ │ │ │ ├── __init__.py │ │ │ │ ├── benchmark.py │ │ │ │ ├── build.py │ │ │ │ ├── catalog.py │ │ │ │ ├── common.py │ │ │ │ ├── dataset_mapper.py │ │ │ │ ├── datasets │ │ │ │ │ ├── README.md │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── builtin.py │ │ │ │ │ ├── builtin_meta.py │ │ │ │ │ ├── cityscapes.py │ │ │ │ │ ├── cityscapes_panoptic.py │ │ │ │ │ ├── coco.py │ │ │ │ │ ├── coco_panoptic.py │ │ │ │ │ ├── lvis.py │ │ │ │ │ ├── lvis_v0_5_categories.py │ │ │ │ │ ├── lvis_v1_categories.py │ │ │ │ │ ├── lvis_v1_category_image_count.py │ │ │ │ │ ├── pascal_voc.py │ │ │ │ │ └── register_coco.py │ │ │ │ ├── detection_utils.py │ │ │ │ ├── samplers │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── distributed_sampler.py │ │ │ │ │ └── grouped_batch_sampler.py │ │ │ │ └── transforms │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── augmentation.py │ │ │ │ │ ├── augmentation_impl.py │ │ │ │ │ └── transform.py │ │ │ ├── engine │ │ │ │ ├── __init__.py │ │ │ │ ├── defaults.py │ │ │ │ ├── hooks.py │ │ │ │ ├── launch.py │ │ │ │ └── train_loop.py │ │ │ ├── evaluation │ │ │ │ ├── __init__.py │ │ │ │ ├── cityscapes_evaluation.py │ │ │ │ ├── coco_evaluation.py │ │ │ │ ├── evaluator.py │ │ │ │ ├── fast_eval_api.py │ │ │ │ ├── lvis_evaluation.py │ │ │ │ ├── panoptic_evaluation.py │ │ │ │ ├── pascal_voc_evaluation.py │ │ │ │ ├── rotated_coco_evaluation.py │ │ │ │ ├── sem_seg_evaluation.py │ │ │ │ └── testing.py │ │ │ ├── export │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ ├── api.py │ │ │ │ ├── c10.py │ │ │ │ ├── caffe2_export.py │ │ │ │ ├── caffe2_inference.py │ │ │ │ ├── caffe2_modeling.py │ │ │ │ ├── caffe2_patch.py │ │ │ │ ├── flatten.py │ │ │ │ ├── shared.py │ │ │ │ ├── torchscript.py │ │ │ │ └── torchscript_patch.py │ │ │ ├── layers │ │ │ │ ├── __init__.py │ │ │ │ ├── aspp.py │ │ │ │ ├── batch_norm.py │ │ │ │ ├── blocks.py │ │ │ │ ├── csrc │ │ │ │ │ ├── README.md │ │ │ │ │ ├── ROIAlignRotated │ │ │ │ │ │ ├── ROIAlignRotated.h │ │ │ │ │ │ ├── ROIAlignRotated_cpu.cpp │ │ │ │ │ │ └── ROIAlignRotated_cuda.cu │ │ │ │ │ ├── box_iou_rotated │ │ │ │ │ │ ├── box_iou_rotated.h │ │ │ │ │ │ ├── box_iou_rotated_cpu.cpp │ │ │ │ │ │ ├── box_iou_rotated_cuda.cu │ │ │ │ │ │ └── box_iou_rotated_utils.h │ │ │ │ │ ├── cocoeval │ │ │ │ │ │ ├── cocoeval.cpp │ │ │ │ │ │ └── cocoeval.h │ │ │ │ │ ├── cuda_version.cu │ │ │ │ │ ├── deformable │ │ │ │ │ │ ├── deform_conv.h │ │ │ │ │ │ ├── deform_conv_cuda.cu │ │ │ │ │ │ └── deform_conv_cuda_kernel.cu │ │ │ │ │ ├── nms_rotated │ │ │ │ │ │ ├── nms_rotated.h │ │ │ │ │ │ ├── nms_rotated_cpu.cpp │ │ │ │ │ │ └── nms_rotated_cuda.cu │ │ │ │ │ └── vision.cpp │ │ │ │ ├── deform_conv.py │ │ │ │ ├── losses.py │ │ │ │ ├── mask_ops.py │ │ │ │ ├── nms.py │ │ │ │ ├── roi_align.py │ │ │ │ ├── roi_align_rotated.py │ │ │ │ ├── rotated_boxes.py │ │ │ │ ├── shape_spec.py │ │ │ │ └── wrappers.py │ │ │ ├── model_zoo │ │ │ │ ├── __init__.py │ │ │ │ └── model_zoo.py │ │ │ ├── modeling │ │ │ │ ├── __init__.py │ │ │ │ ├── anchor_generator.py │ │ │ │ ├── backbone │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── backbone.py │ │ │ │ │ ├── build.py │ │ │ │ │ ├── fpn.py │ │ │ │ │ ├── mvit.py │ │ │ │ │ ├── regnet.py │ │ │ │ │ ├── resnet.py │ │ │ │ │ ├── swin.py │ │ │ │ │ ├── utils.py │ │ │ │ │ └── vit.py │ │ │ │ ├── box_regression.py │ │ │ │ ├── matcher.py │ │ │ │ ├── meta_arch │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── build.py │ │ │ │ │ ├── dense_detector.py │ │ │ │ │ ├── fcos.py │ │ │ │ │ ├── panoptic_fpn.py │ │ │ │ │ ├── rcnn.py │ │ │ │ │ ├── retinanet.py │ │ │ │ │ └── semantic_seg.py │ │ │ │ ├── mmdet_wrapper.py │ │ │ │ ├── poolers.py │ │ │ │ ├── postprocessing.py │ │ │ │ ├── proposal_generator │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── build.py │ │ │ │ │ ├── proposal_utils.py │ │ │ │ │ ├── rpn.py │ │ │ │ │ └── rrpn.py │ │ │ │ ├── roi_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── box_head.py │ │ │ │ │ ├── cascade_rcnn.py │ │ │ │ │ ├── fast_rcnn.py │ │ │ │ │ ├── keypoint_head.py │ │ │ │ │ ├── mask_head.py │ │ │ │ │ ├── roi_heads.py │ │ │ │ │ └── rotated_fast_rcnn.py │ │ │ │ ├── sampling.py │ │ │ │ └── test_time_augmentation.py │ │ │ ├── projects │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ └── deeplab │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── build_solver.py │ │ │ │ │ ├── config.py │ │ │ │ │ ├── loss.py │ │ │ │ │ ├── lr_scheduler.py │ │ │ │ │ ├── resnet.py │ │ │ │ │ └── semantic_seg.py │ │ │ ├── solver │ │ │ │ ├── __init__.py │ │ │ │ ├── build.py │ │ │ │ └── lr_scheduler.py │ │ │ ├── structures │ │ │ │ ├── __init__.py │ │ │ │ ├── boxes.py │ │ │ │ ├── image_list.py │ │ │ │ ├── instances.py │ │ │ │ ├── keypoints.py │ │ │ │ ├── masks.py │ │ │ │ └── rotated_boxes.py │ │ │ ├── tracking │ │ │ │ ├── __init__.py │ │ │ │ ├── base_tracker.py │ │ │ │ ├── bbox_iou_tracker.py │ │ │ │ ├── hungarian_tracker.py │ │ │ │ ├── iou_weighted_hungarian_bbox_iou_tracker.py │ │ │ │ ├── utils.py │ │ │ │ └── vanilla_hungarian_bbox_iou_tracker.py │ │ │ └── utils │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ ├── analysis.py │ │ │ │ ├── collect_env.py │ │ │ │ ├── colormap.py │ │ │ │ ├── comm.py │ │ │ │ ├── develop.py │ │ │ │ ├── env.py │ │ │ │ ├── events.py │ │ │ │ ├── file_io.py │ │ │ │ ├── logger.py │ │ │ │ ├── memory.py │ │ │ │ ├── registry.py │ │ │ │ ├── serialize.py │ │ │ │ ├── testing.py │ │ │ │ ├── tracing.py │ │ │ │ ├── video_visualizer.py │ │ │ │ └── visualizer.py │ │ └── oneformer │ │ │ ├── __init__.py │ │ │ ├── config.py │ │ │ ├── data │ │ │ ├── __init__.py │ │ │ ├── bpe_simple_vocab_16e6.txt │ │ │ ├── bpe_simple_vocab_16e6.txt.gz │ │ │ ├── build.py │ │ │ ├── dataset_mappers │ │ │ │ ├── __init__.py │ │ │ │ ├── coco_unified_new_baseline_dataset_mapper.py │ │ │ │ ├── dataset_mapper.py │ │ │ │ └── oneformer_unified_dataset_mapper.py │ │ │ ├── datasets │ │ │ │ ├── __init__.py │ │ │ │ ├── register_ade20k_instance.py │ │ │ │ ├── register_ade20k_panoptic.py │ │ │ │ ├── register_cityscapes_panoptic.py │ │ │ │ ├── register_coco_panoptic2instance.py │ │ │ │ └── register_coco_panoptic_annos_semseg.py │ │ │ └── tokenizer.py │ │ │ ├── demo │ │ │ ├── colormap.py │ │ │ ├── defaults.py │ │ │ ├── predictor.py │ │ │ └── visualizer.py │ │ │ ├── evaluation │ │ │ ├── __init__.py │ │ │ ├── cityscapes_evaluation.py │ │ │ ├── coco_evaluator.py │ │ │ ├── detection_coco_evaluator.py │ │ │ ├── evaluator.py │ │ │ └── instance_evaluation.py │ │ │ ├── modeling │ │ │ ├── __init__.py │ │ │ ├── backbone │ │ │ │ ├── __init__.py │ │ │ │ ├── dinat.py │ │ │ │ └── swin.py │ │ │ ├── matcher.py │ │ │ ├── meta_arch │ │ │ │ ├── __init__.py │ │ │ │ └── oneformer_head.py │ │ │ ├── pixel_decoder │ │ │ │ ├── __init__.py │ │ │ │ ├── fpn.py │ │ │ │ ├── msdeformattn.py │ │ │ │ └── ops │ │ │ │ │ ├── functions │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── ms_deform_attn_func.py │ │ │ │ │ ├── make.sh │ │ │ │ │ ├── modules │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── ms_deform_attn.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 │ │ │ └── transformer_decoder │ │ │ │ ├── __init__.py │ │ │ │ ├── oneformer_transformer_decoder.py │ │ │ │ ├── position_encoding.py │ │ │ │ ├── text_transformer.py │ │ │ │ └── transformer.py │ │ │ ├── oneformer_model.py │ │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── box_ops.py │ │ │ ├── events.py │ │ │ ├── misc.py │ │ │ └── pos_embed.py │ ├── openpose │ │ ├── LICENSE │ │ ├── __init__.py │ │ ├── body.py │ │ ├── face.py │ │ ├── hand.py │ │ ├── model.py │ │ └── util.py │ ├── pidinet │ │ ├── LICENSE │ │ ├── __init__.py │ │ └── model.py │ ├── shuffle │ │ └── __init__.py │ ├── uniformer │ │ ├── LICENSE │ │ ├── __init__.py │ │ ├── configs │ │ │ └── _base_ │ │ │ │ ├── datasets │ │ │ │ ├── ade20k.py │ │ │ │ ├── chase_db1.py │ │ │ │ ├── cityscapes.py │ │ │ │ ├── cityscapes_769x769.py │ │ │ │ ├── drive.py │ │ │ │ ├── hrf.py │ │ │ │ ├── pascal_context.py │ │ │ │ ├── pascal_context_59.py │ │ │ │ ├── pascal_voc12.py │ │ │ │ ├── pascal_voc12_aug.py │ │ │ │ └── stare.py │ │ │ │ ├── default_runtime.py │ │ │ │ ├── models │ │ │ │ ├── ann_r50-d8.py │ │ │ │ ├── apcnet_r50-d8.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 │ │ │ │ ├── emanet_r50-d8.py │ │ │ │ ├── encnet_r50-d8.py │ │ │ │ ├── fast_scnn.py │ │ │ │ ├── fcn_hr18.py │ │ │ │ ├── fcn_r50-d8.py │ │ │ │ ├── fcn_unet_s5-d16.py │ │ │ │ ├── fpn_r50.py │ │ │ │ ├── fpn_uniformer.py │ │ │ │ ├── gcnet_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 │ │ │ │ ├── upernet_r50.py │ │ │ │ └── upernet_uniformer.py │ │ │ │ └── schedules │ │ │ │ ├── schedule_160k.py │ │ │ │ ├── schedule_20k.py │ │ │ │ ├── schedule_40k.py │ │ │ │ └── schedule_80k.py │ │ ├── exp │ │ │ └── upernet_global_small │ │ │ │ ├── config.py │ │ │ │ ├── run.sh │ │ │ │ ├── test.sh │ │ │ │ ├── test_config_g.py │ │ │ │ ├── test_config_h32.py │ │ │ │ └── test_config_w32.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 │ │ │ ├── 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 │ │ │ ├── ops │ │ │ │ ├── __init__.py │ │ │ │ ├── assign_score_withk.py │ │ │ │ ├── ball_query.py │ │ │ │ ├── bbox.py │ │ │ │ ├── border_align.py │ │ │ │ ├── box_iou_rotated.py │ │ │ │ ├── carafe.py │ │ │ │ ├── cc_attention.py │ │ │ │ ├── contour_expand.py │ │ │ │ ├── corner_pool.py │ │ │ │ ├── correlation.py │ │ │ │ ├── deform_conv.py │ │ │ │ ├── deform_roi_pool.py │ │ │ │ ├── deprecated_wrappers.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 │ │ │ │ ├── modulated_deform_conv.py │ │ │ │ ├── multi_scale_deform_attn.py │ │ │ │ ├── nms.py │ │ │ │ ├── pixel_group.py │ │ │ │ ├── point_sample.py │ │ │ │ ├── points_in_boxes.py │ │ │ │ ├── points_sampler.py │ │ │ │ ├── psa_mask.py │ │ │ │ ├── roi_align.py │ │ │ │ ├── roi_align_rotated.py │ │ │ │ ├── roi_pool.py │ │ │ │ ├── roiaware_pool3d.py │ │ │ │ ├── roipoint_pool3d.py │ │ │ │ ├── saconv.py │ │ │ │ ├── scatter_points.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 │ │ │ │ │ │ ├── dvclive.py │ │ │ │ │ │ ├── mlflow.py │ │ │ │ │ │ ├── neptune.py │ │ │ │ │ │ ├── pavi.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 │ │ │ ├── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── config.py │ │ │ │ ├── env.py │ │ │ │ ├── ext_loader.py │ │ │ │ ├── logging.py │ │ │ │ ├── misc.py │ │ │ │ ├── parrots_jit.py │ │ │ │ ├── parrots_wrapper.py │ │ │ │ ├── path.py │ │ │ │ ├── progressbar.py │ │ │ │ ├── registry.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 │ │ ├── mmcv_custom │ │ │ ├── __init__.py │ │ │ └── checkpoint.py │ │ └── mmseg │ │ │ ├── apis │ │ │ ├── __init__.py │ │ │ ├── inference.py │ │ │ ├── test.py │ │ │ └── train.py │ │ │ ├── core │ │ │ ├── __init__.py │ │ │ ├── evaluation │ │ │ │ ├── __init__.py │ │ │ │ ├── class_names.py │ │ │ │ ├── eval_hooks.py │ │ │ │ └── metrics.py │ │ │ ├── seg │ │ │ │ ├── __init__.py │ │ │ │ ├── builder.py │ │ │ │ └── sampler │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_pixel_sampler.py │ │ │ │ │ └── ohem_pixel_sampler.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ └── misc.py │ │ │ ├── datasets │ │ │ ├── __init__.py │ │ │ ├── ade.py │ │ │ ├── builder.py │ │ │ ├── chase_db1.py │ │ │ ├── cityscapes.py │ │ │ ├── custom.py │ │ │ ├── dataset_wrappers.py │ │ │ ├── drive.py │ │ │ ├── hrf.py │ │ │ ├── pascal_context.py │ │ │ ├── pipelines │ │ │ │ ├── __init__.py │ │ │ │ ├── compose.py │ │ │ │ ├── formating.py │ │ │ │ ├── loading.py │ │ │ │ ├── test_time_aug.py │ │ │ │ └── transforms.py │ │ │ ├── stare.py │ │ │ └── voc.py │ │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── backbones │ │ │ │ ├── __init__.py │ │ │ │ ├── cgnet.py │ │ │ │ ├── fast_scnn.py │ │ │ │ ├── hrnet.py │ │ │ │ ├── mobilenet_v2.py │ │ │ │ ├── mobilenet_v3.py │ │ │ │ ├── resnest.py │ │ │ │ ├── resnet.py │ │ │ │ ├── resnext.py │ │ │ │ ├── unet.py │ │ │ │ ├── uniformer.py │ │ │ │ └── vit.py │ │ │ ├── builder.py │ │ │ ├── decode_heads │ │ │ │ ├── __init__.py │ │ │ │ ├── ann_head.py │ │ │ │ ├── apc_head.py │ │ │ │ ├── aspp_head.py │ │ │ │ ├── cascade_decode_head.py │ │ │ │ ├── cc_head.py │ │ │ │ ├── da_head.py │ │ │ │ ├── decode_head.py │ │ │ │ ├── dm_head.py │ │ │ │ ├── dnl_head.py │ │ │ │ ├── ema_head.py │ │ │ │ ├── enc_head.py │ │ │ │ ├── fcn_head.py │ │ │ │ ├── fpn_head.py │ │ │ │ ├── gc_head.py │ │ │ │ ├── lraspp_head.py │ │ │ │ ├── nl_head.py │ │ │ │ ├── ocr_head.py │ │ │ │ ├── point_head.py │ │ │ │ ├── psa_head.py │ │ │ │ ├── psp_head.py │ │ │ │ ├── sep_aspp_head.py │ │ │ │ ├── sep_fcn_head.py │ │ │ │ └── uper_head.py │ │ │ ├── losses │ │ │ │ ├── __init__.py │ │ │ │ ├── accuracy.py │ │ │ │ ├── cross_entropy_loss.py │ │ │ │ ├── dice_loss.py │ │ │ │ ├── lovasz_loss.py │ │ │ │ └── utils.py │ │ │ ├── necks │ │ │ │ ├── __init__.py │ │ │ │ ├── fpn.py │ │ │ │ └── multilevel_neck.py │ │ │ ├── segmentors │ │ │ │ ├── __init__.py │ │ │ │ ├── base.py │ │ │ │ ├── cascade_encoder_decoder.py │ │ │ │ └── encoder_decoder.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── drop.py │ │ │ │ ├── inverted_residual.py │ │ │ │ ├── make_divisible.py │ │ │ │ ├── res_layer.py │ │ │ │ ├── se_layer.py │ │ │ │ ├── self_attention_block.py │ │ │ │ ├── up_conv_block.py │ │ │ │ └── weight_init.py │ │ │ ├── ops │ │ │ ├── __init__.py │ │ │ ├── encoding.py │ │ │ └── wrappers.py │ │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── collect_env.py │ │ │ └── logger.py │ ├── util.py │ └── zoe │ │ ├── LICENSE │ │ ├── __init__.py │ │ └── zoedepth │ │ ├── data │ │ ├── __init__.py │ │ ├── data_mono.py │ │ ├── ddad.py │ │ ├── diml_indoor_test.py │ │ ├── diml_outdoor_test.py │ │ ├── diode.py │ │ ├── hypersim.py │ │ ├── ibims.py │ │ ├── preprocess.py │ │ ├── sun_rgbd_loader.py │ │ ├── transforms.py │ │ ├── vkitti.py │ │ └── vkitti2.py │ │ ├── models │ │ ├── __init__.py │ │ ├── base_models │ │ │ ├── __init__.py │ │ │ ├── midas.py │ │ │ └── midas_repo │ │ │ │ ├── .gitignore │ │ │ │ ├── Dockerfile │ │ │ │ ├── LICENSE │ │ │ │ ├── README.md │ │ │ │ ├── environment.yaml │ │ │ │ ├── hubconf.py │ │ │ │ ├── input │ │ │ │ └── .placeholder │ │ │ │ ├── midas │ │ │ │ ├── backbones │ │ │ │ │ ├── beit.py │ │ │ │ │ ├── levit.py │ │ │ │ │ ├── next_vit.py │ │ │ │ │ ├── swin.py │ │ │ │ │ ├── swin2.py │ │ │ │ │ ├── swin_common.py │ │ │ │ │ ├── utils.py │ │ │ │ │ └── vit.py │ │ │ │ ├── base_model.py │ │ │ │ ├── blocks.py │ │ │ │ ├── dpt_depth.py │ │ │ │ ├── midas_net.py │ │ │ │ ├── midas_net_custom.py │ │ │ │ ├── model_loader.py │ │ │ │ └── transforms.py │ │ │ │ ├── mobile │ │ │ │ ├── README.md │ │ │ │ ├── android │ │ │ │ │ ├── .gitignore │ │ │ │ │ ├── EXPLORE_THE_CODE.md │ │ │ │ │ ├── LICENSE │ │ │ │ │ ├── README.md │ │ │ │ │ ├── app │ │ │ │ │ │ ├── .gitignore │ │ │ │ │ │ ├── build.gradle │ │ │ │ │ │ ├── proguard-rules.pro │ │ │ │ │ │ └── src │ │ │ │ │ │ │ ├── androidTest │ │ │ │ │ │ │ ├── assets │ │ │ │ │ │ │ │ ├── fox-mobilenet_v1_1.0_224_support.txt │ │ │ │ │ │ │ │ └── fox-mobilenet_v1_1.0_224_task_api.txt │ │ │ │ │ │ │ └── java │ │ │ │ │ │ │ │ ├── AndroidManifest.xml │ │ │ │ │ │ │ │ └── org │ │ │ │ │ │ │ │ └── tensorflow │ │ │ │ │ │ │ │ └── lite │ │ │ │ │ │ │ │ └── examples │ │ │ │ │ │ │ │ └── classification │ │ │ │ │ │ │ │ └── ClassifierTest.java │ │ │ │ │ │ │ └── main │ │ │ │ │ │ │ ├── AndroidManifest.xml │ │ │ │ │ │ │ ├── java │ │ │ │ │ │ │ └── org │ │ │ │ │ │ │ │ └── tensorflow │ │ │ │ │ │ │ │ └── lite │ │ │ │ │ │ │ │ └── examples │ │ │ │ │ │ │ │ └── classification │ │ │ │ │ │ │ │ ├── CameraActivity.java │ │ │ │ │ │ │ │ ├── CameraConnectionFragment.java │ │ │ │ │ │ │ │ ├── ClassifierActivity.java │ │ │ │ │ │ │ │ ├── LegacyCameraConnectionFragment.java │ │ │ │ │ │ │ │ └── customview │ │ │ │ │ │ │ │ ├── AutoFitTextureView.java │ │ │ │ │ │ │ │ ├── OverlayView.java │ │ │ │ │ │ │ │ ├── RecognitionScoreView.java │ │ │ │ │ │ │ │ └── ResultsView.java │ │ │ │ │ │ │ └── res │ │ │ │ │ │ │ ├── drawable-v24 │ │ │ │ │ │ │ └── ic_launcher_foreground.xml │ │ │ │ │ │ │ ├── drawable │ │ │ │ │ │ │ ├── bottom_sheet_bg.xml │ │ │ │ │ │ │ ├── ic_baseline_add.xml │ │ │ │ │ │ │ ├── ic_baseline_remove.xml │ │ │ │ │ │ │ ├── ic_launcher_background.xml │ │ │ │ │ │ │ └── rectangle.xml │ │ │ │ │ │ │ ├── layout │ │ │ │ │ │ │ ├── tfe_ic_activity_camera.xml │ │ │ │ │ │ │ ├── tfe_ic_camera_connection_fragment.xml │ │ │ │ │ │ │ └── tfe_ic_layout_bottom_sheet.xml │ │ │ │ │ │ │ ├── mipmap-anydpi-v26 │ │ │ │ │ │ │ ├── ic_launcher.xml │ │ │ │ │ │ │ └── ic_launcher_round.xml │ │ │ │ │ │ │ └── values │ │ │ │ │ │ │ ├── colors.xml │ │ │ │ │ │ │ ├── dimens.xml │ │ │ │ │ │ │ ├── strings.xml │ │ │ │ │ │ │ └── styles.xml │ │ │ │ │ ├── build.gradle │ │ │ │ │ ├── gradle.properties │ │ │ │ │ ├── gradle │ │ │ │ │ │ └── wrapper │ │ │ │ │ │ │ ├── gradle-wrapper.jar │ │ │ │ │ │ │ └── gradle-wrapper.properties │ │ │ │ │ ├── gradlew │ │ │ │ │ ├── gradlew.bat │ │ │ │ │ ├── lib_support │ │ │ │ │ │ ├── build.gradle │ │ │ │ │ │ ├── proguard-rules.pro │ │ │ │ │ │ └── src │ │ │ │ │ │ │ └── main │ │ │ │ │ │ │ ├── AndroidManifest.xml │ │ │ │ │ │ │ └── java │ │ │ │ │ │ │ └── org │ │ │ │ │ │ │ └── tensorflow │ │ │ │ │ │ │ └── lite │ │ │ │ │ │ │ └── examples │ │ │ │ │ │ │ └── classification │ │ │ │ │ │ │ └── tflite │ │ │ │ │ │ │ ├── Classifier.java │ │ │ │ │ │ │ ├── ClassifierFloatEfficientNet.java │ │ │ │ │ │ │ ├── ClassifierFloatMobileNet.java │ │ │ │ │ │ │ ├── ClassifierQuantizedEfficientNet.java │ │ │ │ │ │ │ └── ClassifierQuantizedMobileNet.java │ │ │ │ │ ├── lib_task_api │ │ │ │ │ │ ├── build.gradle │ │ │ │ │ │ ├── proguard-rules.pro │ │ │ │ │ │ └── src │ │ │ │ │ │ │ └── main │ │ │ │ │ │ │ ├── AndroidManifest.xml │ │ │ │ │ │ │ └── java │ │ │ │ │ │ │ └── org │ │ │ │ │ │ │ └── tensorflow │ │ │ │ │ │ │ └── lite │ │ │ │ │ │ │ └── examples │ │ │ │ │ │ │ └── classification │ │ │ │ │ │ │ └── tflite │ │ │ │ │ │ │ ├── Classifier.java │ │ │ │ │ │ │ ├── ClassifierFloatEfficientNet.java │ │ │ │ │ │ │ ├── ClassifierFloatMobileNet.java │ │ │ │ │ │ │ ├── ClassifierQuantizedEfficientNet.java │ │ │ │ │ │ │ └── ClassifierQuantizedMobileNet.java │ │ │ │ │ ├── models │ │ │ │ │ │ ├── build.gradle │ │ │ │ │ │ ├── download.gradle │ │ │ │ │ │ ├── proguard-rules.pro │ │ │ │ │ │ └── src │ │ │ │ │ │ │ └── main │ │ │ │ │ │ │ ├── AndroidManifest.xml │ │ │ │ │ │ │ └── assets │ │ │ │ │ │ │ ├── labels.txt │ │ │ │ │ │ │ ├── labels_without_background.txt │ │ │ │ │ │ │ └── run_tflite.py │ │ │ │ │ └── settings.gradle │ │ │ │ └── ios │ │ │ │ │ ├── .gitignore │ │ │ │ │ ├── LICENSE │ │ │ │ │ ├── Midas.xcodeproj │ │ │ │ │ ├── project.pbxproj │ │ │ │ │ ├── project.xcworkspace │ │ │ │ │ │ ├── contents.xcworkspacedata │ │ │ │ │ │ ├── xcshareddata │ │ │ │ │ │ │ └── IDEWorkspaceChecks.plist │ │ │ │ │ │ └── xcuserdata │ │ │ │ │ │ │ └── admin.xcuserdatad │ │ │ │ │ │ │ └── UserInterfaceState.xcuserstate │ │ │ │ │ └── xcuserdata │ │ │ │ │ │ └── admin.xcuserdatad │ │ │ │ │ │ └── xcschemes │ │ │ │ │ │ └── xcschememanagement.plist │ │ │ │ │ ├── Midas │ │ │ │ │ ├── AppDelegate.swift │ │ │ │ │ ├── Assets.xcassets │ │ │ │ │ │ ├── AppIcon.appiconset │ │ │ │ │ │ │ └── Contents.json │ │ │ │ │ │ └── Contents.json │ │ │ │ │ ├── Camera Feed │ │ │ │ │ │ ├── CameraFeedManager.swift │ │ │ │ │ │ └── PreviewView.swift │ │ │ │ │ ├── Cells │ │ │ │ │ │ └── InfoCell.swift │ │ │ │ │ ├── Constants.swift │ │ │ │ │ ├── Extensions │ │ │ │ │ │ ├── CGSizeExtension.swift │ │ │ │ │ │ ├── CVPixelBufferExtension.swift │ │ │ │ │ │ └── TFLiteExtension.swift │ │ │ │ │ ├── Info.plist │ │ │ │ │ ├── ModelDataHandler │ │ │ │ │ │ └── ModelDataHandler.swift │ │ │ │ │ ├── Storyboards │ │ │ │ │ │ └── Base.lproj │ │ │ │ │ │ │ ├── Launch Screen.storyboard │ │ │ │ │ │ │ └── Main.storyboard │ │ │ │ │ ├── ViewControllers │ │ │ │ │ │ └── ViewController.swift │ │ │ │ │ └── Views │ │ │ │ │ │ └── OverlayView.swift │ │ │ │ │ ├── Podfile │ │ │ │ │ ├── README.md │ │ │ │ │ └── RunScripts │ │ │ │ │ └── download_models.sh │ │ │ │ ├── output │ │ │ │ └── .placeholder │ │ │ │ ├── ros │ │ │ │ ├── LICENSE │ │ │ │ ├── README.md │ │ │ │ ├── additions │ │ │ │ │ ├── do_catkin_make.sh │ │ │ │ │ ├── downloads.sh │ │ │ │ │ ├── install_ros_melodic_ubuntu_17_18.sh │ │ │ │ │ ├── install_ros_noetic_ubuntu_20.sh │ │ │ │ │ └── make_package_cpp.sh │ │ │ │ ├── launch_midas_cpp.sh │ │ │ │ ├── midas_cpp │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ ├── launch │ │ │ │ │ │ ├── midas_cpp.launch │ │ │ │ │ │ └── midas_talker_listener.launch │ │ │ │ │ ├── package.xml │ │ │ │ │ ├── scripts │ │ │ │ │ │ ├── listener.py │ │ │ │ │ │ ├── listener_original.py │ │ │ │ │ │ └── talker.py │ │ │ │ │ └── src │ │ │ │ │ │ └── main.cpp │ │ │ │ └── run_talker_listener_test.sh │ │ │ │ ├── run.py │ │ │ │ ├── tf │ │ │ │ ├── README.md │ │ │ │ ├── input │ │ │ │ │ └── .placeholder │ │ │ │ ├── make_onnx_model.py │ │ │ │ ├── output │ │ │ │ │ └── .placeholder │ │ │ │ ├── run_onnx.py │ │ │ │ ├── run_pb.py │ │ │ │ ├── transforms.py │ │ │ │ └── utils.py │ │ │ │ ├── utils.py │ │ │ │ └── weights │ │ │ │ └── .placeholder │ │ ├── builder.py │ │ ├── depth_model.py │ │ ├── layers │ │ │ ├── attractor.py │ │ │ ├── dist_layers.py │ │ │ ├── localbins_layers.py │ │ │ └── patch_transformer.py │ │ ├── model_io.py │ │ ├── zoedepth │ │ │ ├── __init__.py │ │ │ ├── config_zoedepth.json │ │ │ ├── config_zoedepth_kitti.json │ │ │ └── zoedepth_v1.py │ │ └── zoedepth_nk │ │ │ ├── __init__.py │ │ │ ├── config_zoedepth_nk.json │ │ │ └── zoedepth_nk_v1.py │ │ ├── trainers │ │ ├── base_trainer.py │ │ ├── builder.py │ │ ├── loss.py │ │ ├── zoedepth_nk_trainer.py │ │ └── zoedepth_trainer.py │ │ └── utils │ │ ├── __init__.py │ │ ├── arg_utils.py │ │ ├── config.py │ │ ├── easydict │ │ └── __init__.py │ │ ├── geometry.py │ │ └── misc.py ├── diffusers │ ├── __init__.py │ ├── callbacks.py │ ├── commands │ │ ├── __init__.py │ │ ├── diffusers_cli.py │ │ ├── env.py │ │ └── fp16_safetensors.py │ ├── configuration_utils.py │ ├── dependency_versions_check.py │ ├── dependency_versions_table.py │ ├── experimental │ │ ├── README.md │ │ ├── __init__.py │ │ └── rl │ │ │ ├── __init__.py │ │ │ └── value_guided_sampling.py │ ├── image_processor.py │ ├── loaders │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-310.pyc │ │ │ ├── __init__.cpython-38.pyc │ │ │ ├── controlnet.cpython-310.pyc │ │ │ ├── peft.cpython-310.pyc │ │ │ ├── peft.cpython-38.pyc │ │ │ ├── single_file_model.cpython-310.pyc │ │ │ ├── single_file_model.cpython-38.pyc │ │ │ ├── single_file_utils.cpython-310.pyc │ │ │ ├── single_file_utils.cpython-38.pyc │ │ │ ├── unet.cpython-310.pyc │ │ │ ├── unet.cpython-38.pyc │ │ │ ├── unet_loader_utils.cpython-310.pyc │ │ │ ├── unet_loader_utils.cpython-38.pyc │ │ │ ├── utils.cpython-310.pyc │ │ │ └── utils.cpython-38.pyc │ │ ├── autoencoder.py │ │ ├── controlnet.py │ │ ├── ip_adapter.py │ │ ├── lora.py │ │ ├── lora_conversion_utils.py │ │ ├── peft.py │ │ ├── single_file.py │ │ ├── single_file_model.py │ │ ├── single_file_utils.py │ │ ├── textual_inversion.py │ │ ├── unet.py │ │ ├── unet_loader_utils.py │ │ └── utils.py │ ├── models │ │ ├── README.md │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-310.pyc │ │ │ ├── __init__.cpython-38.pyc │ │ │ ├── activations.cpython-310.pyc │ │ │ ├── activations.cpython-38.pyc │ │ │ ├── attention.cpython-310.pyc │ │ │ ├── attention.cpython-38.pyc │ │ │ ├── attention_processor.cpython-310.pyc │ │ │ ├── attention_processor.cpython-38.pyc │ │ │ ├── controlnet.cpython-310.pyc │ │ │ ├── controlnet.cpython-38.pyc │ │ │ ├── controlnet1x1.cpython-310.pyc │ │ │ ├── downsampling.cpython-310.pyc │ │ │ ├── downsampling.cpython-38.pyc │ │ │ ├── dual_transformer_2d.cpython-310.pyc │ │ │ ├── dual_transformer_2d.cpython-38.pyc │ │ │ ├── embeddings.cpython-310.pyc │ │ │ ├── embeddings.cpython-38.pyc │ │ │ ├── lora.cpython-310.pyc │ │ │ ├── lora.cpython-38.pyc │ │ │ ├── model_loading_utils.cpython-310.pyc │ │ │ ├── model_loading_utils.cpython-38.pyc │ │ │ ├── modeling_outputs.cpython-310.pyc │ │ │ ├── modeling_outputs.cpython-38.pyc │ │ │ ├── modeling_utils.cpython-310.pyc │ │ │ ├── modeling_utils.cpython-38.pyc │ │ │ ├── normalization.cpython-310.pyc │ │ │ ├── normalization.cpython-38.pyc │ │ │ ├── resnet.cpython-310.pyc │ │ │ ├── resnet.cpython-38.pyc │ │ │ ├── upsampling.cpython-310.pyc │ │ │ └── upsampling.cpython-38.pyc │ │ ├── activations.py │ │ ├── adapter.py │ │ ├── attention.py │ │ ├── attention_flax.py │ │ ├── attention_processor.py │ │ ├── autoencoders │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ ├── __init__.cpython-310.pyc │ │ │ │ ├── __init__.cpython-38.pyc │ │ │ │ ├── autoencoder_asym_kl.cpython-310.pyc │ │ │ │ ├── autoencoder_asym_kl.cpython-38.pyc │ │ │ │ ├── autoencoder_kl.cpython-310.pyc │ │ │ │ ├── autoencoder_kl.cpython-38.pyc │ │ │ │ ├── autoencoder_kl_temporal_decoder.cpython-310.pyc │ │ │ │ ├── autoencoder_kl_temporal_decoder.cpython-38.pyc │ │ │ │ ├── autoencoder_tiny.cpython-310.pyc │ │ │ │ ├── autoencoder_tiny.cpython-38.pyc │ │ │ │ ├── consistency_decoder_vae.cpython-310.pyc │ │ │ │ ├── consistency_decoder_vae.cpython-38.pyc │ │ │ │ ├── vae.cpython-310.pyc │ │ │ │ └── vae.cpython-38.pyc │ │ │ ├── autoencoder_asym_kl.py │ │ │ ├── autoencoder_kl.py │ │ │ ├── autoencoder_kl_temporal_decoder.py │ │ │ ├── autoencoder_tiny.py │ │ │ ├── consistency_decoder_vae.py │ │ │ └── vae.py │ │ ├── controlnet.py │ │ ├── controlnet1x1.py │ │ ├── controlnet_flax.py │ │ ├── controlnet_sin.py │ │ ├── controlnet_xs.py │ │ ├── downsampling.py │ │ ├── dual_transformer_2d.py │ │ ├── embeddings.py │ │ ├── embeddings_flax.py │ │ ├── lora.py │ │ ├── model_loading_utils.py │ │ ├── modeling_flax_pytorch_utils.py │ │ ├── modeling_flax_utils.py │ │ ├── modeling_outputs.py │ │ ├── modeling_pytorch_flax_utils.py │ │ ├── modeling_utils.py │ │ ├── normalization.py │ │ ├── prior_transformer.py │ │ ├── resnet.py │ │ ├── resnet_flax.py │ │ ├── t5_film_transformer.py │ │ ├── transformer_2d.py │ │ ├── transformer_temporal.py │ │ ├── transformers │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ ├── __init__.cpython-310.pyc │ │ │ │ ├── __init__.cpython-38.pyc │ │ │ │ ├── dual_transformer_2d.cpython-310.pyc │ │ │ │ ├── dual_transformer_2d.cpython-38.pyc │ │ │ │ ├── prior_transformer.cpython-310.pyc │ │ │ │ ├── prior_transformer.cpython-38.pyc │ │ │ │ ├── t5_film_transformer.cpython-310.pyc │ │ │ │ ├── t5_film_transformer.cpython-38.pyc │ │ │ │ ├── transformer_2d.cpython-310.pyc │ │ │ │ ├── transformer_2d.cpython-38.pyc │ │ │ │ ├── transformer_temporal.cpython-310.pyc │ │ │ │ └── transformer_temporal.cpython-38.pyc │ │ │ ├── dual_transformer_2d.py │ │ │ ├── prior_transformer.py │ │ │ ├── t5_film_transformer.py │ │ │ ├── transformer_2d.py │ │ │ └── transformer_temporal.py │ │ ├── unet_1d.py │ │ ├── unet_1d_blocks.py │ │ ├── unet_2d.py │ │ ├── unet_2d_blocks.py │ │ ├── unet_2d_condition.py │ │ ├── unets │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ ├── __init__.cpython-310.pyc │ │ │ │ ├── __init__.cpython-38.pyc │ │ │ │ ├── unet_1d.cpython-310.pyc │ │ │ │ ├── unet_1d.cpython-38.pyc │ │ │ │ ├── unet_1d_blocks.cpython-310.pyc │ │ │ │ ├── unet_1d_blocks.cpython-38.pyc │ │ │ │ ├── unet_2d.cpython-310.pyc │ │ │ │ ├── unet_2d.cpython-38.pyc │ │ │ │ ├── unet_2d_blocks.cpython-310.pyc │ │ │ │ ├── unet_2d_blocks.cpython-38.pyc │ │ │ │ ├── unet_2d_condition.cpython-310.pyc │ │ │ │ ├── unet_2d_condition.cpython-38.pyc │ │ │ │ ├── unet_3d_blocks.cpython-310.pyc │ │ │ │ ├── unet_3d_blocks.cpython-38.pyc │ │ │ │ ├── unet_3d_condition.cpython-310.pyc │ │ │ │ ├── unet_3d_condition.cpython-38.pyc │ │ │ │ ├── unet_i2vgen_xl.cpython-310.pyc │ │ │ │ ├── unet_i2vgen_xl.cpython-38.pyc │ │ │ │ ├── unet_kandinsky3.cpython-310.pyc │ │ │ │ ├── unet_kandinsky3.cpython-38.pyc │ │ │ │ ├── unet_motion_model.cpython-310.pyc │ │ │ │ ├── unet_motion_model.cpython-38.pyc │ │ │ │ ├── unet_spatio_temporal_condition.cpython-310.pyc │ │ │ │ ├── unet_spatio_temporal_condition.cpython-38.pyc │ │ │ │ ├── unet_stable_cascade.cpython-310.pyc │ │ │ │ ├── unet_stable_cascade.cpython-38.pyc │ │ │ │ ├── uvit_2d.cpython-310.pyc │ │ │ │ └── uvit_2d.cpython-38.pyc │ │ │ ├── unet_1d.py │ │ │ ├── unet_1d_blocks.py │ │ │ ├── unet_2d.py │ │ │ ├── unet_2d_blocks.py │ │ │ ├── unet_2d_blocks_flax.py │ │ │ ├── unet_2d_condition.py │ │ │ ├── unet_2d_condition_flax.py │ │ │ ├── unet_3d_blocks.py │ │ │ ├── unet_3d_condition.py │ │ │ ├── unet_i2vgen_xl.py │ │ │ ├── unet_kandinsky3.py │ │ │ ├── unet_motion_model.py │ │ │ ├── unet_spatio_temporal_condition.py │ │ │ ├── unet_stable_cascade.py │ │ │ └── uvit_2d.py │ │ ├── upsampling.py │ │ ├── vae_flax.py │ │ └── vq_model.py │ ├── optimization.py │ ├── pipelines │ │ ├── README.md │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-310.pyc │ │ │ ├── __init__.cpython-38.pyc │ │ │ ├── pipeline_loading_utils.cpython-310.pyc │ │ │ ├── pipeline_loading_utils.cpython-38.pyc │ │ │ ├── pipeline_utils.cpython-310.pyc │ │ │ └── pipeline_utils.cpython-38.pyc │ │ ├── amused │ │ │ ├── __init__.py │ │ │ ├── pipeline_amused.py │ │ │ ├── pipeline_amused_img2img.py │ │ │ └── pipeline_amused_inpaint.py │ │ ├── animatediff │ │ │ ├── __init__.py │ │ │ ├── pipeline_animatediff.py │ │ │ ├── pipeline_animatediff_sdxl.py │ │ │ ├── pipeline_animatediff_video2video.py │ │ │ └── pipeline_output.py │ │ ├── audioldm │ │ │ ├── __init__.py │ │ │ └── pipeline_audioldm.py │ │ ├── audioldm2 │ │ │ ├── __init__.py │ │ │ ├── modeling_audioldm2.py │ │ │ └── pipeline_audioldm2.py │ │ ├── auto_pipeline.py │ │ ├── blip_diffusion │ │ │ ├── __init__.py │ │ │ ├── blip_image_processing.py │ │ │ ├── modeling_blip2.py │ │ │ ├── modeling_ctx_clip.py │ │ │ └── pipeline_blip_diffusion.py │ │ ├── consistency_models │ │ │ ├── __init__.py │ │ │ └── pipeline_consistency_models.py │ │ ├── controlnet │ │ │ ├── __init__.py │ │ │ ├── multicontrolnet.py │ │ │ ├── pipeline_controlnet.py │ │ │ ├── pipeline_controlnet_blip_diffusion.py │ │ │ ├── pipeline_controlnet_img2img.py │ │ │ ├── pipeline_controlnet_inpaint.py │ │ │ ├── pipeline_controlnet_inpaint_sd_xl.py │ │ │ ├── pipeline_controlnet_sd_xl.py │ │ │ ├── pipeline_controlnet_sd_xl_img2img.py │ │ │ └── pipeline_flax_controlnet.py │ │ ├── controlnet_xs │ │ │ ├── __init__.py │ │ │ ├── pipeline_controlnet_xs.py │ │ │ └── pipeline_controlnet_xs_sd_xl.py │ │ ├── dance_diffusion │ │ │ ├── __init__.py │ │ │ └── pipeline_dance_diffusion.py │ │ ├── ddim │ │ │ ├── __init__.py │ │ │ └── pipeline_ddim.py │ │ ├── ddpm │ │ │ ├── __init__.py │ │ │ └── pipeline_ddpm.py │ │ ├── deepfloyd_if │ │ │ ├── __init__.py │ │ │ ├── pipeline_if.py │ │ │ ├── pipeline_if_img2img.py │ │ │ ├── pipeline_if_img2img_superresolution.py │ │ │ ├── pipeline_if_inpainting.py │ │ │ ├── pipeline_if_inpainting_superresolution.py │ │ │ ├── pipeline_if_superresolution.py │ │ │ ├── pipeline_output.py │ │ │ ├── safety_checker.py │ │ │ ├── timesteps.py │ │ │ └── watermark.py │ │ ├── deprecated │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── alt_diffusion │ │ │ │ ├── __init__.py │ │ │ │ ├── modeling_roberta_series.py │ │ │ │ ├── pipeline_alt_diffusion.py │ │ │ │ ├── pipeline_alt_diffusion_img2img.py │ │ │ │ └── pipeline_output.py │ │ │ ├── audio_diffusion │ │ │ │ ├── __init__.py │ │ │ │ ├── mel.py │ │ │ │ └── pipeline_audio_diffusion.py │ │ │ ├── latent_diffusion_uncond │ │ │ │ ├── __init__.py │ │ │ │ └── pipeline_latent_diffusion_uncond.py │ │ │ ├── pndm │ │ │ │ ├── __init__.py │ │ │ │ └── pipeline_pndm.py │ │ │ ├── repaint │ │ │ │ ├── __init__.py │ │ │ │ └── pipeline_repaint.py │ │ │ ├── score_sde_ve │ │ │ │ ├── __init__.py │ │ │ │ └── pipeline_score_sde_ve.py │ │ │ ├── spectrogram_diffusion │ │ │ │ ├── __init__.py │ │ │ │ ├── continuous_encoder.py │ │ │ │ ├── midi_utils.py │ │ │ │ ├── notes_encoder.py │ │ │ │ └── pipeline_spectrogram_diffusion.py │ │ │ ├── stable_diffusion_variants │ │ │ │ ├── __init__.py │ │ │ │ ├── pipeline_cycle_diffusion.py │ │ │ │ ├── pipeline_onnx_stable_diffusion_inpaint_legacy.py │ │ │ │ ├── pipeline_stable_diffusion_inpaint_legacy.py │ │ │ │ ├── pipeline_stable_diffusion_model_editing.py │ │ │ │ ├── pipeline_stable_diffusion_paradigms.py │ │ │ │ └── pipeline_stable_diffusion_pix2pix_zero.py │ │ │ ├── stochastic_karras_ve │ │ │ │ ├── __init__.py │ │ │ │ └── pipeline_stochastic_karras_ve.py │ │ │ ├── versatile_diffusion │ │ │ │ ├── __init__.py │ │ │ │ ├── modeling_text_unet.py │ │ │ │ ├── pipeline_versatile_diffusion.py │ │ │ │ ├── pipeline_versatile_diffusion_dual_guided.py │ │ │ │ ├── pipeline_versatile_diffusion_image_variation.py │ │ │ │ └── pipeline_versatile_diffusion_text_to_image.py │ │ │ └── vq_diffusion │ │ │ │ ├── __init__.py │ │ │ │ └── pipeline_vq_diffusion.py │ │ ├── dit │ │ │ ├── __init__.py │ │ │ └── pipeline_dit.py │ │ ├── free_init_utils.py │ │ ├── i2vgen_xl │ │ │ ├── __init__.py │ │ │ └── pipeline_i2vgen_xl.py │ │ ├── kandinsky │ │ │ ├── __init__.py │ │ │ ├── pipeline_kandinsky.py │ │ │ ├── pipeline_kandinsky_combined.py │ │ │ ├── pipeline_kandinsky_img2img.py │ │ │ ├── pipeline_kandinsky_inpaint.py │ │ │ ├── pipeline_kandinsky_prior.py │ │ │ └── text_encoder.py │ │ ├── kandinsky2_2 │ │ │ ├── __init__.py │ │ │ ├── pipeline_kandinsky2_2.py │ │ │ ├── pipeline_kandinsky2_2_combined.py │ │ │ ├── pipeline_kandinsky2_2_controlnet.py │ │ │ ├── pipeline_kandinsky2_2_controlnet_img2img.py │ │ │ ├── pipeline_kandinsky2_2_img2img.py │ │ │ ├── pipeline_kandinsky2_2_inpainting.py │ │ │ ├── pipeline_kandinsky2_2_prior.py │ │ │ └── pipeline_kandinsky2_2_prior_emb2emb.py │ │ ├── kandinsky3 │ │ │ ├── __init__.py │ │ │ ├── convert_kandinsky3_unet.py │ │ │ ├── pipeline_kandinsky3.py │ │ │ └── pipeline_kandinsky3_img2img.py │ │ ├── latent_consistency_models │ │ │ ├── __init__.py │ │ │ ├── pipeline_latent_consistency_img2img.py │ │ │ └── pipeline_latent_consistency_text2img.py │ │ ├── latent_diffusion │ │ │ ├── __init__.py │ │ │ ├── pipeline_latent_diffusion.py │ │ │ └── pipeline_latent_diffusion_superresolution.py │ │ ├── ledits_pp │ │ │ ├── __init__.py │ │ │ ├── pipeline_leditspp_stable_diffusion.py │ │ │ ├── pipeline_leditspp_stable_diffusion_xl.py │ │ │ └── pipeline_output.py │ │ ├── marigold │ │ │ ├── __init__.py │ │ │ ├── marigold_image_processing.py │ │ │ ├── pipeline_marigold_depth.py │ │ │ └── pipeline_marigold_normals.py │ │ ├── musicldm │ │ │ ├── __init__.py │ │ │ └── pipeline_musicldm.py │ │ ├── onnx_utils.py │ │ ├── paint_by_example │ │ │ ├── __init__.py │ │ │ ├── image_encoder.py │ │ │ └── pipeline_paint_by_example.py │ │ ├── pia │ │ │ ├── __init__.py │ │ │ └── pipeline_pia.py │ │ ├── pipeline_flax_utils.py │ │ ├── pipeline_loading_utils.py │ │ ├── pipeline_utils.py │ │ ├── pixart_alpha │ │ │ ├── __init__.py │ │ │ ├── pipeline_pixart_alpha.py │ │ │ └── pipeline_pixart_sigma.py │ │ ├── semantic_stable_diffusion │ │ │ ├── __init__.py │ │ │ ├── pipeline_output.py │ │ │ └── pipeline_semantic_stable_diffusion.py │ │ ├── shap_e │ │ │ ├── __init__.py │ │ │ ├── camera.py │ │ │ ├── pipeline_shap_e.py │ │ │ ├── pipeline_shap_e_img2img.py │ │ │ └── renderer.py │ │ ├── stable_cascade │ │ │ ├── __init__.py │ │ │ ├── pipeline_stable_cascade.py │ │ │ ├── pipeline_stable_cascade_combined.py │ │ │ └── pipeline_stable_cascade_prior.py │ │ ├── stable_diffusion │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── clip_image_project_model.py │ │ │ ├── convert_from_ckpt.py │ │ │ ├── pipeline_flax_stable_diffusion.py │ │ │ ├── pipeline_flax_stable_diffusion_img2img.py │ │ │ ├── pipeline_flax_stable_diffusion_inpaint.py │ │ │ ├── pipeline_onnx_stable_diffusion.py │ │ │ ├── pipeline_onnx_stable_diffusion_img2img.py │ │ │ ├── pipeline_onnx_stable_diffusion_inpaint.py │ │ │ ├── pipeline_onnx_stable_diffusion_upscale.py │ │ │ ├── pipeline_output.py │ │ │ ├── pipeline_stable_diffusion.py │ │ │ ├── pipeline_stable_diffusion_depth2img.py │ │ │ ├── pipeline_stable_diffusion_image_variation.py │ │ │ ├── pipeline_stable_diffusion_img2img.py │ │ │ ├── pipeline_stable_diffusion_inpaint.py │ │ │ ├── pipeline_stable_diffusion_instruct_pix2pix.py │ │ │ ├── pipeline_stable_diffusion_latent_upscale.py │ │ │ ├── pipeline_stable_diffusion_upscale.py │ │ │ ├── pipeline_stable_unclip.py │ │ │ ├── pipeline_stable_unclip_img2img.py │ │ │ ├── safety_checker.py │ │ │ ├── safety_checker_flax.py │ │ │ └── stable_unclip_image_normalizer.py │ │ ├── stable_diffusion_attend_and_excite │ │ │ ├── __init__.py │ │ │ └── pipeline_stable_diffusion_attend_and_excite.py │ │ ├── stable_diffusion_diffedit │ │ │ ├── __init__.py │ │ │ └── pipeline_stable_diffusion_diffedit.py │ │ ├── stable_diffusion_gligen │ │ │ ├── __init__.py │ │ │ ├── pipeline_stable_diffusion_gligen.py │ │ │ └── pipeline_stable_diffusion_gligen_text_image.py │ │ ├── stable_diffusion_k_diffusion │ │ │ ├── __init__.py │ │ │ ├── pipeline_stable_diffusion_k_diffusion.py │ │ │ └── pipeline_stable_diffusion_xl_k_diffusion.py │ │ ├── stable_diffusion_ldm3d │ │ │ ├── __init__.py │ │ │ └── pipeline_stable_diffusion_ldm3d.py │ │ ├── stable_diffusion_panorama │ │ │ ├── __init__.py │ │ │ └── pipeline_stable_diffusion_panorama.py │ │ ├── stable_diffusion_safe │ │ │ ├── __init__.py │ │ │ ├── pipeline_output.py │ │ │ ├── pipeline_stable_diffusion_safe.py │ │ │ └── safety_checker.py │ │ ├── stable_diffusion_sag │ │ │ ├── __init__.py │ │ │ └── pipeline_stable_diffusion_sag.py │ │ ├── stable_diffusion_xl │ │ │ ├── __init__.py │ │ │ ├── pipeline_flax_stable_diffusion_xl.py │ │ │ ├── pipeline_output.py │ │ │ ├── pipeline_stable_diffusion_xl.py │ │ │ ├── pipeline_stable_diffusion_xl_img2img.py │ │ │ ├── pipeline_stable_diffusion_xl_inpaint.py │ │ │ ├── pipeline_stable_diffusion_xl_instruct_pix2pix.py │ │ │ └── watermark.py │ │ ├── stable_video_diffusion │ │ │ ├── __init__.py │ │ │ └── pipeline_stable_video_diffusion.py │ │ ├── t2i_adapter │ │ │ ├── __init__.py │ │ │ ├── pipeline_stable_diffusion_adapter.py │ │ │ └── pipeline_stable_diffusion_xl_adapter.py │ │ ├── text_to_video_synthesis │ │ │ ├── __init__.py │ │ │ ├── pipeline_output.py │ │ │ ├── pipeline_text_to_video_synth.py │ │ │ ├── pipeline_text_to_video_synth_img2img.py │ │ │ ├── pipeline_text_to_video_zero.py │ │ │ └── pipeline_text_to_video_zero_sdxl.py │ │ ├── unclip │ │ │ ├── __init__.py │ │ │ ├── pipeline_unclip.py │ │ │ ├── pipeline_unclip_image_variation.py │ │ │ └── text_proj.py │ │ ├── unidiffuser │ │ │ ├── __init__.py │ │ │ ├── modeling_text_decoder.py │ │ │ ├── modeling_uvit.py │ │ │ └── pipeline_unidiffuser.py │ │ └── wuerstchen │ │ │ ├── __init__.py │ │ │ ├── modeling_paella_vq_model.py │ │ │ ├── modeling_wuerstchen_common.py │ │ │ ├── modeling_wuerstchen_diffnext.py │ │ │ ├── modeling_wuerstchen_prior.py │ │ │ ├── pipeline_wuerstchen.py │ │ │ ├── pipeline_wuerstchen_combined.py │ │ │ └── pipeline_wuerstchen_prior.py │ ├── py.typed │ ├── schedulers │ │ ├── README.md │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-310.pyc │ │ │ ├── __init__.cpython-38.pyc │ │ │ ├── scheduling_consistency_decoder.cpython-310.pyc │ │ │ ├── scheduling_consistency_decoder.cpython-38.pyc │ │ │ ├── scheduling_ddim.cpython-310.pyc │ │ │ ├── scheduling_ddim.cpython-38.pyc │ │ │ ├── scheduling_ddpm.cpython-310.pyc │ │ │ ├── scheduling_ddpm.cpython-38.pyc │ │ │ ├── scheduling_deis_multistep.cpython-310.pyc │ │ │ ├── scheduling_deis_multistep.cpython-38.pyc │ │ │ ├── scheduling_dpmsolver_multistep.cpython-310.pyc │ │ │ ├── scheduling_dpmsolver_multistep.cpython-38.pyc │ │ │ ├── scheduling_dpmsolver_sde.cpython-310.pyc │ │ │ ├── scheduling_dpmsolver_sde.cpython-38.pyc │ │ │ ├── scheduling_dpmsolver_singlestep.cpython-310.pyc │ │ │ ├── scheduling_dpmsolver_singlestep.cpython-38.pyc │ │ │ ├── scheduling_edm_dpmsolver_multistep.cpython-310.pyc │ │ │ ├── scheduling_edm_dpmsolver_multistep.cpython-38.pyc │ │ │ ├── scheduling_edm_euler.cpython-310.pyc │ │ │ ├── scheduling_edm_euler.cpython-38.pyc │ │ │ ├── scheduling_euler_ancestral_discrete.cpython-310.pyc │ │ │ ├── scheduling_euler_ancestral_discrete.cpython-38.pyc │ │ │ ├── scheduling_euler_discrete.cpython-310.pyc │ │ │ ├── scheduling_euler_discrete.cpython-38.pyc │ │ │ ├── scheduling_heun_discrete.cpython-310.pyc │ │ │ ├── scheduling_heun_discrete.cpython-38.pyc │ │ │ ├── scheduling_k_dpm_2_ancestral_discrete.cpython-310.pyc │ │ │ ├── scheduling_k_dpm_2_ancestral_discrete.cpython-38.pyc │ │ │ ├── scheduling_k_dpm_2_discrete.cpython-310.pyc │ │ │ ├── scheduling_k_dpm_2_discrete.cpython-38.pyc │ │ │ ├── scheduling_lms_discrete.cpython-310.pyc │ │ │ ├── scheduling_lms_discrete.cpython-38.pyc │ │ │ ├── scheduling_pndm.cpython-310.pyc │ │ │ ├── scheduling_pndm.cpython-38.pyc │ │ │ ├── scheduling_unipc_multistep.cpython-310.pyc │ │ │ ├── scheduling_unipc_multistep.cpython-38.pyc │ │ │ ├── scheduling_utils.cpython-310.pyc │ │ │ └── scheduling_utils.cpython-38.pyc │ │ ├── deprecated │ │ │ ├── __init__.py │ │ │ ├── scheduling_karras_ve.py │ │ │ └── scheduling_sde_vp.py │ │ ├── scheduling_amused.py │ │ ├── scheduling_consistency_decoder.py │ │ ├── scheduling_consistency_models.py │ │ ├── scheduling_ddim.py │ │ ├── scheduling_ddim_flax.py │ │ ├── scheduling_ddim_inverse.py │ │ ├── scheduling_ddim_parallel.py │ │ ├── scheduling_ddpm.py │ │ ├── scheduling_ddpm_flax.py │ │ ├── scheduling_ddpm_parallel.py │ │ ├── scheduling_ddpm_wuerstchen.py │ │ ├── scheduling_deis_multistep.py │ │ ├── scheduling_dpmsolver_multistep.py │ │ ├── scheduling_dpmsolver_multistep_flax.py │ │ ├── scheduling_dpmsolver_multistep_inverse.py │ │ ├── scheduling_dpmsolver_sde.py │ │ ├── scheduling_dpmsolver_singlestep.py │ │ ├── scheduling_edm_dpmsolver_multistep.py │ │ ├── scheduling_edm_euler.py │ │ ├── scheduling_euler_ancestral_discrete.py │ │ ├── scheduling_euler_discrete.py │ │ ├── scheduling_euler_discrete_flax.py │ │ ├── scheduling_heun_discrete.py │ │ ├── scheduling_ipndm.py │ │ ├── scheduling_k_dpm_2_ancestral_discrete.py │ │ ├── scheduling_k_dpm_2_discrete.py │ │ ├── scheduling_karras_ve_flax.py │ │ ├── scheduling_lcm.py │ │ ├── scheduling_lms_discrete.py │ │ ├── scheduling_lms_discrete_flax.py │ │ ├── scheduling_pndm.py │ │ ├── scheduling_pndm_flax.py │ │ ├── scheduling_repaint.py │ │ ├── scheduling_sasolver.py │ │ ├── scheduling_sde_ve.py │ │ ├── scheduling_sde_ve_flax.py │ │ ├── scheduling_tcd.py │ │ ├── scheduling_unclip.py │ │ ├── scheduling_unipc_multistep.py │ │ ├── scheduling_utils.py │ │ ├── scheduling_utils_flax.py │ │ └── scheduling_vq_diffusion.py │ ├── training_utils.py │ ├── utils │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-310.pyc │ │ │ ├── __init__.cpython-38.pyc │ │ │ ├── accelerate_utils.cpython-310.pyc │ │ │ ├── accelerate_utils.cpython-38.pyc │ │ │ ├── constants.cpython-310.pyc │ │ │ ├── constants.cpython-38.pyc │ │ │ ├── deprecation_utils.cpython-310.pyc │ │ │ ├── deprecation_utils.cpython-38.pyc │ │ │ ├── doc_utils.cpython-310.pyc │ │ │ ├── doc_utils.cpython-38.pyc │ │ │ ├── dummy_flax_and_transformers_objects.cpython-310.pyc │ │ │ ├── dummy_flax_and_transformers_objects.cpython-38.pyc │ │ │ ├── dummy_flax_objects.cpython-310.pyc │ │ │ ├── dummy_flax_objects.cpython-38.pyc │ │ │ ├── dummy_note_seq_objects.cpython-310.pyc │ │ │ ├── dummy_note_seq_objects.cpython-38.pyc │ │ │ ├── dummy_onnx_objects.cpython-310.pyc │ │ │ ├── dummy_onnx_objects.cpython-38.pyc │ │ │ ├── dummy_torch_and_librosa_objects.cpython-310.pyc │ │ │ ├── dummy_torch_and_librosa_objects.cpython-38.pyc │ │ │ ├── dummy_torch_and_scipy_objects.cpython-38.pyc │ │ │ ├── dummy_torch_and_torchsde_objects.cpython-310.pyc │ │ │ ├── dummy_torch_and_torchsde_objects.cpython-38.pyc │ │ │ ├── dummy_torch_and_transformers_and_k_diffusion_objects.cpython-310.pyc │ │ │ ├── dummy_torch_and_transformers_and_k_diffusion_objects.cpython-38.pyc │ │ │ ├── dummy_torch_and_transformers_and_onnx_objects.cpython-310.pyc │ │ │ ├── dummy_torch_and_transformers_and_onnx_objects.cpython-38.pyc │ │ │ ├── dummy_transformers_and_torch_and_note_seq_objects.cpython-310.pyc │ │ │ ├── dummy_transformers_and_torch_and_note_seq_objects.cpython-38.pyc │ │ │ ├── dynamic_modules_utils.cpython-310.pyc │ │ │ ├── dynamic_modules_utils.cpython-38.pyc │ │ │ ├── export_utils.cpython-310.pyc │ │ │ ├── export_utils.cpython-38.pyc │ │ │ ├── hub_utils.cpython-310.pyc │ │ │ ├── hub_utils.cpython-38.pyc │ │ │ ├── import_utils.cpython-310.pyc │ │ │ ├── import_utils.cpython-38.pyc │ │ │ ├── loading_utils.cpython-310.pyc │ │ │ ├── loading_utils.cpython-38.pyc │ │ │ ├── logging.cpython-310.pyc │ │ │ ├── logging.cpython-38.pyc │ │ │ ├── outputs.cpython-310.pyc │ │ │ ├── outputs.cpython-38.pyc │ │ │ ├── peft_utils.cpython-310.pyc │ │ │ ├── peft_utils.cpython-38.pyc │ │ │ ├── pil_utils.cpython-310.pyc │ │ │ ├── pil_utils.cpython-38.pyc │ │ │ ├── state_dict_utils.cpython-310.pyc │ │ │ ├── state_dict_utils.cpython-38.pyc │ │ │ ├── torch_utils.cpython-310.pyc │ │ │ ├── torch_utils.cpython-38.pyc │ │ │ ├── versions.cpython-310.pyc │ │ │ └── versions.cpython-38.pyc │ │ ├── accelerate_utils.py │ │ ├── constants.py │ │ ├── deprecation_utils.py │ │ ├── doc_utils.py │ │ ├── dummy_flax_and_transformers_objects.py │ │ ├── dummy_flax_objects.py │ │ ├── dummy_note_seq_objects.py │ │ ├── dummy_onnx_objects.py │ │ ├── dummy_pt_objects.py │ │ ├── dummy_torch_and_librosa_objects.py │ │ ├── dummy_torch_and_scipy_objects.py │ │ ├── dummy_torch_and_torchsde_objects.py │ │ ├── dummy_torch_and_transformers_and_k_diffusion_objects.py │ │ ├── dummy_torch_and_transformers_and_onnx_objects.py │ │ ├── dummy_torch_and_transformers_objects.py │ │ ├── dummy_transformers_and_torch_and_note_seq_objects.py │ │ ├── dynamic_modules_utils.py │ │ ├── export_utils.py │ │ ├── hub_utils.py │ │ ├── import_utils.py │ │ ├── loading_utils.py │ │ ├── logging.py │ │ ├── model_card_template.md │ │ ├── outputs.py │ │ ├── peft_utils.py │ │ ├── pil_utils.py │ │ ├── state_dict_utils.py │ │ ├── testing_utils.py │ │ ├── torch_utils.py │ │ └── versions.py │ └── video_processor.py └── models │ ├── attention.py │ ├── attention_processor.py │ ├── dino_model.py │ ├── dinov2 │ ├── .gitignore │ ├── CODE_OF_CONDUCT.md │ ├── CONTRIBUTING.md │ ├── LICENSE │ ├── MODEL_CARD.md │ ├── README.md │ ├── conda-extras.yaml │ ├── conda.yaml │ ├── dinov2 │ │ ├── __init__.py │ │ ├── configs │ │ │ ├── __init__.py │ │ │ ├── eval │ │ │ │ ├── vitb14_pretrain.yaml │ │ │ │ ├── vitb14_reg4_pretrain.yaml │ │ │ │ ├── vitg14_pretrain.yaml │ │ │ │ ├── vitg14_reg4_pretrain.yaml │ │ │ │ ├── vitl14_pretrain.yaml │ │ │ │ ├── vitl14_reg4_pretrain.yaml │ │ │ │ ├── vits14_pretrain.yaml │ │ │ │ └── vits14_reg4_pretrain.yaml │ │ │ ├── ssl_default_config.yaml │ │ │ └── train │ │ │ │ ├── vitg14.yaml │ │ │ │ ├── vitl14.yaml │ │ │ │ └── vitl16_short.yaml │ │ ├── data │ │ │ ├── __init__.py │ │ │ ├── adapters.py │ │ │ ├── augmentations.py │ │ │ ├── collate.py │ │ │ ├── datasets │ │ │ │ ├── __init__.py │ │ │ │ ├── decoders.py │ │ │ │ ├── extended.py │ │ │ │ ├── image_net.py │ │ │ │ └── image_net_22k.py │ │ │ ├── loaders.py │ │ │ ├── masking.py │ │ │ ├── samplers.py │ │ │ └── transforms.py │ │ ├── distributed │ │ │ └── __init__.py │ │ ├── eval │ │ │ ├── __init__.py │ │ │ ├── depth │ │ │ │ ├── __init__.py │ │ │ │ ├── models │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── backbones │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── vision_transformer.py │ │ │ │ │ ├── builder.py │ │ │ │ │ ├── decode_heads │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── decode_head.py │ │ │ │ │ │ ├── dpt_head.py │ │ │ │ │ │ └── linear_head.py │ │ │ │ │ ├── depther │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── base.py │ │ │ │ │ │ └── encoder_decoder.py │ │ │ │ │ └── losses │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── gradientloss.py │ │ │ │ │ │ └── sigloss.py │ │ │ │ └── ops │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── wrappers.py │ │ │ ├── knn.py │ │ │ ├── linear.py │ │ │ ├── log_regression.py │ │ │ ├── metrics.py │ │ │ ├── segmentation │ │ │ │ ├── __init__.py │ │ │ │ ├── hooks │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── optimizer.py │ │ │ │ ├── models │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── backbones │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── vision_transformer.py │ │ │ │ │ └── decode_heads │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── linear_head.py │ │ │ │ └── utils │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── colormaps.py │ │ │ ├── segmentation_m2f │ │ │ │ ├── __init__.py │ │ │ │ ├── core │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── anchor │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── builder.py │ │ │ │ │ │ └── point_generator.py │ │ │ │ │ ├── box │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── builder.py │ │ │ │ │ │ └── samplers │ │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ │ ├── base_sampler.py │ │ │ │ │ │ │ ├── mask_pseudo_sampler.py │ │ │ │ │ │ │ ├── mask_sampling_result.py │ │ │ │ │ │ │ └── sampling_result.py │ │ │ │ │ └── utils │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── dist_utils.py │ │ │ │ │ │ └── misc.py │ │ │ │ ├── models │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── backbones │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── adapter_modules.py │ │ │ │ │ │ ├── drop_path.py │ │ │ │ │ │ ├── vit.py │ │ │ │ │ │ └── vit_adapter.py │ │ │ │ │ ├── builder.py │ │ │ │ │ ├── decode_heads │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── mask2former_head.py │ │ │ │ │ ├── losses │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── cross_entropy_loss.py │ │ │ │ │ │ ├── dice_loss.py │ │ │ │ │ │ └── match_costs.py │ │ │ │ │ ├── plugins │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── msdeformattn_pixel_decoder.py │ │ │ │ │ ├── segmentors │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ └── encoder_decoder_mask2former.py │ │ │ │ │ └── utils │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── assigner.py │ │ │ │ │ │ ├── point_sample.py │ │ │ │ │ │ ├── positional_encoding.py │ │ │ │ │ │ └── transformer.py │ │ │ │ └── ops │ │ │ │ │ └── modules │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── ms_deform_attn.py │ │ │ ├── setup.py │ │ │ └── utils.py │ │ ├── fsdp │ │ │ └── __init__.py │ │ ├── hub │ │ │ ├── __init__.py │ │ │ ├── backbones.py │ │ │ ├── classifiers.py │ │ │ ├── depth │ │ │ │ ├── __init__.py │ │ │ │ ├── decode_heads.py │ │ │ │ ├── encoder_decoder.py │ │ │ │ └── ops.py │ │ │ ├── depthers.py │ │ │ └── utils.py │ │ ├── layers │ │ │ ├── __init__.py │ │ │ ├── attention.py │ │ │ ├── block.py │ │ │ ├── dino_head.py │ │ │ ├── drop_path.py │ │ │ ├── layer_scale.py │ │ │ ├── mlp.py │ │ │ ├── patch_embed.py │ │ │ └── swiglu_ffn.py │ │ ├── logging │ │ │ ├── __init__.py │ │ │ └── helpers.py │ │ ├── loss │ │ │ ├── __init__.py │ │ │ ├── dino_clstoken_loss.py │ │ │ ├── ibot_patch_loss.py │ │ │ └── koleo_loss.py │ │ ├── models │ │ │ ├── __init__.py │ │ │ └── vision_transformer.py │ │ ├── run │ │ │ ├── __init__.py │ │ │ ├── eval │ │ │ │ ├── knn.py │ │ │ │ ├── linear.py │ │ │ │ └── log_regression.py │ │ │ ├── submit.py │ │ │ └── train │ │ │ │ └── train.py │ │ ├── train │ │ │ ├── __init__.py │ │ │ ├── ssl_meta_arch.py │ │ │ └── train.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── cluster.py │ │ │ ├── config.py │ │ │ ├── dtype.py │ │ │ ├── param_groups.py │ │ │ └── utils.py │ ├── hubconf.py │ ├── pyproject.toml │ ├── requirements-dev.txt │ ├── requirements-extras.txt │ ├── requirements.txt │ ├── scripts │ │ └── lint.sh │ ├── setup.cfg │ └── setup.py │ ├── mutual_self_attention.py │ ├── mutual_self_attention_multi_scale.py │ ├── projection.py │ ├── refunet_2d_condition.py │ ├── transformer_2d.py │ ├── unet_2d_blocks.py │ └── unet_2d_condition.py ├── training ├── .DS_Store ├── dataset_configuration.py └── setting_config.py └── utils ├── data_util.py ├── diffusion_util.py ├── image_util.py └── seed_all.py /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/.DS_Store -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/README.md -------------------------------------------------------------------------------- /asset/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/asset/.DS_Store -------------------------------------------------------------------------------- /asset/UI.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/asset/UI.jpg -------------------------------------------------------------------------------- /asset/badge-website.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/asset/badge-website.svg -------------------------------------------------------------------------------- /asset/intro.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/asset/intro.jpg -------------------------------------------------------------------------------- /asset/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/asset/logo.png -------------------------------------------------------------------------------- /asset/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/asset/teaser.jpg -------------------------------------------------------------------------------- /automatic_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/automatic_label.py -------------------------------------------------------------------------------- /ckpt/controlnet/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/ckpt/controlnet/config.json -------------------------------------------------------------------------------- /ckpt/download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/ckpt/download.sh -------------------------------------------------------------------------------- /data/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/.DS_Store -------------------------------------------------------------------------------- /data/test1/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test1/1.jpg -------------------------------------------------------------------------------- /data/test1/2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test1/2.jpg -------------------------------------------------------------------------------- /data/test1/3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test1/3.jpg -------------------------------------------------------------------------------- /data/test1/masks/mask_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test1/masks/mask_1.png -------------------------------------------------------------------------------- /data/test1/masks/mask_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test1/masks/mask_2.png -------------------------------------------------------------------------------- /data/test1/masks/mask_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test1/masks/mask_3.png -------------------------------------------------------------------------------- /data/test1/sketch.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test1/sketch.jpg -------------------------------------------------------------------------------- /data/test2/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test2/1.jpg -------------------------------------------------------------------------------- /data/test2/2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test2/2.jpg -------------------------------------------------------------------------------- /data/test2/3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test2/3.jpg -------------------------------------------------------------------------------- /data/test2/4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test2/4.jpg -------------------------------------------------------------------------------- /data/test2/5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test2/5.jpg -------------------------------------------------------------------------------- /data/test2/masks/mask_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test2/masks/mask_1.png -------------------------------------------------------------------------------- /data/test2/masks/mask_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test2/masks/mask_2.png -------------------------------------------------------------------------------- /data/test2/masks/mask_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test2/masks/mask_3.png -------------------------------------------------------------------------------- /data/test2/masks/mask_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test2/masks/mask_4.png -------------------------------------------------------------------------------- /data/test2/masks/mask_5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test2/masks/mask_5.png -------------------------------------------------------------------------------- /data/test2/sketch.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/data/test2/sketch.jpg -------------------------------------------------------------------------------- /dataloader/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/dataloader/coco.py -------------------------------------------------------------------------------- /dataloader/file_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/dataloader/file_io.py -------------------------------------------------------------------------------- /dataloader/image_multipair_loader_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/dataloader/image_multipair_loader_add.py -------------------------------------------------------------------------------- /dataloader/image_pair_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/dataloader/image_pair_loader.py -------------------------------------------------------------------------------- /dataloader/inference_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/dataloader/inference_loader.py -------------------------------------------------------------------------------- /dataloader/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/dataloader/transforms.py -------------------------------------------------------------------------------- /dataloader/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/dataloader/utils.py -------------------------------------------------------------------------------- /dataloader/video_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/dataloader/video_loader.py -------------------------------------------------------------------------------- /inference/gradio_app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/inference/gradio_app.py -------------------------------------------------------------------------------- /inference/image_pair_edit_pipeline_multi_ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/inference/image_pair_edit_pipeline_multi_ref.py -------------------------------------------------------------------------------- /inference/multi_ref_infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/inference/multi_ref_infer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/scripts/.DS_Store -------------------------------------------------------------------------------- /scripts/multi_ref_infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/scripts/multi_ref_infer.sh -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/__init__.py -------------------------------------------------------------------------------- /src/annotator/canny/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/canny/__init__.py -------------------------------------------------------------------------------- /src/annotator/hed/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/hed/__init__.py -------------------------------------------------------------------------------- /src/annotator/lineart/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/lineart/LICENSE -------------------------------------------------------------------------------- /src/annotator/lineart/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/lineart/__init__.py -------------------------------------------------------------------------------- /src/annotator/lineart_anime/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/lineart_anime/LICENSE -------------------------------------------------------------------------------- /src/annotator/lineart_anime/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/lineart_anime/__init__.py -------------------------------------------------------------------------------- /src/annotator/midas/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/midas/LICENSE -------------------------------------------------------------------------------- /src/annotator/midas/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/midas/__init__.py -------------------------------------------------------------------------------- /src/annotator/midas/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/midas/api.py -------------------------------------------------------------------------------- /src/annotator/midas/midas/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/annotator/midas/midas/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/midas/midas/base_model.py -------------------------------------------------------------------------------- /src/annotator/midas/midas/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/midas/midas/blocks.py -------------------------------------------------------------------------------- /src/annotator/midas/midas/dpt_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/midas/midas/dpt_depth.py -------------------------------------------------------------------------------- /src/annotator/midas/midas/midas_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/midas/midas/midas_net.py -------------------------------------------------------------------------------- /src/annotator/midas/midas/midas_net_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/midas/midas/midas_net_custom.py -------------------------------------------------------------------------------- /src/annotator/midas/midas/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/midas/midas/transforms.py -------------------------------------------------------------------------------- /src/annotator/midas/midas/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/midas/midas/vit.py -------------------------------------------------------------------------------- /src/annotator/midas/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/midas/utils.py -------------------------------------------------------------------------------- /src/annotator/mlsd/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/mlsd/LICENSE -------------------------------------------------------------------------------- /src/annotator/mlsd/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/mlsd/__init__.py -------------------------------------------------------------------------------- /src/annotator/mlsd/models/mbv2_mlsd_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/mlsd/models/mbv2_mlsd_large.py -------------------------------------------------------------------------------- /src/annotator/mlsd/models/mbv2_mlsd_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/mlsd/models/mbv2_mlsd_tiny.py -------------------------------------------------------------------------------- /src/annotator/mlsd/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/mlsd/utils.py -------------------------------------------------------------------------------- /src/annotator/normalbae/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/normalbae/LICENSE -------------------------------------------------------------------------------- /src/annotator/normalbae/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/normalbae/__init__.py -------------------------------------------------------------------------------- /src/annotator/normalbae/models/NNET.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/normalbae/models/NNET.py -------------------------------------------------------------------------------- /src/annotator/normalbae/models/baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/normalbae/models/baseline.py -------------------------------------------------------------------------------- /src/annotator/normalbae/models/submodules/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/normalbae/models/submodules/decoder.py -------------------------------------------------------------------------------- /src/annotator/normalbae/models/submodules/efficientnet_repo/geffnet/version.py: -------------------------------------------------------------------------------- 1 | __version__ = '1.0.2' 2 | -------------------------------------------------------------------------------- /src/annotator/normalbae/models/submodules/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/normalbae/models/submodules/encoder.py -------------------------------------------------------------------------------- /src/annotator/normalbae/models/submodules/submodules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/normalbae/models/submodules/submodules.py -------------------------------------------------------------------------------- /src/annotator/normalbae/utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/normalbae/utils/losses.py -------------------------------------------------------------------------------- /src/annotator/normalbae/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/normalbae/utils/utils.py -------------------------------------------------------------------------------- /src/annotator/oneformer/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/LICENSE -------------------------------------------------------------------------------- /src/annotator/oneformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/api.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/checkpoint/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/checkpoint/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/checkpoint/catalog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/checkpoint/catalog.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/config/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/config/compat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/config/compat.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/config/config.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/config/defaults.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/config/instantiate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/config/instantiate.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/config/lazy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/config/lazy.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/data/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/data/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/data/benchmark.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/data/build.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/data/catalog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/data/catalog.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/data/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/data/common.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/data/dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/data/dataset_mapper.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/data/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/data/datasets/coco.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/data/datasets/lvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/data/datasets/lvis.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/engine/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/engine/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/engine/defaults.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/engine/hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/engine/hooks.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/engine/launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/engine/launch.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/engine/train_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/engine/train_loop.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/evaluation/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/evaluation/testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/evaluation/testing.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/export/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/export/README.md -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/export/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/export/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/export/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/export/api.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/export/c10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/export/c10.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/export/caffe2_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/export/caffe2_patch.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/export/flatten.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/export/flatten.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/export/shared.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/export/shared.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/export/torchscript.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/export/torchscript.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/layers/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/layers/aspp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/layers/aspp.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/layers/batch_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/layers/batch_norm.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/layers/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/layers/blocks.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/layers/csrc/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/layers/csrc/README.md -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/layers/csrc/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/layers/csrc/vision.cpp -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/layers/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/layers/deform_conv.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/layers/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/layers/losses.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/layers/mask_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/layers/mask_ops.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/layers/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/layers/nms.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/layers/roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/layers/roi_align.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/layers/shape_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/layers/shape_spec.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/layers/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/layers/wrappers.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/model_zoo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/model_zoo/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/model_zoo/model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/model_zoo/model_zoo.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/modeling/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/modeling/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/modeling/matcher.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/modeling/poolers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/modeling/poolers.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/modeling/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/modeling/sampling.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/projects/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/projects/README.md -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/projects/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/projects/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/solver/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/solver/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/solver/build.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/solver/lr_scheduler.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/structures/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/structures/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/structures/boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/structures/boxes.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/structures/masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/structures/masks.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/tracking/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/tracking/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/tracking/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/tracking/utils.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/utils/README.md -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/utils/analysis.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/collect_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/utils/collect_env.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/colormap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/utils/colormap.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/utils/comm.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/develop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/utils/develop.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/utils/env.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/events.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/utils/events.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/file_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/utils/file_io.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/utils/logger.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/utils/memory.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/utils/registry.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/serialize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/utils/serialize.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/utils/testing.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/tracing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/utils/tracing.py -------------------------------------------------------------------------------- /src/annotator/oneformer/detectron2/utils/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/detectron2/utils/visualizer.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/config.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/data/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/data/build.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/data/dataset_mappers/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/data/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/data/tokenizer.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/demo/colormap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/demo/colormap.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/demo/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/demo/defaults.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/demo/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/demo/predictor.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/demo/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/demo/visualizer.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/evaluation/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/evaluation/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/evaluation/evaluator.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/modeling/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/modeling/backbone/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/modeling/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/modeling/matcher.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/modeling/meta_arch/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/modeling/pixel_decoder/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/oneformer_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/oneformer_model.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/utils/__init__.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/utils/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/utils/box_ops.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/utils/events.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/utils/events.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/utils/misc.py -------------------------------------------------------------------------------- /src/annotator/oneformer/oneformer/utils/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/oneformer/oneformer/utils/pos_embed.py -------------------------------------------------------------------------------- /src/annotator/openpose/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/openpose/LICENSE -------------------------------------------------------------------------------- /src/annotator/openpose/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/openpose/__init__.py -------------------------------------------------------------------------------- /src/annotator/openpose/body.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/openpose/body.py -------------------------------------------------------------------------------- /src/annotator/openpose/face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/openpose/face.py -------------------------------------------------------------------------------- /src/annotator/openpose/hand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/openpose/hand.py -------------------------------------------------------------------------------- /src/annotator/openpose/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/openpose/model.py -------------------------------------------------------------------------------- /src/annotator/openpose/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/openpose/util.py -------------------------------------------------------------------------------- /src/annotator/pidinet/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/pidinet/LICENSE -------------------------------------------------------------------------------- /src/annotator/pidinet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/pidinet/__init__.py -------------------------------------------------------------------------------- /src/annotator/pidinet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/pidinet/model.py -------------------------------------------------------------------------------- /src/annotator/shuffle/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/shuffle/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/LICENSE -------------------------------------------------------------------------------- /src/annotator/uniformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/configs/_base_/datasets/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/configs/_base_/datasets/ade20k.py -------------------------------------------------------------------------------- /src/annotator/uniformer/configs/_base_/datasets/drive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/configs/_base_/datasets/drive.py -------------------------------------------------------------------------------- /src/annotator/uniformer/configs/_base_/datasets/hrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/configs/_base_/datasets/hrf.py -------------------------------------------------------------------------------- /src/annotator/uniformer/configs/_base_/datasets/stare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/configs/_base_/datasets/stare.py -------------------------------------------------------------------------------- /src/annotator/uniformer/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /src/annotator/uniformer/configs/_base_/models/cgnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/configs/_base_/models/cgnet.py -------------------------------------------------------------------------------- /src/annotator/uniformer/configs/_base_/models/fcn_hr18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/configs/_base_/models/fcn_hr18.py -------------------------------------------------------------------------------- /src/annotator/uniformer/configs/_base_/models/fpn_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/configs/_base_/models/fpn_r50.py -------------------------------------------------------------------------------- /src/annotator/uniformer/exp/upernet_global_small/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/exp/upernet_global_small/run.sh -------------------------------------------------------------------------------- /src/annotator/uniformer/exp/upernet_global_small/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/exp/upernet_global_small/test.sh -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/arraymisc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/arraymisc/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/arraymisc/quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/arraymisc/quantization.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/alexnet.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/activation.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/context_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/context_block.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/conv.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/conv_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/conv_module.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/conv_ws.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/conv_ws.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/drop.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/hsigmoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/hsigmoid.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/hswish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/hswish.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/non_local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/non_local.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/norm.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/padding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/padding.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/plugin.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/registry.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/scale.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/swish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/swish.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/transformer.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/upsample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/upsample.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/bricks/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/bricks/wrappers.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/builder.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/resnet.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/utils/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/utils/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/utils/flops_counter.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/utils/fuse_conv_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/utils/fuse_conv_bn.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/utils/sync_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/utils/sync_bn.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/utils/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/utils/weight_init.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/cnn/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/cnn/vgg.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/engine/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/engine/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/engine/test.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/fileio/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/fileio/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/fileio/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/fileio/file_client.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/fileio/handlers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/fileio/handlers/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/fileio/handlers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/fileio/handlers/base.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/fileio/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/fileio/io.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/fileio/parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/fileio/parse.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/image/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/image/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/image/colorspace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/image/colorspace.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/image/geometric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/image/geometric.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/image/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/image/io.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/image/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/image/misc.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/image/photometric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/image/photometric.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/model_zoo/deprecated.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/model_zoo/deprecated.json -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/model_zoo/mmcls.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/model_zoo/mmcls.json -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/model_zoo/open_mmlab.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/model_zoo/open_mmlab.json -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/assign_score_withk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/assign_score_withk.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/ball_query.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/ball_query.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/bbox.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/border_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/border_align.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/box_iou_rotated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/box_iou_rotated.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/carafe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/carafe.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/cc_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/cc_attention.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/contour_expand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/contour_expand.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/corner_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/corner_pool.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/correlation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/correlation.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/deform_conv.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/deform_roi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/deform_roi_pool.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/deprecated_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/deprecated_wrappers.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/focal_loss.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/furthest_point_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/furthest_point_sample.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/fused_bias_leakyrelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/fused_bias_leakyrelu.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/gather_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/gather_points.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/group_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/group_points.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/info.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/iou3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/iou3d.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/knn.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/masked_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/masked_conv.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/merge_cells.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/merge_cells.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/modulated_deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/modulated_deform_conv.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/nms.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/pixel_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/pixel_group.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/point_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/point_sample.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/points_in_boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/points_in_boxes.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/points_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/points_sampler.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/psa_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/psa_mask.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/roi_align.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/roi_align_rotated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/roi_align_rotated.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/roi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/roi_pool.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/roiaware_pool3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/roiaware_pool3d.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/roipoint_pool3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/roipoint_pool3d.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/saconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/saconv.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/scatter_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/scatter_points.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/sync_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/sync_bn.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/three_interpolate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/three_interpolate.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/three_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/three_nn.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/tin_shift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/tin_shift.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/upfirdn2d.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/ops/voxelize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/ops/voxelize.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/parallel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/parallel/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/parallel/_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/parallel/_functions.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/parallel/collate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/parallel/collate.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/parallel/data_container.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/parallel/data_container.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/parallel/data_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/parallel/data_parallel.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/parallel/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/parallel/distributed.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/parallel/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/parallel/registry.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/parallel/scatter_gather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/parallel/scatter_gather.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/parallel/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/parallel/utils.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/base_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/base_module.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/base_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/base_runner.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/builder.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/checkpoint.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/dist_utils.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/epoch_based_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/epoch_based_runner.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/fp16_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/fp16_utils.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/checkpoint.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/closure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/closure.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/ema.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/evaluation.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/hook.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/iter_timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/iter_timer.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/logger/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/logger/base.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/logger/pavi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/logger/pavi.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/logger/text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/logger/text.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/logger/wandb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/logger/wandb.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/lr_updater.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/lr_updater.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/memory.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/optimizer.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/profiler.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/sampler_seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/sampler_seed.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/hooks/sync_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/hooks/sync_buffer.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/iter_based_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/iter_based_runner.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/log_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/log_buffer.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/optimizer/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/optimizer/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/optimizer/builder.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/priority.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/priority.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/runner/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/runner/utils.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/utils/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/utils/config.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/utils/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/utils/env.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/utils/ext_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/utils/ext_loader.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/utils/logging.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/utils/misc.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/utils/parrots_jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/utils/parrots_jit.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/utils/parrots_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/utils/parrots_wrapper.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/utils/path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/utils/path.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/utils/progressbar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/utils/progressbar.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/utils/registry.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/utils/testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/utils/testing.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/utils/timer.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/utils/trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/utils/trace.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/utils/version_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/utils/version_utils.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/version.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/video/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/video/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/video/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/video/io.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/video/optflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/video/optflow.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/video/processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/video/processing.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/visualization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/visualization/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/visualization/color.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/visualization/color.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/visualization/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/visualization/image.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv/visualization/optflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv/visualization/optflow.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv_custom/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv_custom/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmcv_custom/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmcv_custom/checkpoint.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/apis/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/apis/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/apis/inference.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/apis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/apis/test.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/apis/train.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/core/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/core/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/core/evaluation/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/core/evaluation/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/core/evaluation/metrics.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/core/seg/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/core/seg/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/core/seg/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/core/seg/builder.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/core/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/core/utils/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/core/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/core/utils/misc.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/datasets/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/datasets/ade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/datasets/ade.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/datasets/builder.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/datasets/chase_db1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/datasets/chase_db1.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/datasets/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/datasets/cityscapes.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/datasets/custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/datasets/custom.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/datasets/drive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/datasets/drive.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/datasets/hrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/datasets/hrf.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/datasets/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/datasets/pascal_context.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/datasets/stare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/datasets/stare.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/datasets/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/datasets/voc.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/backbones/cgnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/backbones/cgnet.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/backbones/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/backbones/hrnet.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/backbones/resnest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/backbones/resnest.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/backbones/resnet.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/backbones/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/backbones/resnext.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/backbones/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/backbones/unet.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/backbones/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/backbones/vit.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/builder.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/losses/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/losses/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/losses/accuracy.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/losses/dice_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/losses/dice_loss.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/losses/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/losses/utils.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/necks/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/necks/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/necks/fpn.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/segmentors/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/segmentors/base.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/utils/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/utils/drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/utils/drop.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/utils/res_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/utils/res_layer.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/utils/se_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/utils/se_layer.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/models/utils/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/models/utils/weight_init.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/ops/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/ops/encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/ops/encoding.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/ops/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/ops/wrappers.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/utils/__init__.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/utils/collect_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/utils/collect_env.py -------------------------------------------------------------------------------- /src/annotator/uniformer/mmseg/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/uniformer/mmseg/utils/logger.py -------------------------------------------------------------------------------- /src/annotator/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/util.py -------------------------------------------------------------------------------- /src/annotator/zoe/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/LICENSE -------------------------------------------------------------------------------- /src/annotator/zoe/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/__init__.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/data/__init__.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/data/data_mono.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/data/data_mono.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/data/ddad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/data/ddad.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/data/diml_indoor_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/data/diml_indoor_test.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/data/diml_outdoor_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/data/diml_outdoor_test.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/data/diode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/data/diode.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/data/hypersim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/data/hypersim.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/data/ibims.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/data/ibims.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/data/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/data/preprocess.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/data/sun_rgbd_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/data/sun_rgbd_loader.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/data/transforms.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/data/vkitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/data/vkitti.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/data/vkitti2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/data/vkitti2.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/models/__init__.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/models/base_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/models/base_models/__init__.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/models/base_models/midas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/models/base_models/midas.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/models/base_models/midas_repo/input/.placeholder: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/models/base_models/midas_repo/mobile/ios/.gitignore: -------------------------------------------------------------------------------- 1 | # ignore model file 2 | #*.tflite 3 | -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/models/base_models/midas_repo/output/.placeholder: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/models/base_models/midas_repo/tf/input/.placeholder: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/models/base_models/midas_repo/tf/output/.placeholder: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/models/base_models/midas_repo/weights/.placeholder: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/models/builder.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/models/depth_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/models/depth_model.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/models/layers/attractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/models/layers/attractor.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/models/model_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/models/model_io.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/models/zoedepth/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/models/zoedepth/__init__.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/trainers/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/trainers/base_trainer.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/trainers/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/trainers/builder.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/trainers/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/trainers/loss.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/utils/__init__.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/utils/arg_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/utils/arg_utils.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/utils/config.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/utils/easydict/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/utils/easydict/__init__.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/utils/geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/utils/geometry.py -------------------------------------------------------------------------------- /src/annotator/zoe/zoedepth/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/annotator/zoe/zoedepth/utils/misc.py -------------------------------------------------------------------------------- /src/diffusers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/__init__.py -------------------------------------------------------------------------------- /src/diffusers/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/callbacks.py -------------------------------------------------------------------------------- /src/diffusers/commands/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/commands/__init__.py -------------------------------------------------------------------------------- /src/diffusers/commands/diffusers_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/commands/diffusers_cli.py -------------------------------------------------------------------------------- /src/diffusers/commands/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/commands/env.py -------------------------------------------------------------------------------- /src/diffusers/commands/fp16_safetensors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/commands/fp16_safetensors.py -------------------------------------------------------------------------------- /src/diffusers/configuration_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/configuration_utils.py -------------------------------------------------------------------------------- /src/diffusers/dependency_versions_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/dependency_versions_check.py -------------------------------------------------------------------------------- /src/diffusers/dependency_versions_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/dependency_versions_table.py -------------------------------------------------------------------------------- /src/diffusers/experimental/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/experimental/README.md -------------------------------------------------------------------------------- /src/diffusers/experimental/__init__.py: -------------------------------------------------------------------------------- 1 | from .rl import ValueGuidedRLPipeline 2 | -------------------------------------------------------------------------------- /src/diffusers/experimental/rl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/experimental/rl/__init__.py -------------------------------------------------------------------------------- /src/diffusers/experimental/rl/value_guided_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/experimental/rl/value_guided_sampling.py -------------------------------------------------------------------------------- /src/diffusers/image_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/image_processor.py -------------------------------------------------------------------------------- /src/diffusers/loaders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/__init__.py -------------------------------------------------------------------------------- /src/diffusers/loaders/__pycache__/peft.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/__pycache__/peft.cpython-310.pyc -------------------------------------------------------------------------------- /src/diffusers/loaders/__pycache__/peft.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/__pycache__/peft.cpython-38.pyc -------------------------------------------------------------------------------- /src/diffusers/loaders/__pycache__/unet.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/__pycache__/unet.cpython-310.pyc -------------------------------------------------------------------------------- /src/diffusers/loaders/__pycache__/unet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/__pycache__/unet.cpython-38.pyc -------------------------------------------------------------------------------- /src/diffusers/loaders/__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /src/diffusers/loaders/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/autoencoder.py -------------------------------------------------------------------------------- /src/diffusers/loaders/controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/controlnet.py -------------------------------------------------------------------------------- /src/diffusers/loaders/ip_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/ip_adapter.py -------------------------------------------------------------------------------- /src/diffusers/loaders/lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/lora.py -------------------------------------------------------------------------------- /src/diffusers/loaders/lora_conversion_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/lora_conversion_utils.py -------------------------------------------------------------------------------- /src/diffusers/loaders/peft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/peft.py -------------------------------------------------------------------------------- /src/diffusers/loaders/single_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/single_file.py -------------------------------------------------------------------------------- /src/diffusers/loaders/single_file_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/single_file_model.py -------------------------------------------------------------------------------- /src/diffusers/loaders/single_file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/single_file_utils.py -------------------------------------------------------------------------------- /src/diffusers/loaders/textual_inversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/textual_inversion.py -------------------------------------------------------------------------------- /src/diffusers/loaders/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/unet.py -------------------------------------------------------------------------------- /src/diffusers/loaders/unet_loader_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/unet_loader_utils.py -------------------------------------------------------------------------------- /src/diffusers/loaders/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/loaders/utils.py -------------------------------------------------------------------------------- /src/diffusers/models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/README.md -------------------------------------------------------------------------------- /src/diffusers/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/__init__.py -------------------------------------------------------------------------------- /src/diffusers/models/__pycache__/lora.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/__pycache__/lora.cpython-310.pyc -------------------------------------------------------------------------------- /src/diffusers/models/__pycache__/lora.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/__pycache__/lora.cpython-38.pyc -------------------------------------------------------------------------------- /src/diffusers/models/__pycache__/resnet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/__pycache__/resnet.cpython-38.pyc -------------------------------------------------------------------------------- /src/diffusers/models/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/activations.py -------------------------------------------------------------------------------- /src/diffusers/models/adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/adapter.py -------------------------------------------------------------------------------- /src/diffusers/models/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/attention.py -------------------------------------------------------------------------------- /src/diffusers/models/attention_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/attention_flax.py -------------------------------------------------------------------------------- /src/diffusers/models/attention_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/attention_processor.py -------------------------------------------------------------------------------- /src/diffusers/models/autoencoders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/autoencoders/__init__.py -------------------------------------------------------------------------------- /src/diffusers/models/autoencoders/autoencoder_kl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/autoencoders/autoencoder_kl.py -------------------------------------------------------------------------------- /src/diffusers/models/autoencoders/autoencoder_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/autoencoders/autoencoder_tiny.py -------------------------------------------------------------------------------- /src/diffusers/models/autoencoders/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/autoencoders/vae.py -------------------------------------------------------------------------------- /src/diffusers/models/controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/controlnet.py -------------------------------------------------------------------------------- /src/diffusers/models/controlnet1x1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/controlnet1x1.py -------------------------------------------------------------------------------- /src/diffusers/models/controlnet_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/controlnet_flax.py -------------------------------------------------------------------------------- /src/diffusers/models/controlnet_sin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/controlnet_sin.py -------------------------------------------------------------------------------- /src/diffusers/models/controlnet_xs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/controlnet_xs.py -------------------------------------------------------------------------------- /src/diffusers/models/downsampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/downsampling.py -------------------------------------------------------------------------------- /src/diffusers/models/dual_transformer_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/dual_transformer_2d.py -------------------------------------------------------------------------------- /src/diffusers/models/embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/embeddings.py -------------------------------------------------------------------------------- /src/diffusers/models/embeddings_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/embeddings_flax.py -------------------------------------------------------------------------------- /src/diffusers/models/lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/lora.py -------------------------------------------------------------------------------- /src/diffusers/models/model_loading_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/model_loading_utils.py -------------------------------------------------------------------------------- /src/diffusers/models/modeling_flax_pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/modeling_flax_pytorch_utils.py -------------------------------------------------------------------------------- /src/diffusers/models/modeling_flax_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/modeling_flax_utils.py -------------------------------------------------------------------------------- /src/diffusers/models/modeling_outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/modeling_outputs.py -------------------------------------------------------------------------------- /src/diffusers/models/modeling_pytorch_flax_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/modeling_pytorch_flax_utils.py -------------------------------------------------------------------------------- /src/diffusers/models/modeling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/modeling_utils.py -------------------------------------------------------------------------------- /src/diffusers/models/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/normalization.py -------------------------------------------------------------------------------- /src/diffusers/models/prior_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/prior_transformer.py -------------------------------------------------------------------------------- /src/diffusers/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/resnet.py -------------------------------------------------------------------------------- /src/diffusers/models/resnet_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/resnet_flax.py -------------------------------------------------------------------------------- /src/diffusers/models/t5_film_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/t5_film_transformer.py -------------------------------------------------------------------------------- /src/diffusers/models/transformer_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/transformer_2d.py -------------------------------------------------------------------------------- /src/diffusers/models/transformer_temporal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/transformer_temporal.py -------------------------------------------------------------------------------- /src/diffusers/models/transformers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/transformers/__init__.py -------------------------------------------------------------------------------- /src/diffusers/models/transformers/prior_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/transformers/prior_transformer.py -------------------------------------------------------------------------------- /src/diffusers/models/transformers/transformer_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/transformers/transformer_2d.py -------------------------------------------------------------------------------- /src/diffusers/models/unet_1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unet_1d.py -------------------------------------------------------------------------------- /src/diffusers/models/unet_1d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unet_1d_blocks.py -------------------------------------------------------------------------------- /src/diffusers/models/unet_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unet_2d.py -------------------------------------------------------------------------------- /src/diffusers/models/unet_2d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unet_2d_blocks.py -------------------------------------------------------------------------------- /src/diffusers/models/unet_2d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unet_2d_condition.py -------------------------------------------------------------------------------- /src/diffusers/models/unets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unets/__init__.py -------------------------------------------------------------------------------- /src/diffusers/models/unets/unet_1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unets/unet_1d.py -------------------------------------------------------------------------------- /src/diffusers/models/unets/unet_1d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unets/unet_1d_blocks.py -------------------------------------------------------------------------------- /src/diffusers/models/unets/unet_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unets/unet_2d.py -------------------------------------------------------------------------------- /src/diffusers/models/unets/unet_2d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unets/unet_2d_blocks.py -------------------------------------------------------------------------------- /src/diffusers/models/unets/unet_2d_blocks_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unets/unet_2d_blocks_flax.py -------------------------------------------------------------------------------- /src/diffusers/models/unets/unet_2d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unets/unet_2d_condition.py -------------------------------------------------------------------------------- /src/diffusers/models/unets/unet_2d_condition_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unets/unet_2d_condition_flax.py -------------------------------------------------------------------------------- /src/diffusers/models/unets/unet_3d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unets/unet_3d_blocks.py -------------------------------------------------------------------------------- /src/diffusers/models/unets/unet_3d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unets/unet_3d_condition.py -------------------------------------------------------------------------------- /src/diffusers/models/unets/unet_i2vgen_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unets/unet_i2vgen_xl.py -------------------------------------------------------------------------------- /src/diffusers/models/unets/unet_kandinsky3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unets/unet_kandinsky3.py -------------------------------------------------------------------------------- /src/diffusers/models/unets/unet_motion_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unets/unet_motion_model.py -------------------------------------------------------------------------------- /src/diffusers/models/unets/unet_stable_cascade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unets/unet_stable_cascade.py -------------------------------------------------------------------------------- /src/diffusers/models/unets/uvit_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/unets/uvit_2d.py -------------------------------------------------------------------------------- /src/diffusers/models/upsampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/upsampling.py -------------------------------------------------------------------------------- /src/diffusers/models/vae_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/vae_flax.py -------------------------------------------------------------------------------- /src/diffusers/models/vq_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/models/vq_model.py -------------------------------------------------------------------------------- /src/diffusers/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/optimization.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/README.md -------------------------------------------------------------------------------- /src/diffusers/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/amused/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/amused/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/amused/pipeline_amused.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/amused/pipeline_amused.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/animatediff/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/animatediff/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/animatediff/pipeline_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/animatediff/pipeline_output.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/audioldm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/audioldm/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/audioldm/pipeline_audioldm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/audioldm/pipeline_audioldm.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/audioldm2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/audioldm2/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/auto_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/auto_pipeline.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/blip_diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/blip_diffusion/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/consistency_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/consistency_models/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/controlnet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/controlnet/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/controlnet/multicontrolnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/controlnet/multicontrolnet.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/controlnet_xs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/controlnet_xs/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/dance_diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/dance_diffusion/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/ddim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/ddim/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/ddim/pipeline_ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/ddim/pipeline_ddim.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/ddpm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/ddpm/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/ddpm/pipeline_ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/ddpm/pipeline_ddpm.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/deepfloyd_if/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/deepfloyd_if/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/deepfloyd_if/pipeline_if.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/deepfloyd_if/pipeline_if.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/deepfloyd_if/safety_checker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/deepfloyd_if/safety_checker.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/deepfloyd_if/timesteps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/deepfloyd_if/timesteps.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/deepfloyd_if/watermark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/deepfloyd_if/watermark.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/deprecated/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/deprecated/README.md -------------------------------------------------------------------------------- /src/diffusers/pipelines/deprecated/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/deprecated/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/deprecated/pndm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/deprecated/pndm/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/deprecated/repaint/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/deprecated/repaint/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/dit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/dit/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/dit/pipeline_dit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/dit/pipeline_dit.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/free_init_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/free_init_utils.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/i2vgen_xl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/i2vgen_xl/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/kandinsky/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/kandinsky/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/kandinsky/text_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/kandinsky/text_encoder.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/kandinsky2_2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/kandinsky2_2/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/kandinsky3/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/kandinsky3/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/latent_diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/latent_diffusion/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/ledits_pp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/ledits_pp/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/ledits_pp/pipeline_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/ledits_pp/pipeline_output.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/marigold/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/marigold/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/musicldm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/musicldm/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/musicldm/pipeline_musicldm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/musicldm/pipeline_musicldm.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/onnx_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/onnx_utils.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/paint_by_example/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/paint_by_example/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/pia/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/pia/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/pia/pipeline_pia.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/pia/pipeline_pia.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/pipeline_flax_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/pipeline_flax_utils.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/pipeline_loading_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/pipeline_loading_utils.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/pipeline_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/pipeline_utils.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/pixart_alpha/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/pixart_alpha/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/shap_e/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/shap_e/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/shap_e/camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/shap_e/camera.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/shap_e/pipeline_shap_e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/shap_e/pipeline_shap_e.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/shap_e/renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/shap_e/renderer.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/stable_cascade/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/stable_cascade/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/stable_diffusion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/stable_diffusion/README.md -------------------------------------------------------------------------------- /src/diffusers/pipelines/stable_diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/stable_diffusion/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/t2i_adapter/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/t2i_adapter/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/unclip/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/unclip/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/unclip/pipeline_unclip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/unclip/pipeline_unclip.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/unclip/text_proj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/unclip/text_proj.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/unidiffuser/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/unidiffuser/__init__.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/unidiffuser/modeling_uvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/unidiffuser/modeling_uvit.py -------------------------------------------------------------------------------- /src/diffusers/pipelines/wuerstchen/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/pipelines/wuerstchen/__init__.py -------------------------------------------------------------------------------- /src/diffusers/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/diffusers/schedulers/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/README.md -------------------------------------------------------------------------------- /src/diffusers/schedulers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/__init__.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/deprecated/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/deprecated/__init__.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_amused.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_amused.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_ddim.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_ddim_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_ddim_flax.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_ddim_inverse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_ddim_inverse.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_ddim_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_ddim_parallel.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_ddpm.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_ddpm_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_ddpm_flax.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_ddpm_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_ddpm_parallel.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_ddpm_wuerstchen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_ddpm_wuerstchen.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_deis_multistep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_deis_multistep.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_dpmsolver_sde.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_dpmsolver_sde.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_edm_euler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_edm_euler.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_euler_discrete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_euler_discrete.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_heun_discrete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_heun_discrete.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_ipndm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_ipndm.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_karras_ve_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_karras_ve_flax.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_lcm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_lcm.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_lms_discrete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_lms_discrete.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_pndm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_pndm.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_pndm_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_pndm_flax.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_repaint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_repaint.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_sasolver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_sasolver.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_sde_ve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_sde_ve.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_sde_ve_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_sde_ve_flax.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_tcd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_tcd.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_unclip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_unclip.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_unipc_multistep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_unipc_multistep.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_utils.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_utils_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_utils_flax.py -------------------------------------------------------------------------------- /src/diffusers/schedulers/scheduling_vq_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/schedulers/scheduling_vq_diffusion.py -------------------------------------------------------------------------------- /src/diffusers/training_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/training_utils.py -------------------------------------------------------------------------------- /src/diffusers/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/__init__.py -------------------------------------------------------------------------------- /src/diffusers/utils/__pycache__/logging.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/__pycache__/logging.cpython-38.pyc -------------------------------------------------------------------------------- /src/diffusers/utils/__pycache__/outputs.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/__pycache__/outputs.cpython-38.pyc -------------------------------------------------------------------------------- /src/diffusers/utils/accelerate_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/accelerate_utils.py -------------------------------------------------------------------------------- /src/diffusers/utils/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/constants.py -------------------------------------------------------------------------------- /src/diffusers/utils/deprecation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/deprecation_utils.py -------------------------------------------------------------------------------- /src/diffusers/utils/doc_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/doc_utils.py -------------------------------------------------------------------------------- /src/diffusers/utils/dummy_flax_objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/dummy_flax_objects.py -------------------------------------------------------------------------------- /src/diffusers/utils/dummy_note_seq_objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/dummy_note_seq_objects.py -------------------------------------------------------------------------------- /src/diffusers/utils/dummy_onnx_objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/dummy_onnx_objects.py -------------------------------------------------------------------------------- /src/diffusers/utils/dummy_pt_objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/dummy_pt_objects.py -------------------------------------------------------------------------------- /src/diffusers/utils/dummy_torch_and_librosa_objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/dummy_torch_and_librosa_objects.py -------------------------------------------------------------------------------- /src/diffusers/utils/dummy_torch_and_scipy_objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/dummy_torch_and_scipy_objects.py -------------------------------------------------------------------------------- /src/diffusers/utils/dynamic_modules_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/dynamic_modules_utils.py -------------------------------------------------------------------------------- /src/diffusers/utils/export_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/export_utils.py -------------------------------------------------------------------------------- /src/diffusers/utils/hub_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/hub_utils.py -------------------------------------------------------------------------------- /src/diffusers/utils/import_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/import_utils.py -------------------------------------------------------------------------------- /src/diffusers/utils/loading_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/loading_utils.py -------------------------------------------------------------------------------- /src/diffusers/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/logging.py -------------------------------------------------------------------------------- /src/diffusers/utils/model_card_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/model_card_template.md -------------------------------------------------------------------------------- /src/diffusers/utils/outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/outputs.py -------------------------------------------------------------------------------- /src/diffusers/utils/peft_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/peft_utils.py -------------------------------------------------------------------------------- /src/diffusers/utils/pil_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/pil_utils.py -------------------------------------------------------------------------------- /src/diffusers/utils/state_dict_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/state_dict_utils.py -------------------------------------------------------------------------------- /src/diffusers/utils/testing_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/testing_utils.py -------------------------------------------------------------------------------- /src/diffusers/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/torch_utils.py -------------------------------------------------------------------------------- /src/diffusers/utils/versions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/utils/versions.py -------------------------------------------------------------------------------- /src/diffusers/video_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/diffusers/video_processor.py -------------------------------------------------------------------------------- /src/models/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/attention.py -------------------------------------------------------------------------------- /src/models/attention_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/attention_processor.py -------------------------------------------------------------------------------- /src/models/dino_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dino_model.py -------------------------------------------------------------------------------- /src/models/dinov2/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/.gitignore -------------------------------------------------------------------------------- /src/models/dinov2/CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /src/models/dinov2/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/CONTRIBUTING.md -------------------------------------------------------------------------------- /src/models/dinov2/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/LICENSE -------------------------------------------------------------------------------- /src/models/dinov2/MODEL_CARD.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/MODEL_CARD.md -------------------------------------------------------------------------------- /src/models/dinov2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/README.md -------------------------------------------------------------------------------- /src/models/dinov2/conda-extras.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/conda-extras.yaml -------------------------------------------------------------------------------- /src/models/dinov2/conda.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/conda.yaml -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/configs/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/configs/train/vitg14.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/configs/train/vitg14.yaml -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/configs/train/vitl14.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/configs/train/vitl14.yaml -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/data/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/data/adapters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/data/adapters.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/data/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/data/augmentations.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/data/collate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/data/collate.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/data/datasets/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/data/datasets/decoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/data/datasets/decoders.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/data/datasets/extended.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/data/datasets/extended.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/data/datasets/image_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/data/datasets/image_net.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/data/loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/data/loaders.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/data/masking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/data/masking.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/data/samplers.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/data/transforms.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/distributed/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/distributed/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/eval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/eval/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/eval/depth/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/eval/depth/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/eval/depth/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/eval/depth/models/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/eval/depth/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/eval/depth/models/builder.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/eval/depth/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/eval/depth/ops/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/eval/depth/ops/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/eval/depth/ops/wrappers.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/eval/knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/eval/knn.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/eval/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/eval/linear.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/eval/log_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/eval/log_regression.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/eval/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/eval/metrics.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/eval/segmentation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/eval/segmentation/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/eval/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/eval/setup.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/eval/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/eval/utils.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/fsdp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/fsdp/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/hub/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/hub/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/hub/backbones.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/hub/backbones.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/hub/classifiers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/hub/classifiers.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/hub/depth/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/hub/depth/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/hub/depth/decode_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/hub/depth/decode_heads.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/hub/depth/encoder_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/hub/depth/encoder_decoder.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/hub/depth/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/hub/depth/ops.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/hub/depthers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/hub/depthers.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/hub/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/hub/utils.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/layers/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/layers/attention.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/layers/block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/layers/block.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/layers/dino_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/layers/dino_head.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/layers/drop_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/layers/drop_path.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/layers/layer_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/layers/layer_scale.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/layers/mlp.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/layers/patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/layers/patch_embed.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/layers/swiglu_ffn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/layers/swiglu_ffn.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/logging/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/logging/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/logging/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/logging/helpers.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/loss/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/loss/dino_clstoken_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/loss/dino_clstoken_loss.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/loss/ibot_patch_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/loss/ibot_patch_loss.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/loss/koleo_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/loss/koleo_loss.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/models/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/models/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/models/vision_transformer.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/run/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/run/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/run/eval/knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/run/eval/knn.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/run/eval/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/run/eval/linear.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/run/eval/log_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/run/eval/log_regression.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/run/submit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/run/submit.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/run/train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/run/train/train.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/train/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/train/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/train/ssl_meta_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/train/ssl_meta_arch.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/train/train.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/utils/__init__.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/utils/cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/utils/cluster.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/utils/config.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/utils/dtype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/utils/dtype.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/utils/param_groups.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/utils/param_groups.py -------------------------------------------------------------------------------- /src/models/dinov2/dinov2/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/dinov2/utils/utils.py -------------------------------------------------------------------------------- /src/models/dinov2/hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/hubconf.py -------------------------------------------------------------------------------- /src/models/dinov2/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/pyproject.toml -------------------------------------------------------------------------------- /src/models/dinov2/requirements-dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/requirements-dev.txt -------------------------------------------------------------------------------- /src/models/dinov2/requirements-extras.txt: -------------------------------------------------------------------------------- 1 | mmcv-full==1.5.0 2 | mmsegmentation==0.27.0 3 | -------------------------------------------------------------------------------- /src/models/dinov2/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/requirements.txt -------------------------------------------------------------------------------- /src/models/dinov2/scripts/lint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/scripts/lint.sh -------------------------------------------------------------------------------- /src/models/dinov2/setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/setup.cfg -------------------------------------------------------------------------------- /src/models/dinov2/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/dinov2/setup.py -------------------------------------------------------------------------------- /src/models/mutual_self_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/mutual_self_attention.py -------------------------------------------------------------------------------- /src/models/mutual_self_attention_multi_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/mutual_self_attention_multi_scale.py -------------------------------------------------------------------------------- /src/models/projection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/projection.py -------------------------------------------------------------------------------- /src/models/refunet_2d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/refunet_2d_condition.py -------------------------------------------------------------------------------- /src/models/transformer_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/transformer_2d.py -------------------------------------------------------------------------------- /src/models/unet_2d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/unet_2d_blocks.py -------------------------------------------------------------------------------- /src/models/unet_2d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/src/models/unet_2d_condition.py -------------------------------------------------------------------------------- /training/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/training/.DS_Store -------------------------------------------------------------------------------- /training/dataset_configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/training/dataset_configuration.py -------------------------------------------------------------------------------- /training/setting_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/training/setting_config.py -------------------------------------------------------------------------------- /utils/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/utils/data_util.py -------------------------------------------------------------------------------- /utils/diffusion_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/utils/diffusion_util.py -------------------------------------------------------------------------------- /utils/image_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/utils/image_util.py -------------------------------------------------------------------------------- /utils/seed_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YinHan-Zhang/MagicColor/HEAD/utils/seed_all.py --------------------------------------------------------------------------------