├── .clang-format ├── .git-blame-ignore-revs ├── .gitattributes ├── .github ├── ISSUE_TEMPLATE │ ├── bug-report.yml │ ├── config.yml │ ├── documentation.yml │ └── feature-request.yml ├── PULL_REQUEST_TEMPLATE.md ├── failed_schedule_issue_template.md ├── process_commit.py ├── pytorch-probot.yml ├── scripts │ ├── cmake.sh │ ├── export_IS_M1_CONDA_BUILD_JOB.sh │ ├── setup-env.sh │ └── unittest.sh └── workflows │ ├── build-cmake.yml │ ├── build-wheels-aarch64-linux.yml │ ├── build-wheels-linux.yml │ ├── build-wheels-m1.yml │ ├── build-wheels-windows.yml │ ├── build_wheel_windows_arm64.yml │ ├── docs.yml │ ├── lint.yml │ ├── pr-labels.yml │ ├── prototype-tests-linux-gpu.yml │ ├── tests-schedule.yml │ ├── tests.yml │ └── update-viablestrict.yml ├── .gitignore ├── .pre-commit-config.yaml ├── CITATION.cff ├── CMakeLists.txt ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── CONTRIBUTING_MODELS.md ├── LICENSE ├── MANIFEST.in ├── README.md ├── android ├── .gitignore ├── README.md ├── build.gradle ├── gradle.properties ├── gradle │ └── wrapper │ │ ├── gradle-wrapper.jar │ │ └── gradle-wrapper.properties ├── gradle_scripts │ ├── android_tasks.gradle │ └── release.gradle ├── gradlew ├── gradlew.bat ├── ops │ ├── CMakeLists.txt │ ├── build.gradle │ ├── gradle.properties │ └── src │ │ └── main │ │ └── AndroidManifest.xml ├── settings.gradle └── test_app │ ├── app │ ├── build.gradle │ └── src │ │ └── main │ │ ├── AndroidManifest.xml │ │ ├── java │ │ └── org │ │ │ └── pytorch │ │ │ └── testapp │ │ │ ├── BBox.java │ │ │ ├── CameraActivity.java │ │ │ ├── MainActivity.java │ │ │ └── Result.java │ │ └── res │ │ ├── layout │ │ ├── activity_camera.xml │ │ ├── activity_main.xml │ │ └── texture_view.xml │ │ ├── mipmap-mdpi │ │ ├── ic_launcher.png │ │ └── ic_launcher_round.png │ │ └── values │ │ ├── colors.xml │ │ ├── strings.xml │ │ └── styles.xml │ └── make_assets.py ├── benchmarks └── encoding_decoding.py ├── cmake ├── TorchVisionConfig.cmake.in └── iOS.cmake ├── docs ├── Makefile ├── make.bat ├── requirements.txt └── source │ ├── _static │ ├── css │ │ └── custom_torchvision.css │ └── img │ │ ├── pytorch-logo-dark.png │ │ ├── pytorch-logo-dark.svg │ │ ├── pytorch-logo-flame.png │ │ └── pytorch-logo-flame.svg │ ├── _templates │ ├── class.rst │ ├── class_dataset.rst │ ├── function.rst │ └── layout.html │ ├── beta_status.py │ ├── conf.py │ ├── datasets.rst │ ├── docutils.conf │ ├── feature_extraction.rst │ ├── index.rst │ ├── io.rst │ ├── models.rst │ ├── models │ ├── alexnet.rst │ ├── convnext.rst │ ├── deeplabv3.rst │ ├── densenet.rst │ ├── efficientnet.rst │ ├── efficientnetv2.rst │ ├── faster_rcnn.rst │ ├── fcn.rst │ ├── fcos.rst │ ├── googlenet.rst │ ├── googlenet_quant.rst │ ├── inception.rst │ ├── inception_quant.rst │ ├── keypoint_rcnn.rst │ ├── lraspp.rst │ ├── mask_rcnn.rst │ ├── maxvit.rst │ ├── mnasnet.rst │ ├── mobilenetv2.rst │ ├── mobilenetv2_quant.rst │ ├── mobilenetv3.rst │ ├── mobilenetv3_quant.rst │ ├── raft.rst │ ├── regnet.rst │ ├── resnet.rst │ ├── resnet_quant.rst │ ├── resnext.rst │ ├── resnext_quant.rst │ ├── retinanet.rst │ ├── shufflenetv2.rst │ ├── shufflenetv2_quant.rst │ ├── squeezenet.rst │ ├── ssd.rst │ ├── ssdlite.rst │ ├── swin_transformer.rst │ ├── vgg.rst │ ├── video_mvit.rst │ ├── video_resnet.rst │ ├── video_s3d.rst │ ├── video_swin_transformer.rst │ ├── vision_transformer.rst │ └── wide_resnet.rst │ ├── ops.rst │ ├── training_references.rst │ ├── transforms.rst │ ├── tv_tensors.rst │ └── utils.rst ├── examples ├── cpp │ ├── CMakeLists.txt │ ├── README.md │ ├── run_model.cpp │ └── script_model.py └── python │ └── README.md ├── gallery ├── README.rst ├── assets │ ├── FudanPed00054.png │ ├── FudanPed00054_mask.png │ ├── astronaut.jpg │ ├── basketball.mp4 │ ├── coco │ │ ├── images │ │ │ ├── 000000000001.jpg │ │ │ └── 000000000002.jpg │ │ └── instances.json │ ├── dog1.jpg │ ├── dog2.jpg │ ├── imagenet_class_index.json │ ├── leaning_tower.jpg │ ├── person1.jpg │ ├── pottery.jpg │ ├── repurposing_annotations_thumbnail.png │ ├── transforms_thumbnail.png │ └── visualization_utils_thumbnail2.png ├── others │ ├── README.rst │ ├── plot_optical_flow.py │ ├── plot_repurposing_annotations.py │ ├── plot_scripted_tensor_transforms.py │ └── plot_visualization_utils.py └── transforms │ ├── README.rst │ ├── helpers.py │ ├── plot_custom_transforms.py │ ├── plot_custom_tv_tensors.py │ ├── plot_cutmix_mixup.py │ ├── plot_keypoints_transforms.py │ ├── plot_rotated_box_transforms.py │ ├── plot_transforms_e2e.py │ ├── plot_transforms_getting_started.py │ ├── plot_transforms_illustrations.py │ └── plot_tv_tensors.py ├── hubconf.py ├── ios ├── CMakeLists.txt ├── LibTorchvision.podspec ├── README.md ├── VisionTestApp │ ├── VisionTestApp.xcodeproj │ │ ├── project.pbxproj │ │ └── project.xcworkspace │ │ │ ├── contents.xcworkspacedata │ │ │ └── xcshareddata │ │ │ └── IDEWorkspaceChecks.plist │ ├── VisionTestApp │ │ ├── AppDelegate.h │ │ ├── AppDelegate.m │ │ ├── Assets.xcassets │ │ │ ├── AccentColor.colorset │ │ │ │ └── Contents.json │ │ │ ├── AppIcon.appiconset │ │ │ │ └── Contents.json │ │ │ └── Contents.json │ │ ├── Base.lproj │ │ │ ├── LaunchScreen.storyboard │ │ │ └── Main.storyboard │ │ ├── Info.plist │ │ ├── ModelRunner.h │ │ ├── ModelRunner.mm │ │ ├── ViewController.h │ │ ├── ViewController.mm │ │ └── main.m │ ├── clean.sh │ ├── make_assets.py │ └── setup.sh └── build_ios.sh ├── maintainer_guide.md ├── mypy.ini ├── packaging ├── cut_release.sh ├── post_build_script.sh ├── pre_build_script.sh ├── pre_build_script_arm64.sh ├── wheel │ └── relocate.py └── windows │ └── internal │ ├── build_cmake.bat │ ├── build_cpp_example.bat │ └── vc_env_helper.bat ├── pyproject.toml ├── pytest.ini ├── references ├── classification │ ├── README.md │ ├── presets.py │ ├── sampler.py │ ├── train.py │ ├── train_quantization.py │ ├── transforms.py │ └── utils.py ├── depth │ └── stereo │ │ ├── README.md │ │ ├── __init__.py │ │ ├── assets │ │ ├── Loss.jpg │ │ ├── disparity-background-mode-collapse.jpg │ │ ├── disparity-domain-drift.jpg │ │ └── gradient-norm-removal.jpg │ │ ├── cascade_evaluation.py │ │ ├── parsing.py │ │ ├── presets.py │ │ ├── train.py │ │ ├── transforms.py │ │ ├── utils │ │ ├── __init__.py │ │ ├── distributed.py │ │ ├── logger.py │ │ ├── losses.py │ │ ├── metrics.py │ │ ├── norm.py │ │ └── padder.py │ │ └── visualization.py ├── detection │ ├── README.md │ ├── coco_eval.py │ ├── coco_utils.py │ ├── engine.py │ ├── group_by_aspect_ratio.py │ ├── presets.py │ ├── train.py │ ├── transforms.py │ └── utils.py ├── optical_flow │ ├── README.md │ ├── presets.py │ ├── train.py │ ├── transforms.py │ └── utils.py ├── segmentation │ ├── README.md │ ├── coco_utils.py │ ├── presets.py │ ├── train.py │ ├── transforms.py │ ├── utils.py │ └── v2_extras.py ├── similarity │ ├── README.md │ ├── loss.py │ ├── model.py │ ├── sampler.py │ ├── test.py │ └── train.py └── video_classification │ ├── README.md │ ├── datasets.py │ ├── presets.py │ ├── train.py │ ├── transforms.py │ └── utils.py ├── release ├── README.md └── apply-release-changes.py ├── scripts ├── README.rst ├── collect_model_urls.py ├── download_model_urls.py ├── fbcode_to_main_sync.sh └── release_notes │ ├── classify_prs.py │ └── retrieve_prs_data.py ├── setup.cfg ├── setup.py ├── test ├── _utils_internal.py ├── assets │ ├── damaged_jpeg │ │ ├── TensorFlow-LICENSE │ │ ├── bad_huffman.jpg │ │ ├── corrupt.jpg │ │ ├── corrupt34_2.jpg │ │ ├── corrupt34_3.jpg │ │ └── corrupt34_4.jpg │ ├── damaged_png │ │ └── sigsegv.png │ ├── encode_jpeg │ │ ├── grace_hopper_517x606.jpg │ │ └── jpeg_write │ │ │ └── grace_hopper_517x606_pil.jpg │ ├── expected_flow.pt │ ├── fakedata │ │ ├── draw_boxes_different_label_background_colors.png │ │ ├── draw_boxes_different_label_colors.png │ │ ├── draw_boxes_different_label_fill_colors.png │ │ ├── draw_boxes_util.png │ │ ├── draw_boxes_vanilla.png │ │ ├── draw_keypoint_vanilla.png │ │ ├── draw_keypoints_visibility.png │ │ ├── draw_rotated_boxes.png │ │ ├── draw_rotated_boxes_fill.png │ │ ├── imagefolder │ │ │ ├── a │ │ │ │ ├── a1.png │ │ │ │ ├── a2.png │ │ │ │ └── a3.png │ │ │ └── b │ │ │ │ ├── b1.png │ │ │ │ ├── b2.png │ │ │ │ ├── b3.png │ │ │ │ └── b4.png │ │ └── logos │ │ │ ├── cmyk_pytorch.jpg │ │ │ ├── gray_pytorch.jpg │ │ │ ├── gray_pytorch.png │ │ │ ├── grayalpha_pytorch.png │ │ │ ├── palette_pytorch.png │ │ │ ├── rgb_pytorch.avif │ │ │ ├── rgb_pytorch.jpg │ │ │ ├── rgb_pytorch.png │ │ │ ├── rgb_pytorch.webp │ │ │ ├── rgb_pytorch16.png │ │ │ ├── rgb_pytorch_incorrectly_encoded_but_who_cares.heic │ │ │ ├── rgbalpha_pytorch.png │ │ │ └── rgbalpha_pytorch16.png │ ├── gaussian_blur_opencv_results.pt │ ├── interlaced_png │ │ ├── wizard_low-interlaced.png │ │ └── wizard_low.png │ ├── labeled_image.png │ ├── masks.tiff │ ├── toosmall_png │ │ └── heapbof.png │ └── videos │ │ ├── R6llTwEh07w.mp4 │ │ ├── RATRACE_wave_f_nm_np1_fr_goo_37.avi │ │ ├── README │ │ ├── SOX5yA1l24A.mp4 │ │ ├── SchoolRulesHowTheyHelpUs_wave_f_nm_np1_ba_med_0.avi │ │ ├── TrumanShow_wave_f_nm_np1_fr_med_26.avi │ │ ├── WUzgd7C1pWA.mp4 │ │ ├── hmdb51_Turnk_r_Pippi_Michel_cartwheel_f_cm_np2_le_med_6.avi │ │ ├── v_SoccerJuggling_g23_c01.avi │ │ └── v_SoccerJuggling_g24_c01.avi ├── builtin_dataset_mocks.py ├── common_extended_utils.py ├── common_utils.py ├── conftest.py ├── cpp │ └── test_custom_operators.cpp ├── datasets_utils.py ├── expect │ ├── ModelTester.test_alexnet_expect.pkl │ ├── ModelTester.test_convnext_base_expect.pkl │ ├── ModelTester.test_convnext_large_expect.pkl │ ├── ModelTester.test_convnext_small_expect.pkl │ ├── ModelTester.test_convnext_tiny_expect.pkl │ ├── ModelTester.test_crestereo_base_expect.pkl │ ├── ModelTester.test_deeplabv3_mobilenet_v3_large_expect.pkl │ ├── ModelTester.test_deeplabv3_resnet101_expect.pkl │ ├── ModelTester.test_deeplabv3_resnet50_expect.pkl │ ├── ModelTester.test_densenet121_expect.pkl │ ├── ModelTester.test_densenet161_expect.pkl │ ├── ModelTester.test_densenet169_expect.pkl │ ├── ModelTester.test_densenet201_expect.pkl │ ├── ModelTester.test_efficientnet_b0_expect.pkl │ ├── ModelTester.test_efficientnet_b1_expect.pkl │ ├── ModelTester.test_efficientnet_b2_expect.pkl │ ├── ModelTester.test_efficientnet_b3_expect.pkl │ ├── ModelTester.test_efficientnet_b4_expect.pkl │ ├── ModelTester.test_efficientnet_b5_expect.pkl │ ├── ModelTester.test_efficientnet_b6_expect.pkl │ ├── ModelTester.test_efficientnet_b7_expect.pkl │ ├── ModelTester.test_efficientnet_v2_l_expect.pkl │ ├── ModelTester.test_efficientnet_v2_m_expect.pkl │ ├── ModelTester.test_efficientnet_v2_s_expect.pkl │ ├── ModelTester.test_fasterrcnn_mobilenet_v3_large_320_fpn_expect.pkl │ ├── ModelTester.test_fasterrcnn_mobilenet_v3_large_fpn_expect.pkl │ ├── ModelTester.test_fasterrcnn_resnet50_fpn_expect.pkl │ ├── ModelTester.test_fasterrcnn_resnet50_fpn_v2_expect.pkl │ ├── ModelTester.test_fcn_resnet101_expect.pkl │ ├── ModelTester.test_fcn_resnet50_expect.pkl │ ├── ModelTester.test_fcos_resnet50_fpn_expect.pkl │ ├── ModelTester.test_googlenet_expect.pkl │ ├── ModelTester.test_googlenet_quantized_expect.pkl │ ├── ModelTester.test_inception_v3_expect.pkl │ ├── ModelTester.test_keypointrcnn_resnet50_fpn_expect.pkl │ ├── ModelTester.test_lraspp_mobilenet_v3_large_expect.pkl │ ├── ModelTester.test_maskrcnn_resnet50_fpn_expect.pkl │ ├── ModelTester.test_maskrcnn_resnet50_fpn_v2_expect.pkl │ ├── ModelTester.test_maxvit_t_expect.pkl │ ├── ModelTester.test_mc3_18_expect.pkl │ ├── ModelTester.test_mnasnet0_5_expect.pkl │ ├── ModelTester.test_mnasnet0_75_expect.pkl │ ├── ModelTester.test_mnasnet1_0_expect.pkl │ ├── ModelTester.test_mnasnet1_3_expect.pkl │ ├── ModelTester.test_mobilenet_v2_expect.pkl │ ├── ModelTester.test_mobilenet_v2_quantized_expect.pkl │ ├── ModelTester.test_mobilenet_v3_large_expect.pkl │ ├── ModelTester.test_mobilenet_v3_large_quantized_expect.pkl │ ├── ModelTester.test_mobilenet_v3_small_expect.pkl │ ├── ModelTester.test_mobilenetv2_residual_setting_expect.pkl │ ├── ModelTester.test_mvit_v1_b_expect.pkl │ ├── ModelTester.test_mvit_v2_s_expect.pkl │ ├── ModelTester.test_r2plus1d_18_expect.pkl │ ├── ModelTester.test_r3d_18_expect.pkl │ ├── ModelTester.test_raft_large_expect.pkl │ ├── ModelTester.test_raft_small_expect.pkl │ ├── ModelTester.test_raft_stereo_base_expect.pkl │ ├── ModelTester.test_raft_stereo_realtime_expect.pkl │ ├── ModelTester.test_regnet_x_16gf_expect.pkl │ ├── ModelTester.test_regnet_x_1_6gf_expect.pkl │ ├── ModelTester.test_regnet_x_32gf_expect.pkl │ ├── ModelTester.test_regnet_x_3_2gf_expect.pkl │ ├── ModelTester.test_regnet_x_400mf_expect.pkl │ ├── ModelTester.test_regnet_x_800mf_expect.pkl │ ├── ModelTester.test_regnet_x_8gf_expect.pkl │ ├── ModelTester.test_regnet_y_128gf_expect.pkl │ ├── ModelTester.test_regnet_y_16gf_expect.pkl │ ├── ModelTester.test_regnet_y_1_6gf_expect.pkl │ ├── ModelTester.test_regnet_y_32gf_expect.pkl │ ├── ModelTester.test_regnet_y_3_2gf_expect.pkl │ ├── ModelTester.test_regnet_y_400mf_expect.pkl │ ├── ModelTester.test_regnet_y_800mf_expect.pkl │ ├── ModelTester.test_regnet_y_8gf_expect.pkl │ ├── ModelTester.test_resnet101_expect.pkl │ ├── ModelTester.test_resnet152_expect.pkl │ ├── ModelTester.test_resnet18_expect.pkl │ ├── ModelTester.test_resnet18_quantized_expect.pkl │ ├── ModelTester.test_resnet34_expect.pkl │ ├── ModelTester.test_resnet50_expect.pkl │ ├── ModelTester.test_resnet50_quantized_expect.pkl │ ├── ModelTester.test_resnext101_32x8d_expect.pkl │ ├── ModelTester.test_resnext101_32x8d_quantized_expect.pkl │ ├── ModelTester.test_resnext101_64x4d_expect.pkl │ ├── ModelTester.test_resnext101_64x4d_quantized_expect.pkl │ ├── ModelTester.test_resnext50_32x4d_expect.pkl │ ├── ModelTester.test_retinanet_resnet50_fpn_expect.pkl │ ├── ModelTester.test_retinanet_resnet50_fpn_v2_expect.pkl │ ├── ModelTester.test_s3d_expect.pkl │ ├── ModelTester.test_shufflenet_v2_x0_5_expect.pkl │ ├── ModelTester.test_shufflenet_v2_x0_5_quantized_expect.pkl │ ├── ModelTester.test_shufflenet_v2_x1_0_expect.pkl │ ├── ModelTester.test_shufflenet_v2_x1_0_quantized_expect.pkl │ ├── ModelTester.test_shufflenet_v2_x1_5_expect.pkl │ ├── ModelTester.test_shufflenet_v2_x1_5_quantized_expect.pkl │ ├── ModelTester.test_shufflenet_v2_x2_0_expect.pkl │ ├── ModelTester.test_shufflenet_v2_x2_0_quantized_expect.pkl │ ├── ModelTester.test_squeezenet1_0_expect.pkl │ ├── ModelTester.test_squeezenet1_1_expect.pkl │ ├── ModelTester.test_ssd300_vgg16_expect.pkl │ ├── ModelTester.test_ssdlite320_mobilenet_v3_large_expect.pkl │ ├── ModelTester.test_swin3d_b_expect.pkl │ ├── ModelTester.test_swin3d_s_expect.pkl │ ├── ModelTester.test_swin3d_t_expect.pkl │ ├── ModelTester.test_swin_b_expect.pkl │ ├── ModelTester.test_swin_s_expect.pkl │ ├── ModelTester.test_swin_t_expect.pkl │ ├── ModelTester.test_swin_v2_b_expect.pkl │ ├── ModelTester.test_swin_v2_s_expect.pkl │ ├── ModelTester.test_swin_v2_t_expect.pkl │ ├── ModelTester.test_vgg11_bn_expect.pkl │ ├── ModelTester.test_vgg11_expect.pkl │ ├── ModelTester.test_vgg13_bn_expect.pkl │ ├── ModelTester.test_vgg13_expect.pkl │ ├── ModelTester.test_vgg16_bn_expect.pkl │ ├── ModelTester.test_vgg16_expect.pkl │ ├── ModelTester.test_vgg19_bn_expect.pkl │ ├── ModelTester.test_vgg19_expect.pkl │ ├── ModelTester.test_vit_b_16_expect.pkl │ ├── ModelTester.test_vit_b_32_expect.pkl │ ├── ModelTester.test_vit_h_14_expect.pkl │ ├── ModelTester.test_vit_l_16_expect.pkl │ ├── ModelTester.test_vit_l_32_expect.pkl │ ├── ModelTester.test_vitc_b_16_expect.pkl │ ├── ModelTester.test_wide_resnet101_2_expect.pkl │ └── ModelTester.test_wide_resnet50_2_expect.pkl ├── optests_failures_dict.json ├── preprocess-bench.py ├── smoke_test.py ├── test_architecture_ops.py ├── test_backbone_utils.py ├── test_datasets.py ├── test_datasets_download.py ├── test_datasets_samplers.py ├── test_datasets_utils.py ├── test_datasets_video_utils.py ├── test_datasets_video_utils_opt.py ├── test_extended_models.py ├── test_functional_tensor.py ├── test_image.py ├── test_internal_utils.py ├── test_internet.py ├── test_io.py ├── test_io_opt.py ├── test_models.py ├── test_models_detection_anchor_utils.py ├── test_models_detection_negative_samples.py ├── test_models_detection_utils.py ├── test_onnx.py ├── test_ops.py ├── test_prototype_datasets_builtin.py ├── test_prototype_datasets_utils.py ├── test_prototype_models.py ├── test_prototype_transforms.py ├── test_transforms.py ├── test_transforms_tensor.py ├── test_transforms_v2.py ├── test_transforms_v2_utils.py ├── test_transforms_video.py ├── test_tv_tensors.py ├── test_utils.py ├── test_video_gpu_decoder.py ├── test_video_reader.py └── test_videoapi.py ├── torchvision ├── __init__.py ├── _internally_replaced_utils.py ├── _meta_registrations.py ├── _utils.py ├── csrc │ ├── io │ │ ├── decoder │ │ │ ├── audio_sampler.cpp │ │ │ ├── audio_sampler.h │ │ │ ├── audio_stream.cpp │ │ │ ├── audio_stream.h │ │ │ ├── cc_stream.cpp │ │ │ ├── cc_stream.h │ │ │ ├── decoder.cpp │ │ │ ├── decoder.h │ │ │ ├── defs.h │ │ │ ├── gpu │ │ │ │ ├── README.rst │ │ │ │ ├── decoder.cpp │ │ │ │ ├── decoder.h │ │ │ │ ├── demuxer.h │ │ │ │ ├── gpu_decoder.cpp │ │ │ │ └── gpu_decoder.h │ │ │ ├── memory_buffer.cpp │ │ │ ├── memory_buffer.h │ │ │ ├── seekable_buffer.cpp │ │ │ ├── seekable_buffer.h │ │ │ ├── stream.cpp │ │ │ ├── stream.h │ │ │ ├── subtitle_sampler.cpp │ │ │ ├── subtitle_sampler.h │ │ │ ├── subtitle_stream.cpp │ │ │ ├── subtitle_stream.h │ │ │ ├── sync_decoder.cpp │ │ │ ├── sync_decoder.h │ │ │ ├── sync_decoder_test.cpp │ │ │ ├── time_keeper.cpp │ │ │ ├── time_keeper.h │ │ │ ├── util.cpp │ │ │ ├── util.h │ │ │ ├── util_test.cpp │ │ │ ├── video_sampler.cpp │ │ │ ├── video_sampler.h │ │ │ ├── video_stream.cpp │ │ │ └── video_stream.h │ │ ├── image │ │ │ ├── common.cpp │ │ │ ├── common.h │ │ │ ├── cpu │ │ │ │ ├── common_jpeg.cpp │ │ │ │ ├── common_jpeg.h │ │ │ │ ├── common_png.h │ │ │ │ ├── decode_gif.cpp │ │ │ │ ├── decode_gif.h │ │ │ │ ├── decode_image.cpp │ │ │ │ ├── decode_image.h │ │ │ │ ├── decode_jpeg.cpp │ │ │ │ ├── decode_jpeg.h │ │ │ │ ├── decode_png.cpp │ │ │ │ ├── decode_png.h │ │ │ │ ├── decode_webp.cpp │ │ │ │ ├── decode_webp.h │ │ │ │ ├── encode_jpeg.cpp │ │ │ │ ├── encode_jpeg.h │ │ │ │ ├── encode_png.cpp │ │ │ │ ├── encode_png.h │ │ │ │ ├── exif.h │ │ │ │ ├── giflib │ │ │ │ │ ├── README │ │ │ │ │ ├── dgif_lib.c │ │ │ │ │ ├── gif_hash.c │ │ │ │ │ ├── gif_hash.h │ │ │ │ │ ├── gif_lib.h │ │ │ │ │ ├── gif_lib_private.h │ │ │ │ │ ├── gifalloc.c │ │ │ │ │ └── openbsd-reallocarray.c │ │ │ │ ├── read_write_file.cpp │ │ │ │ └── read_write_file.h │ │ │ ├── cuda │ │ │ │ ├── decode_jpegs_cuda.cpp │ │ │ │ ├── decode_jpegs_cuda.h │ │ │ │ ├── encode_decode_jpegs_cuda.h │ │ │ │ ├── encode_jpegs_cuda.cpp │ │ │ │ └── encode_jpegs_cuda.h │ │ │ ├── image.cpp │ │ │ └── image.h │ │ ├── video │ │ │ ├── video.cpp │ │ │ └── video.h │ │ └── video_reader │ │ │ ├── video_reader.cpp │ │ │ └── video_reader.h │ ├── macros.h │ ├── ops │ │ ├── autocast │ │ │ ├── deform_conv2d_kernel.cpp │ │ │ ├── nms_kernel.cpp │ │ │ ├── ps_roi_align_kernel.cpp │ │ │ ├── ps_roi_pool_kernel.cpp │ │ │ ├── roi_align_kernel.cpp │ │ │ └── roi_pool_kernel.cpp │ │ ├── autograd │ │ │ ├── deform_conv2d_kernel.cpp │ │ │ ├── ps_roi_align_kernel.cpp │ │ │ ├── ps_roi_pool_kernel.cpp │ │ │ ├── roi_align_kernel.cpp │ │ │ └── roi_pool_kernel.cpp │ │ ├── cpu │ │ │ ├── deform_conv2d_kernel.cpp │ │ │ ├── nms_kernel.cpp │ │ │ ├── ps_roi_align_kernel.cpp │ │ │ ├── ps_roi_pool_kernel.cpp │ │ │ ├── roi_align_common.h │ │ │ ├── roi_align_kernel.cpp │ │ │ └── roi_pool_kernel.cpp │ │ ├── cuda │ │ │ ├── cuda_helpers.h │ │ │ ├── deform_conv2d_kernel.cu │ │ │ ├── nms_kernel.cu │ │ │ ├── ps_roi_align_kernel.cu │ │ │ ├── ps_roi_pool_kernel.cu │ │ │ ├── roi_align_kernel.cu │ │ │ └── roi_pool_kernel.cu │ │ ├── deform_conv2d.cpp │ │ ├── deform_conv2d.h │ │ ├── mps │ │ │ ├── deform_conv2d_kernel.mm │ │ │ ├── mps_helpers.h │ │ │ ├── mps_kernels.h │ │ │ ├── nms_kernel.mm │ │ │ ├── ps_roi_align_kernel.mm │ │ │ ├── ps_roi_pool_kernel.mm │ │ │ ├── roi_align_kernel.mm │ │ │ └── roi_pool_kernel.mm │ │ ├── nms.cpp │ │ ├── nms.h │ │ ├── ops.h │ │ ├── ps_roi_align.cpp │ │ ├── ps_roi_align.h │ │ ├── ps_roi_pool.cpp │ │ ├── ps_roi_pool.h │ │ ├── quantized │ │ │ └── cpu │ │ │ │ ├── qnms_kernel.cpp │ │ │ │ └── qroi_align_kernel.cpp │ │ ├── roi_align.cpp │ │ ├── roi_align.h │ │ ├── roi_pool.cpp │ │ └── roi_pool.h │ ├── vision.cpp │ └── vision.h ├── datasets │ ├── __init__.py │ ├── _optical_flow.py │ ├── _stereo_matching.py │ ├── caltech.py │ ├── celeba.py │ ├── cifar.py │ ├── cityscapes.py │ ├── clevr.py │ ├── coco.py │ ├── country211.py │ ├── dtd.py │ ├── eurosat.py │ ├── fakedata.py │ ├── fer2013.py │ ├── fgvc_aircraft.py │ ├── flickr.py │ ├── flowers102.py │ ├── folder.py │ ├── food101.py │ ├── gtsrb.py │ ├── hmdb51.py │ ├── imagenet.py │ ├── imagenette.py │ ├── inaturalist.py │ ├── kinetics.py │ ├── kitti.py │ ├── lfw.py │ ├── lsun.py │ ├── mnist.py │ ├── moving_mnist.py │ ├── omniglot.py │ ├── oxford_iiit_pet.py │ ├── pcam.py │ ├── phototour.py │ ├── places365.py │ ├── rendered_sst2.py │ ├── samplers │ │ ├── __init__.py │ │ └── clip_sampler.py │ ├── sbd.py │ ├── sbu.py │ ├── semeion.py │ ├── stanford_cars.py │ ├── stl10.py │ ├── sun397.py │ ├── svhn.py │ ├── ucf101.py │ ├── usps.py │ ├── utils.py │ ├── video_utils.py │ ├── vision.py │ ├── voc.py │ └── widerface.py ├── extension.py ├── io │ ├── __init__.py │ ├── _load_gpu_decoder.py │ ├── _video_deprecation_warning.py │ ├── _video_opt.py │ ├── image.py │ ├── video.py │ └── video_reader.py ├── models │ ├── __init__.py │ ├── _api.py │ ├── _meta.py │ ├── _utils.py │ ├── alexnet.py │ ├── convnext.py │ ├── densenet.py │ ├── detection │ │ ├── __init__.py │ │ ├── _utils.py │ │ ├── anchor_utils.py │ │ ├── backbone_utils.py │ │ ├── faster_rcnn.py │ │ ├── fcos.py │ │ ├── generalized_rcnn.py │ │ ├── image_list.py │ │ ├── keypoint_rcnn.py │ │ ├── mask_rcnn.py │ │ ├── retinanet.py │ │ ├── roi_heads.py │ │ ├── rpn.py │ │ ├── ssd.py │ │ ├── ssdlite.py │ │ └── transform.py │ ├── efficientnet.py │ ├── feature_extraction.py │ ├── googlenet.py │ ├── inception.py │ ├── maxvit.py │ ├── mnasnet.py │ ├── mobilenet.py │ ├── mobilenetv2.py │ ├── mobilenetv3.py │ ├── optical_flow │ │ ├── __init__.py │ │ ├── _utils.py │ │ └── raft.py │ ├── quantization │ │ ├── __init__.py │ │ ├── googlenet.py │ │ ├── inception.py │ │ ├── mobilenet.py │ │ ├── mobilenetv2.py │ │ ├── mobilenetv3.py │ │ ├── resnet.py │ │ ├── shufflenetv2.py │ │ └── utils.py │ ├── regnet.py │ ├── resnet.py │ ├── segmentation │ │ ├── __init__.py │ │ ├── _utils.py │ │ ├── deeplabv3.py │ │ ├── fcn.py │ │ └── lraspp.py │ ├── shufflenetv2.py │ ├── squeezenet.py │ ├── swin_transformer.py │ ├── vgg.py │ ├── video │ │ ├── __init__.py │ │ ├── mvit.py │ │ ├── resnet.py │ │ ├── s3d.py │ │ └── swin_transformer.py │ └── vision_transformer.py ├── ops │ ├── __init__.py │ ├── _box_convert.py │ ├── _register_onnx_ops.py │ ├── _utils.py │ ├── boxes.py │ ├── ciou_loss.py │ ├── deform_conv.py │ ├── diou_loss.py │ ├── drop_block.py │ ├── feature_pyramid_network.py │ ├── focal_loss.py │ ├── giou_loss.py │ ├── misc.py │ ├── poolers.py │ ├── ps_roi_align.py │ ├── ps_roi_pool.py │ ├── roi_align.py │ ├── roi_pool.py │ └── stochastic_depth.py ├── prototype │ ├── __init__.py │ ├── datasets │ │ ├── README.md │ │ ├── __init__.py │ │ ├── _api.py │ │ ├── _builtin │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── caltech.py │ │ │ ├── caltech101.categories │ │ │ ├── caltech256.categories │ │ │ ├── celeba.py │ │ │ ├── cifar.py │ │ │ ├── cifar10.categories │ │ │ ├── cifar100.categories │ │ │ ├── clevr.py │ │ │ ├── coco.categories │ │ │ ├── coco.py │ │ │ ├── country211.categories │ │ │ ├── country211.py │ │ │ ├── cub200.categories │ │ │ ├── cub200.py │ │ │ ├── dtd.categories │ │ │ ├── dtd.py │ │ │ ├── eurosat.py │ │ │ ├── fer2013.py │ │ │ ├── food101.categories │ │ │ ├── food101.py │ │ │ ├── gtsrb.py │ │ │ ├── imagenet.categories │ │ │ ├── imagenet.py │ │ │ ├── mnist.py │ │ │ ├── oxford-iiit-pet.categories │ │ │ ├── oxford_iiit_pet.py │ │ │ ├── pcam.py │ │ │ ├── sbd.categories │ │ │ ├── sbd.py │ │ │ ├── semeion.py │ │ │ ├── stanford-cars.categories │ │ │ ├── stanford_cars.py │ │ │ ├── svhn.py │ │ │ ├── usps.py │ │ │ ├── voc.categories │ │ │ └── voc.py │ │ ├── _folder.py │ │ ├── _home.py │ │ ├── benchmark.py │ │ ├── generate_category_files.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── _dataset.py │ │ │ ├── _encoded.py │ │ │ ├── _internal.py │ │ │ └── _resource.py │ ├── models │ │ ├── __init__.py │ │ └── depth │ │ │ ├── __init__.py │ │ │ └── stereo │ │ │ ├── __init__.py │ │ │ ├── crestereo.py │ │ │ └── raft_stereo.py │ ├── transforms │ │ ├── __init__.py │ │ ├── _augment.py │ │ ├── _geometry.py │ │ ├── _misc.py │ │ ├── _presets.py │ │ └── _type_conversion.py │ ├── tv_tensors │ │ ├── __init__.py │ │ └── _label.py │ └── utils │ │ ├── __init__.py │ │ └── _internal.py ├── transforms │ ├── __init__.py │ ├── _functional_pil.py │ ├── _functional_tensor.py │ ├── _functional_video.py │ ├── _presets.py │ ├── _transforms_video.py │ ├── autoaugment.py │ ├── functional.py │ ├── transforms.py │ └── v2 │ │ ├── __init__.py │ │ ├── _augment.py │ │ ├── _auto_augment.py │ │ ├── _color.py │ │ ├── _container.py │ │ ├── _deprecated.py │ │ ├── _geometry.py │ │ ├── _meta.py │ │ ├── _misc.py │ │ ├── _temporal.py │ │ ├── _transform.py │ │ ├── _type_conversion.py │ │ ├── _utils.py │ │ └── functional │ │ ├── __init__.py │ │ ├── _augment.py │ │ ├── _color.py │ │ ├── _deprecated.py │ │ ├── _geometry.py │ │ ├── _meta.py │ │ ├── _misc.py │ │ ├── _temporal.py │ │ ├── _type_conversion.py │ │ └── _utils.py ├── tv_tensors │ ├── __init__.py │ ├── _bounding_boxes.py │ ├── _dataset_wrapper.py │ ├── _image.py │ ├── _keypoints.py │ ├── _mask.py │ ├── _torch_function_helpers.py │ ├── _tv_tensor.py │ └── _video.py └── utils.py └── version.txt /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.clang-format -------------------------------------------------------------------------------- /.git-blame-ignore-revs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.git-blame-ignore-revs -------------------------------------------------------------------------------- /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.gitattributes -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug-report.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/ISSUE_TEMPLATE/bug-report.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/documentation.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/ISSUE_TEMPLATE/documentation.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature-request.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/ISSUE_TEMPLATE/feature-request.yml -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.github/failed_schedule_issue_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/failed_schedule_issue_template.md -------------------------------------------------------------------------------- /.github/process_commit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/process_commit.py -------------------------------------------------------------------------------- /.github/pytorch-probot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/pytorch-probot.yml -------------------------------------------------------------------------------- /.github/scripts/cmake.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/scripts/cmake.sh -------------------------------------------------------------------------------- /.github/scripts/export_IS_M1_CONDA_BUILD_JOB.sh: -------------------------------------------------------------------------------- 1 | #!/bin/sh 2 | 3 | export IS_M1_CONDA_BUILD_JOB=1 4 | -------------------------------------------------------------------------------- /.github/scripts/setup-env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/scripts/setup-env.sh -------------------------------------------------------------------------------- /.github/scripts/unittest.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/scripts/unittest.sh -------------------------------------------------------------------------------- /.github/workflows/build-cmake.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/workflows/build-cmake.yml -------------------------------------------------------------------------------- /.github/workflows/build-wheels-aarch64-linux.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/workflows/build-wheels-aarch64-linux.yml -------------------------------------------------------------------------------- /.github/workflows/build-wheels-linux.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/workflows/build-wheels-linux.yml -------------------------------------------------------------------------------- /.github/workflows/build-wheels-m1.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/workflows/build-wheels-m1.yml -------------------------------------------------------------------------------- /.github/workflows/build-wheels-windows.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/workflows/build-wheels-windows.yml -------------------------------------------------------------------------------- /.github/workflows/build_wheel_windows_arm64.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/workflows/build_wheel_windows_arm64.yml -------------------------------------------------------------------------------- /.github/workflows/docs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/workflows/docs.yml -------------------------------------------------------------------------------- /.github/workflows/lint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/workflows/lint.yml -------------------------------------------------------------------------------- /.github/workflows/pr-labels.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/workflows/pr-labels.yml -------------------------------------------------------------------------------- /.github/workflows/prototype-tests-linux-gpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/workflows/prototype-tests-linux-gpu.yml -------------------------------------------------------------------------------- /.github/workflows/tests-schedule.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/workflows/tests-schedule.yml -------------------------------------------------------------------------------- /.github/workflows/tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/workflows/tests.yml -------------------------------------------------------------------------------- /.github/workflows/update-viablestrict.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.github/workflows/update-viablestrict.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/CITATION.cff -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /CONTRIBUTING_MODELS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/CONTRIBUTING_MODELS.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/README.md -------------------------------------------------------------------------------- /android/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/.gitignore -------------------------------------------------------------------------------- /android/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/README.md -------------------------------------------------------------------------------- /android/build.gradle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/build.gradle -------------------------------------------------------------------------------- /android/gradle.properties: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/gradle.properties -------------------------------------------------------------------------------- /android/gradle/wrapper/gradle-wrapper.jar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/gradle/wrapper/gradle-wrapper.jar -------------------------------------------------------------------------------- /android/gradle/wrapper/gradle-wrapper.properties: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/gradle/wrapper/gradle-wrapper.properties -------------------------------------------------------------------------------- /android/gradle_scripts/android_tasks.gradle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/gradle_scripts/android_tasks.gradle -------------------------------------------------------------------------------- /android/gradle_scripts/release.gradle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/gradle_scripts/release.gradle -------------------------------------------------------------------------------- /android/gradlew: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/gradlew -------------------------------------------------------------------------------- /android/gradlew.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/gradlew.bat -------------------------------------------------------------------------------- /android/ops/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/ops/CMakeLists.txt -------------------------------------------------------------------------------- /android/ops/build.gradle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/ops/build.gradle -------------------------------------------------------------------------------- /android/ops/gradle.properties: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/ops/gradle.properties -------------------------------------------------------------------------------- /android/ops/src/main/AndroidManifest.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/ops/src/main/AndroidManifest.xml -------------------------------------------------------------------------------- /android/settings.gradle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/settings.gradle -------------------------------------------------------------------------------- /android/test_app/app/build.gradle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/test_app/app/build.gradle -------------------------------------------------------------------------------- /android/test_app/app/src/main/AndroidManifest.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/test_app/app/src/main/AndroidManifest.xml -------------------------------------------------------------------------------- /android/test_app/app/src/main/res/layout/activity_main.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/test_app/app/src/main/res/layout/activity_main.xml -------------------------------------------------------------------------------- /android/test_app/app/src/main/res/layout/texture_view.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/test_app/app/src/main/res/layout/texture_view.xml -------------------------------------------------------------------------------- /android/test_app/app/src/main/res/values/colors.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/test_app/app/src/main/res/values/colors.xml -------------------------------------------------------------------------------- /android/test_app/app/src/main/res/values/strings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/test_app/app/src/main/res/values/strings.xml -------------------------------------------------------------------------------- /android/test_app/app/src/main/res/values/styles.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/test_app/app/src/main/res/values/styles.xml -------------------------------------------------------------------------------- /android/test_app/make_assets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/android/test_app/make_assets.py -------------------------------------------------------------------------------- /benchmarks/encoding_decoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/benchmarks/encoding_decoding.py -------------------------------------------------------------------------------- /cmake/TorchVisionConfig.cmake.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/cmake/TorchVisionConfig.cmake.in -------------------------------------------------------------------------------- /cmake/iOS.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/cmake/iOS.cmake -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/source/_static/css/custom_torchvision.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/_static/css/custom_torchvision.css -------------------------------------------------------------------------------- /docs/source/_static/img/pytorch-logo-dark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/_static/img/pytorch-logo-dark.png -------------------------------------------------------------------------------- /docs/source/_static/img/pytorch-logo-dark.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/_static/img/pytorch-logo-dark.svg -------------------------------------------------------------------------------- /docs/source/_static/img/pytorch-logo-flame.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/_static/img/pytorch-logo-flame.png -------------------------------------------------------------------------------- /docs/source/_static/img/pytorch-logo-flame.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/_static/img/pytorch-logo-flame.svg -------------------------------------------------------------------------------- /docs/source/_templates/class.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/_templates/class.rst -------------------------------------------------------------------------------- /docs/source/_templates/class_dataset.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/_templates/class_dataset.rst -------------------------------------------------------------------------------- /docs/source/_templates/function.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/_templates/function.rst -------------------------------------------------------------------------------- /docs/source/_templates/layout.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/_templates/layout.html -------------------------------------------------------------------------------- /docs/source/beta_status.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/beta_status.py -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/datasets.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/datasets.rst -------------------------------------------------------------------------------- /docs/source/docutils.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/docutils.conf -------------------------------------------------------------------------------- /docs/source/feature_extraction.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/feature_extraction.rst -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/io.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/io.rst -------------------------------------------------------------------------------- /docs/source/models.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models.rst -------------------------------------------------------------------------------- /docs/source/models/alexnet.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/alexnet.rst -------------------------------------------------------------------------------- /docs/source/models/convnext.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/convnext.rst -------------------------------------------------------------------------------- /docs/source/models/deeplabv3.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/deeplabv3.rst -------------------------------------------------------------------------------- /docs/source/models/densenet.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/densenet.rst -------------------------------------------------------------------------------- /docs/source/models/efficientnet.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/efficientnet.rst -------------------------------------------------------------------------------- /docs/source/models/efficientnetv2.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/efficientnetv2.rst -------------------------------------------------------------------------------- /docs/source/models/faster_rcnn.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/faster_rcnn.rst -------------------------------------------------------------------------------- /docs/source/models/fcn.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/fcn.rst -------------------------------------------------------------------------------- /docs/source/models/fcos.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/fcos.rst -------------------------------------------------------------------------------- /docs/source/models/googlenet.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/googlenet.rst -------------------------------------------------------------------------------- /docs/source/models/googlenet_quant.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/googlenet_quant.rst -------------------------------------------------------------------------------- /docs/source/models/inception.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/inception.rst -------------------------------------------------------------------------------- /docs/source/models/inception_quant.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/inception_quant.rst -------------------------------------------------------------------------------- /docs/source/models/keypoint_rcnn.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/keypoint_rcnn.rst -------------------------------------------------------------------------------- /docs/source/models/lraspp.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/lraspp.rst -------------------------------------------------------------------------------- /docs/source/models/mask_rcnn.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/mask_rcnn.rst -------------------------------------------------------------------------------- /docs/source/models/maxvit.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/maxvit.rst -------------------------------------------------------------------------------- /docs/source/models/mnasnet.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/mnasnet.rst -------------------------------------------------------------------------------- /docs/source/models/mobilenetv2.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/mobilenetv2.rst -------------------------------------------------------------------------------- /docs/source/models/mobilenetv2_quant.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/mobilenetv2_quant.rst -------------------------------------------------------------------------------- /docs/source/models/mobilenetv3.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/mobilenetv3.rst -------------------------------------------------------------------------------- /docs/source/models/mobilenetv3_quant.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/mobilenetv3_quant.rst -------------------------------------------------------------------------------- /docs/source/models/raft.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/raft.rst -------------------------------------------------------------------------------- /docs/source/models/regnet.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/regnet.rst -------------------------------------------------------------------------------- /docs/source/models/resnet.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/resnet.rst -------------------------------------------------------------------------------- /docs/source/models/resnet_quant.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/resnet_quant.rst -------------------------------------------------------------------------------- /docs/source/models/resnext.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/resnext.rst -------------------------------------------------------------------------------- /docs/source/models/resnext_quant.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/resnext_quant.rst -------------------------------------------------------------------------------- /docs/source/models/retinanet.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/retinanet.rst -------------------------------------------------------------------------------- /docs/source/models/shufflenetv2.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/shufflenetv2.rst -------------------------------------------------------------------------------- /docs/source/models/shufflenetv2_quant.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/shufflenetv2_quant.rst -------------------------------------------------------------------------------- /docs/source/models/squeezenet.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/squeezenet.rst -------------------------------------------------------------------------------- /docs/source/models/ssd.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/ssd.rst -------------------------------------------------------------------------------- /docs/source/models/ssdlite.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/ssdlite.rst -------------------------------------------------------------------------------- /docs/source/models/swin_transformer.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/swin_transformer.rst -------------------------------------------------------------------------------- /docs/source/models/vgg.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/vgg.rst -------------------------------------------------------------------------------- /docs/source/models/video_mvit.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/video_mvit.rst -------------------------------------------------------------------------------- /docs/source/models/video_resnet.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/video_resnet.rst -------------------------------------------------------------------------------- /docs/source/models/video_s3d.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/video_s3d.rst -------------------------------------------------------------------------------- /docs/source/models/video_swin_transformer.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/video_swin_transformer.rst -------------------------------------------------------------------------------- /docs/source/models/vision_transformer.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/vision_transformer.rst -------------------------------------------------------------------------------- /docs/source/models/wide_resnet.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/models/wide_resnet.rst -------------------------------------------------------------------------------- /docs/source/ops.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/ops.rst -------------------------------------------------------------------------------- /docs/source/training_references.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/training_references.rst -------------------------------------------------------------------------------- /docs/source/transforms.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/transforms.rst -------------------------------------------------------------------------------- /docs/source/tv_tensors.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/tv_tensors.rst -------------------------------------------------------------------------------- /docs/source/utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/docs/source/utils.rst -------------------------------------------------------------------------------- /examples/cpp/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/examples/cpp/CMakeLists.txt -------------------------------------------------------------------------------- /examples/cpp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/examples/cpp/README.md -------------------------------------------------------------------------------- /examples/cpp/run_model.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/examples/cpp/run_model.cpp -------------------------------------------------------------------------------- /examples/cpp/script_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/examples/cpp/script_model.py -------------------------------------------------------------------------------- /examples/python/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/examples/python/README.md -------------------------------------------------------------------------------- /gallery/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/README.rst -------------------------------------------------------------------------------- /gallery/assets/FudanPed00054.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/assets/FudanPed00054.png -------------------------------------------------------------------------------- /gallery/assets/FudanPed00054_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/assets/FudanPed00054_mask.png -------------------------------------------------------------------------------- /gallery/assets/astronaut.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/assets/astronaut.jpg -------------------------------------------------------------------------------- /gallery/assets/basketball.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/assets/basketball.mp4 -------------------------------------------------------------------------------- /gallery/assets/coco/images/000000000001.jpg: -------------------------------------------------------------------------------- 1 | ../../astronaut.jpg -------------------------------------------------------------------------------- /gallery/assets/coco/images/000000000002.jpg: -------------------------------------------------------------------------------- 1 | ../../dog2.jpg -------------------------------------------------------------------------------- /gallery/assets/coco/instances.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/assets/coco/instances.json -------------------------------------------------------------------------------- /gallery/assets/dog1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/assets/dog1.jpg -------------------------------------------------------------------------------- /gallery/assets/dog2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/assets/dog2.jpg -------------------------------------------------------------------------------- /gallery/assets/imagenet_class_index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/assets/imagenet_class_index.json -------------------------------------------------------------------------------- /gallery/assets/leaning_tower.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/assets/leaning_tower.jpg -------------------------------------------------------------------------------- /gallery/assets/person1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/assets/person1.jpg -------------------------------------------------------------------------------- /gallery/assets/pottery.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/assets/pottery.jpg -------------------------------------------------------------------------------- /gallery/assets/repurposing_annotations_thumbnail.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/assets/repurposing_annotations_thumbnail.png -------------------------------------------------------------------------------- /gallery/assets/transforms_thumbnail.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/assets/transforms_thumbnail.png -------------------------------------------------------------------------------- /gallery/assets/visualization_utils_thumbnail2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/assets/visualization_utils_thumbnail2.png -------------------------------------------------------------------------------- /gallery/others/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/others/README.rst -------------------------------------------------------------------------------- /gallery/others/plot_optical_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/others/plot_optical_flow.py -------------------------------------------------------------------------------- /gallery/others/plot_repurposing_annotations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/others/plot_repurposing_annotations.py -------------------------------------------------------------------------------- /gallery/others/plot_scripted_tensor_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/others/plot_scripted_tensor_transforms.py -------------------------------------------------------------------------------- /gallery/others/plot_visualization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/others/plot_visualization_utils.py -------------------------------------------------------------------------------- /gallery/transforms/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/transforms/README.rst -------------------------------------------------------------------------------- /gallery/transforms/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/transforms/helpers.py -------------------------------------------------------------------------------- /gallery/transforms/plot_custom_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/transforms/plot_custom_transforms.py -------------------------------------------------------------------------------- /gallery/transforms/plot_custom_tv_tensors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/transforms/plot_custom_tv_tensors.py -------------------------------------------------------------------------------- /gallery/transforms/plot_cutmix_mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/transforms/plot_cutmix_mixup.py -------------------------------------------------------------------------------- /gallery/transforms/plot_keypoints_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/transforms/plot_keypoints_transforms.py -------------------------------------------------------------------------------- /gallery/transforms/plot_rotated_box_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/transforms/plot_rotated_box_transforms.py -------------------------------------------------------------------------------- /gallery/transforms/plot_transforms_e2e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/transforms/plot_transforms_e2e.py -------------------------------------------------------------------------------- /gallery/transforms/plot_transforms_getting_started.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/transforms/plot_transforms_getting_started.py -------------------------------------------------------------------------------- /gallery/transforms/plot_transforms_illustrations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/transforms/plot_transforms_illustrations.py -------------------------------------------------------------------------------- /gallery/transforms/plot_tv_tensors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/gallery/transforms/plot_tv_tensors.py -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/hubconf.py -------------------------------------------------------------------------------- /ios/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/CMakeLists.txt -------------------------------------------------------------------------------- /ios/LibTorchvision.podspec: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/LibTorchvision.podspec -------------------------------------------------------------------------------- /ios/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/README.md -------------------------------------------------------------------------------- /ios/VisionTestApp/VisionTestApp.xcodeproj/project.pbxproj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/VisionTestApp/VisionTestApp.xcodeproj/project.pbxproj -------------------------------------------------------------------------------- /ios/VisionTestApp/VisionTestApp/AppDelegate.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/VisionTestApp/VisionTestApp/AppDelegate.h -------------------------------------------------------------------------------- /ios/VisionTestApp/VisionTestApp/AppDelegate.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/VisionTestApp/VisionTestApp/AppDelegate.m -------------------------------------------------------------------------------- /ios/VisionTestApp/VisionTestApp/Base.lproj/Main.storyboard: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/VisionTestApp/VisionTestApp/Base.lproj/Main.storyboard -------------------------------------------------------------------------------- /ios/VisionTestApp/VisionTestApp/Info.plist: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/VisionTestApp/VisionTestApp/Info.plist -------------------------------------------------------------------------------- /ios/VisionTestApp/VisionTestApp/ModelRunner.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/VisionTestApp/VisionTestApp/ModelRunner.h -------------------------------------------------------------------------------- /ios/VisionTestApp/VisionTestApp/ModelRunner.mm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/VisionTestApp/VisionTestApp/ModelRunner.mm -------------------------------------------------------------------------------- /ios/VisionTestApp/VisionTestApp/ViewController.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/VisionTestApp/VisionTestApp/ViewController.h -------------------------------------------------------------------------------- /ios/VisionTestApp/VisionTestApp/ViewController.mm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/VisionTestApp/VisionTestApp/ViewController.mm -------------------------------------------------------------------------------- /ios/VisionTestApp/VisionTestApp/main.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/VisionTestApp/VisionTestApp/main.m -------------------------------------------------------------------------------- /ios/VisionTestApp/clean.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/VisionTestApp/clean.sh -------------------------------------------------------------------------------- /ios/VisionTestApp/make_assets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/VisionTestApp/make_assets.py -------------------------------------------------------------------------------- /ios/VisionTestApp/setup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/VisionTestApp/setup.sh -------------------------------------------------------------------------------- /ios/build_ios.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/ios/build_ios.sh -------------------------------------------------------------------------------- /maintainer_guide.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/maintainer_guide.md -------------------------------------------------------------------------------- /mypy.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/mypy.ini -------------------------------------------------------------------------------- /packaging/cut_release.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/packaging/cut_release.sh -------------------------------------------------------------------------------- /packaging/post_build_script.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/packaging/post_build_script.sh -------------------------------------------------------------------------------- /packaging/pre_build_script.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/packaging/pre_build_script.sh -------------------------------------------------------------------------------- /packaging/pre_build_script_arm64.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/packaging/pre_build_script_arm64.sh -------------------------------------------------------------------------------- /packaging/wheel/relocate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/packaging/wheel/relocate.py -------------------------------------------------------------------------------- /packaging/windows/internal/build_cmake.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/packaging/windows/internal/build_cmake.bat -------------------------------------------------------------------------------- /packaging/windows/internal/build_cpp_example.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/packaging/windows/internal/build_cpp_example.bat -------------------------------------------------------------------------------- /packaging/windows/internal/vc_env_helper.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/packaging/windows/internal/vc_env_helper.bat -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/pytest.ini -------------------------------------------------------------------------------- /references/classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/classification/README.md -------------------------------------------------------------------------------- /references/classification/presets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/classification/presets.py -------------------------------------------------------------------------------- /references/classification/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/classification/sampler.py -------------------------------------------------------------------------------- /references/classification/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/classification/train.py -------------------------------------------------------------------------------- /references/classification/train_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/classification/train_quantization.py -------------------------------------------------------------------------------- /references/classification/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/classification/transforms.py -------------------------------------------------------------------------------- /references/classification/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/classification/utils.py -------------------------------------------------------------------------------- /references/depth/stereo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/README.md -------------------------------------------------------------------------------- /references/depth/stereo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /references/depth/stereo/assets/Loss.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/assets/Loss.jpg -------------------------------------------------------------------------------- /references/depth/stereo/assets/disparity-domain-drift.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/assets/disparity-domain-drift.jpg -------------------------------------------------------------------------------- /references/depth/stereo/assets/gradient-norm-removal.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/assets/gradient-norm-removal.jpg -------------------------------------------------------------------------------- /references/depth/stereo/cascade_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/cascade_evaluation.py -------------------------------------------------------------------------------- /references/depth/stereo/parsing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/parsing.py -------------------------------------------------------------------------------- /references/depth/stereo/presets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/presets.py -------------------------------------------------------------------------------- /references/depth/stereo/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/train.py -------------------------------------------------------------------------------- /references/depth/stereo/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/transforms.py -------------------------------------------------------------------------------- /references/depth/stereo/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/utils/__init__.py -------------------------------------------------------------------------------- /references/depth/stereo/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/utils/distributed.py -------------------------------------------------------------------------------- /references/depth/stereo/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/utils/logger.py -------------------------------------------------------------------------------- /references/depth/stereo/utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/utils/losses.py -------------------------------------------------------------------------------- /references/depth/stereo/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/utils/metrics.py -------------------------------------------------------------------------------- /references/depth/stereo/utils/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/utils/norm.py -------------------------------------------------------------------------------- /references/depth/stereo/utils/padder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/utils/padder.py -------------------------------------------------------------------------------- /references/depth/stereo/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/depth/stereo/visualization.py -------------------------------------------------------------------------------- /references/detection/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/detection/README.md -------------------------------------------------------------------------------- /references/detection/coco_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/detection/coco_eval.py -------------------------------------------------------------------------------- /references/detection/coco_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/detection/coco_utils.py -------------------------------------------------------------------------------- /references/detection/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/detection/engine.py -------------------------------------------------------------------------------- /references/detection/group_by_aspect_ratio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/detection/group_by_aspect_ratio.py -------------------------------------------------------------------------------- /references/detection/presets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/detection/presets.py -------------------------------------------------------------------------------- /references/detection/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/detection/train.py -------------------------------------------------------------------------------- /references/detection/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/detection/transforms.py -------------------------------------------------------------------------------- /references/detection/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/detection/utils.py -------------------------------------------------------------------------------- /references/optical_flow/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/optical_flow/README.md -------------------------------------------------------------------------------- /references/optical_flow/presets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/optical_flow/presets.py -------------------------------------------------------------------------------- /references/optical_flow/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/optical_flow/train.py -------------------------------------------------------------------------------- /references/optical_flow/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/optical_flow/transforms.py -------------------------------------------------------------------------------- /references/optical_flow/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/optical_flow/utils.py -------------------------------------------------------------------------------- /references/segmentation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/segmentation/README.md -------------------------------------------------------------------------------- /references/segmentation/coco_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/segmentation/coco_utils.py -------------------------------------------------------------------------------- /references/segmentation/presets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/segmentation/presets.py -------------------------------------------------------------------------------- /references/segmentation/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/segmentation/train.py -------------------------------------------------------------------------------- /references/segmentation/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/segmentation/transforms.py -------------------------------------------------------------------------------- /references/segmentation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/segmentation/utils.py -------------------------------------------------------------------------------- /references/segmentation/v2_extras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/segmentation/v2_extras.py -------------------------------------------------------------------------------- /references/similarity/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/similarity/README.md -------------------------------------------------------------------------------- /references/similarity/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/similarity/loss.py -------------------------------------------------------------------------------- /references/similarity/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/similarity/model.py -------------------------------------------------------------------------------- /references/similarity/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/similarity/sampler.py -------------------------------------------------------------------------------- /references/similarity/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/similarity/test.py -------------------------------------------------------------------------------- /references/similarity/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/similarity/train.py -------------------------------------------------------------------------------- /references/video_classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/video_classification/README.md -------------------------------------------------------------------------------- /references/video_classification/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/video_classification/datasets.py -------------------------------------------------------------------------------- /references/video_classification/presets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/video_classification/presets.py -------------------------------------------------------------------------------- /references/video_classification/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/video_classification/train.py -------------------------------------------------------------------------------- /references/video_classification/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/video_classification/transforms.py -------------------------------------------------------------------------------- /references/video_classification/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/references/video_classification/utils.py -------------------------------------------------------------------------------- /release/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/release/README.md -------------------------------------------------------------------------------- /release/apply-release-changes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/release/apply-release-changes.py -------------------------------------------------------------------------------- /scripts/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/scripts/README.rst -------------------------------------------------------------------------------- /scripts/collect_model_urls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/scripts/collect_model_urls.py -------------------------------------------------------------------------------- /scripts/download_model_urls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/scripts/download_model_urls.py -------------------------------------------------------------------------------- /scripts/fbcode_to_main_sync.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/scripts/fbcode_to_main_sync.sh -------------------------------------------------------------------------------- /scripts/release_notes/classify_prs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/scripts/release_notes/classify_prs.py -------------------------------------------------------------------------------- /scripts/release_notes/retrieve_prs_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/scripts/release_notes/retrieve_prs_data.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/setup.py -------------------------------------------------------------------------------- /test/_utils_internal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/_utils_internal.py -------------------------------------------------------------------------------- /test/assets/damaged_jpeg/TensorFlow-LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/damaged_jpeg/TensorFlow-LICENSE -------------------------------------------------------------------------------- /test/assets/damaged_jpeg/bad_huffman.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/damaged_jpeg/bad_huffman.jpg -------------------------------------------------------------------------------- /test/assets/damaged_jpeg/corrupt.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/damaged_jpeg/corrupt.jpg -------------------------------------------------------------------------------- /test/assets/damaged_jpeg/corrupt34_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/damaged_jpeg/corrupt34_2.jpg -------------------------------------------------------------------------------- /test/assets/damaged_jpeg/corrupt34_3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/damaged_jpeg/corrupt34_3.jpg -------------------------------------------------------------------------------- /test/assets/damaged_jpeg/corrupt34_4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/damaged_jpeg/corrupt34_4.jpg -------------------------------------------------------------------------------- /test/assets/damaged_png/sigsegv.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/damaged_png/sigsegv.png -------------------------------------------------------------------------------- /test/assets/encode_jpeg/grace_hopper_517x606.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/encode_jpeg/grace_hopper_517x606.jpg -------------------------------------------------------------------------------- /test/assets/expected_flow.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/expected_flow.pt -------------------------------------------------------------------------------- /test/assets/fakedata/draw_boxes_different_label_colors.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/draw_boxes_different_label_colors.png -------------------------------------------------------------------------------- /test/assets/fakedata/draw_boxes_util.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/draw_boxes_util.png -------------------------------------------------------------------------------- /test/assets/fakedata/draw_boxes_vanilla.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/draw_boxes_vanilla.png -------------------------------------------------------------------------------- /test/assets/fakedata/draw_keypoint_vanilla.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/draw_keypoint_vanilla.png -------------------------------------------------------------------------------- /test/assets/fakedata/draw_keypoints_visibility.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/draw_keypoints_visibility.png -------------------------------------------------------------------------------- /test/assets/fakedata/draw_rotated_boxes.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/draw_rotated_boxes.png -------------------------------------------------------------------------------- /test/assets/fakedata/draw_rotated_boxes_fill.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/draw_rotated_boxes_fill.png -------------------------------------------------------------------------------- /test/assets/fakedata/imagefolder/a/a1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/imagefolder/a/a1.png -------------------------------------------------------------------------------- /test/assets/fakedata/imagefolder/a/a2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/imagefolder/a/a2.png -------------------------------------------------------------------------------- /test/assets/fakedata/imagefolder/a/a3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/imagefolder/a/a3.png -------------------------------------------------------------------------------- /test/assets/fakedata/imagefolder/b/b1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/imagefolder/b/b1.png -------------------------------------------------------------------------------- /test/assets/fakedata/imagefolder/b/b2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/imagefolder/b/b2.png -------------------------------------------------------------------------------- /test/assets/fakedata/imagefolder/b/b3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/imagefolder/b/b3.png -------------------------------------------------------------------------------- /test/assets/fakedata/imagefolder/b/b4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/imagefolder/b/b4.png -------------------------------------------------------------------------------- /test/assets/fakedata/logos/cmyk_pytorch.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/logos/cmyk_pytorch.jpg -------------------------------------------------------------------------------- /test/assets/fakedata/logos/gray_pytorch.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/logos/gray_pytorch.jpg -------------------------------------------------------------------------------- /test/assets/fakedata/logos/gray_pytorch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/logos/gray_pytorch.png -------------------------------------------------------------------------------- /test/assets/fakedata/logos/grayalpha_pytorch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/logos/grayalpha_pytorch.png -------------------------------------------------------------------------------- /test/assets/fakedata/logos/palette_pytorch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/logos/palette_pytorch.png -------------------------------------------------------------------------------- /test/assets/fakedata/logos/rgb_pytorch.avif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/logos/rgb_pytorch.avif -------------------------------------------------------------------------------- /test/assets/fakedata/logos/rgb_pytorch.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/logos/rgb_pytorch.jpg -------------------------------------------------------------------------------- /test/assets/fakedata/logos/rgb_pytorch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/logos/rgb_pytorch.png -------------------------------------------------------------------------------- /test/assets/fakedata/logos/rgb_pytorch.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/logos/rgb_pytorch.webp -------------------------------------------------------------------------------- /test/assets/fakedata/logos/rgb_pytorch16.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/logos/rgb_pytorch16.png -------------------------------------------------------------------------------- /test/assets/fakedata/logos/rgbalpha_pytorch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/logos/rgbalpha_pytorch.png -------------------------------------------------------------------------------- /test/assets/fakedata/logos/rgbalpha_pytorch16.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/fakedata/logos/rgbalpha_pytorch16.png -------------------------------------------------------------------------------- /test/assets/gaussian_blur_opencv_results.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/gaussian_blur_opencv_results.pt -------------------------------------------------------------------------------- /test/assets/interlaced_png/wizard_low-interlaced.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/interlaced_png/wizard_low-interlaced.png -------------------------------------------------------------------------------- /test/assets/interlaced_png/wizard_low.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/interlaced_png/wizard_low.png -------------------------------------------------------------------------------- /test/assets/labeled_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/labeled_image.png -------------------------------------------------------------------------------- /test/assets/masks.tiff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/masks.tiff -------------------------------------------------------------------------------- /test/assets/toosmall_png/heapbof.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/toosmall_png/heapbof.png -------------------------------------------------------------------------------- /test/assets/videos/R6llTwEh07w.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/videos/R6llTwEh07w.mp4 -------------------------------------------------------------------------------- /test/assets/videos/RATRACE_wave_f_nm_np1_fr_goo_37.avi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/videos/RATRACE_wave_f_nm_np1_fr_goo_37.avi -------------------------------------------------------------------------------- /test/assets/videos/README: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/videos/README -------------------------------------------------------------------------------- /test/assets/videos/SOX5yA1l24A.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/videos/SOX5yA1l24A.mp4 -------------------------------------------------------------------------------- /test/assets/videos/TrumanShow_wave_f_nm_np1_fr_med_26.avi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/videos/TrumanShow_wave_f_nm_np1_fr_med_26.avi -------------------------------------------------------------------------------- /test/assets/videos/WUzgd7C1pWA.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/videos/WUzgd7C1pWA.mp4 -------------------------------------------------------------------------------- /test/assets/videos/v_SoccerJuggling_g23_c01.avi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/videos/v_SoccerJuggling_g23_c01.avi -------------------------------------------------------------------------------- /test/assets/videos/v_SoccerJuggling_g24_c01.avi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/assets/videos/v_SoccerJuggling_g24_c01.avi -------------------------------------------------------------------------------- /test/builtin_dataset_mocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/builtin_dataset_mocks.py -------------------------------------------------------------------------------- /test/common_extended_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/common_extended_utils.py -------------------------------------------------------------------------------- /test/common_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/common_utils.py -------------------------------------------------------------------------------- /test/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/conftest.py -------------------------------------------------------------------------------- /test/cpp/test_custom_operators.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/cpp/test_custom_operators.cpp -------------------------------------------------------------------------------- /test/datasets_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/datasets_utils.py -------------------------------------------------------------------------------- /test/expect/ModelTester.test_alexnet_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_alexnet_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_convnext_base_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_convnext_base_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_convnext_large_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_convnext_large_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_convnext_small_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_convnext_small_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_convnext_tiny_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_convnext_tiny_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_crestereo_base_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_crestereo_base_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_deeplabv3_resnet101_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_deeplabv3_resnet101_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_deeplabv3_resnet50_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_deeplabv3_resnet50_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_densenet121_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_densenet121_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_densenet161_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_densenet161_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_densenet169_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_densenet169_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_densenet201_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_densenet201_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_efficientnet_b0_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_efficientnet_b0_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_efficientnet_b1_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_efficientnet_b1_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_efficientnet_b2_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_efficientnet_b2_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_efficientnet_b3_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_efficientnet_b3_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_efficientnet_b4_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_efficientnet_b4_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_efficientnet_b5_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_efficientnet_b5_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_efficientnet_b6_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_efficientnet_b6_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_efficientnet_b7_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_efficientnet_b7_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_efficientnet_v2_l_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_efficientnet_v2_l_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_efficientnet_v2_m_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_efficientnet_v2_m_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_efficientnet_v2_s_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_efficientnet_v2_s_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_fcn_resnet101_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_fcn_resnet101_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_fcn_resnet50_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_fcn_resnet50_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_fcos_resnet50_fpn_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_fcos_resnet50_fpn_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_googlenet_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_googlenet_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_googlenet_quantized_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_googlenet_quantized_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_inception_v3_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_inception_v3_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_maxvit_t_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_maxvit_t_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_mc3_18_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_mc3_18_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_mnasnet0_5_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_mnasnet0_5_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_mnasnet0_75_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_mnasnet0_75_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_mnasnet1_0_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_mnasnet1_0_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_mnasnet1_3_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_mnasnet1_3_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_mobilenet_v2_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_mobilenet_v2_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_mobilenet_v3_large_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_mobilenet_v3_large_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_mobilenet_v3_small_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_mobilenet_v3_small_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_mvit_v1_b_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_mvit_v1_b_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_mvit_v2_s_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_mvit_v2_s_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_r2plus1d_18_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_r2plus1d_18_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_r3d_18_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_r3d_18_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_raft_large_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_raft_large_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_raft_small_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_raft_small_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_raft_stereo_base_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_raft_stereo_base_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_regnet_x_16gf_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_regnet_x_16gf_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_regnet_x_1_6gf_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_regnet_x_1_6gf_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_regnet_x_32gf_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_regnet_x_32gf_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_regnet_x_3_2gf_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_regnet_x_3_2gf_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_regnet_x_400mf_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_regnet_x_400mf_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_regnet_x_800mf_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_regnet_x_800mf_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_regnet_x_8gf_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_regnet_x_8gf_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_regnet_y_128gf_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_regnet_y_128gf_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_regnet_y_16gf_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_regnet_y_16gf_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_regnet_y_1_6gf_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_regnet_y_1_6gf_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_regnet_y_32gf_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_regnet_y_32gf_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_regnet_y_3_2gf_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_regnet_y_3_2gf_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_regnet_y_400mf_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_regnet_y_400mf_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_regnet_y_800mf_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_regnet_y_800mf_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_regnet_y_8gf_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_regnet_y_8gf_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_resnet101_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_resnet101_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_resnet152_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_resnet152_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_resnet18_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_resnet18_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_resnet18_quantized_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_resnet18_quantized_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_resnet34_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_resnet34_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_resnet50_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_resnet50_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_resnet50_quantized_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_resnet50_quantized_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_resnext101_32x8d_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_resnext101_32x8d_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_resnext101_64x4d_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_resnext101_64x4d_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_resnext50_32x4d_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_resnext50_32x4d_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_s3d_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_s3d_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_shufflenet_v2_x0_5_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_shufflenet_v2_x0_5_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_shufflenet_v2_x1_0_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_shufflenet_v2_x1_0_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_shufflenet_v2_x1_5_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_shufflenet_v2_x1_5_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_shufflenet_v2_x2_0_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_shufflenet_v2_x2_0_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_squeezenet1_0_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_squeezenet1_0_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_squeezenet1_1_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_squeezenet1_1_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_ssd300_vgg16_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_ssd300_vgg16_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_swin3d_b_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_swin3d_b_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_swin3d_s_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_swin3d_s_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_swin3d_t_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_swin3d_t_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_swin_b_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_swin_b_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_swin_s_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_swin_s_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_swin_t_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_swin_t_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_swin_v2_b_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_swin_v2_b_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_swin_v2_s_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_swin_v2_s_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_swin_v2_t_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_swin_v2_t_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_vgg11_bn_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_vgg11_bn_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_vgg11_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_vgg11_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_vgg13_bn_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_vgg13_bn_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_vgg13_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_vgg13_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_vgg16_bn_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_vgg16_bn_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_vgg16_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_vgg16_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_vgg19_bn_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_vgg19_bn_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_vgg19_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_vgg19_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_vit_b_16_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_vit_b_16_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_vit_b_32_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_vit_b_32_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_vit_h_14_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_vit_h_14_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_vit_l_16_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_vit_l_16_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_vit_l_32_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_vit_l_32_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_vitc_b_16_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_vitc_b_16_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_wide_resnet101_2_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_wide_resnet101_2_expect.pkl -------------------------------------------------------------------------------- /test/expect/ModelTester.test_wide_resnet50_2_expect.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/expect/ModelTester.test_wide_resnet50_2_expect.pkl -------------------------------------------------------------------------------- /test/optests_failures_dict.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/optests_failures_dict.json -------------------------------------------------------------------------------- /test/preprocess-bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/preprocess-bench.py -------------------------------------------------------------------------------- /test/smoke_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/smoke_test.py -------------------------------------------------------------------------------- /test/test_architecture_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_architecture_ops.py -------------------------------------------------------------------------------- /test/test_backbone_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_backbone_utils.py -------------------------------------------------------------------------------- /test/test_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_datasets.py -------------------------------------------------------------------------------- /test/test_datasets_download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_datasets_download.py -------------------------------------------------------------------------------- /test/test_datasets_samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_datasets_samplers.py -------------------------------------------------------------------------------- /test/test_datasets_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_datasets_utils.py -------------------------------------------------------------------------------- /test/test_datasets_video_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_datasets_video_utils.py -------------------------------------------------------------------------------- /test/test_datasets_video_utils_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_datasets_video_utils_opt.py -------------------------------------------------------------------------------- /test/test_extended_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_extended_models.py -------------------------------------------------------------------------------- /test/test_functional_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_functional_tensor.py -------------------------------------------------------------------------------- /test/test_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_image.py -------------------------------------------------------------------------------- /test/test_internal_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_internal_utils.py -------------------------------------------------------------------------------- /test/test_internet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_internet.py -------------------------------------------------------------------------------- /test/test_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_io.py -------------------------------------------------------------------------------- /test/test_io_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_io_opt.py -------------------------------------------------------------------------------- /test/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_models.py -------------------------------------------------------------------------------- /test/test_models_detection_anchor_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_models_detection_anchor_utils.py -------------------------------------------------------------------------------- /test/test_models_detection_negative_samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_models_detection_negative_samples.py -------------------------------------------------------------------------------- /test/test_models_detection_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_models_detection_utils.py -------------------------------------------------------------------------------- /test/test_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_onnx.py -------------------------------------------------------------------------------- /test/test_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_ops.py -------------------------------------------------------------------------------- /test/test_prototype_datasets_builtin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_prototype_datasets_builtin.py -------------------------------------------------------------------------------- /test/test_prototype_datasets_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_prototype_datasets_utils.py -------------------------------------------------------------------------------- /test/test_prototype_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_prototype_models.py -------------------------------------------------------------------------------- /test/test_prototype_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_prototype_transforms.py -------------------------------------------------------------------------------- /test/test_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_transforms.py -------------------------------------------------------------------------------- /test/test_transforms_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_transforms_tensor.py -------------------------------------------------------------------------------- /test/test_transforms_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_transforms_v2.py -------------------------------------------------------------------------------- /test/test_transforms_v2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_transforms_v2_utils.py -------------------------------------------------------------------------------- /test/test_transforms_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_transforms_video.py -------------------------------------------------------------------------------- /test/test_tv_tensors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_tv_tensors.py -------------------------------------------------------------------------------- /test/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_utils.py -------------------------------------------------------------------------------- /test/test_video_gpu_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_video_gpu_decoder.py -------------------------------------------------------------------------------- /test/test_video_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_video_reader.py -------------------------------------------------------------------------------- /test/test_videoapi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/test/test_videoapi.py -------------------------------------------------------------------------------- /torchvision/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/__init__.py -------------------------------------------------------------------------------- /torchvision/_internally_replaced_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/_internally_replaced_utils.py -------------------------------------------------------------------------------- /torchvision/_meta_registrations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/_meta_registrations.py -------------------------------------------------------------------------------- /torchvision/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/_utils.py -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/audio_sampler.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/audio_sampler.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/audio_sampler.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/audio_sampler.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/audio_stream.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/audio_stream.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/audio_stream.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/audio_stream.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/cc_stream.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/cc_stream.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/cc_stream.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/cc_stream.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/decoder.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/decoder.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/decoder.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/decoder.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/defs.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/defs.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/gpu/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/gpu/README.rst -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/gpu/decoder.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/gpu/decoder.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/gpu/decoder.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/gpu/decoder.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/gpu/demuxer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/gpu/demuxer.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/gpu/gpu_decoder.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/gpu/gpu_decoder.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/gpu/gpu_decoder.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/gpu/gpu_decoder.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/memory_buffer.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/memory_buffer.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/memory_buffer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/memory_buffer.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/seekable_buffer.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/seekable_buffer.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/seekable_buffer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/seekable_buffer.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/stream.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/stream.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/stream.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/stream.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/subtitle_sampler.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/subtitle_sampler.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/subtitle_sampler.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/subtitle_sampler.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/subtitle_stream.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/subtitle_stream.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/subtitle_stream.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/subtitle_stream.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/sync_decoder.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/sync_decoder.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/sync_decoder.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/sync_decoder.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/sync_decoder_test.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/sync_decoder_test.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/time_keeper.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/time_keeper.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/time_keeper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/time_keeper.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/util.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/util.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/util.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/util_test.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/util_test.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/video_sampler.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/video_sampler.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/video_sampler.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/video_sampler.h -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/video_stream.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/video_stream.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/decoder/video_stream.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/decoder/video_stream.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/common.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/common.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/image/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/common.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/common_jpeg.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/common_jpeg.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/common_jpeg.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/common_jpeg.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/common_png.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/common_png.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/decode_gif.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/decode_gif.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/decode_gif.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/decode_gif.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/decode_image.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/decode_image.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/decode_image.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/decode_image.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/decode_jpeg.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/decode_jpeg.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/decode_jpeg.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/decode_jpeg.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/decode_png.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/decode_png.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/decode_png.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/decode_png.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/decode_webp.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/decode_webp.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/decode_webp.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/decode_webp.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/encode_jpeg.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/encode_jpeg.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/encode_jpeg.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/encode_jpeg.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/encode_png.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/encode_png.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/encode_png.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/encode_png.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/exif.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/exif.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/giflib/README: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/giflib/README -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/giflib/dgif_lib.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/giflib/dgif_lib.c -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/giflib/gif_hash.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/giflib/gif_hash.c -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/giflib/gif_hash.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/giflib/gif_hash.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/giflib/gif_lib.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/giflib/gif_lib.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/giflib/gif_lib_private.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/giflib/gif_lib_private.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/giflib/gifalloc.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/giflib/gifalloc.c -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/giflib/openbsd-reallocarray.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/giflib/openbsd-reallocarray.c -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/read_write_file.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/read_write_file.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cpu/read_write_file.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cpu/read_write_file.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cuda/decode_jpegs_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cuda/decode_jpegs_cuda.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cuda/decode_jpegs_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cuda/decode_jpegs_cuda.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cuda/encode_jpegs_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cuda/encode_jpegs_cuda.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/image/cuda/encode_jpegs_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/cuda/encode_jpegs_cuda.h -------------------------------------------------------------------------------- /torchvision/csrc/io/image/image.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/image.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/image/image.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/image/image.h -------------------------------------------------------------------------------- /torchvision/csrc/io/video/video.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/video/video.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/video/video.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/video/video.h -------------------------------------------------------------------------------- /torchvision/csrc/io/video_reader/video_reader.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/video_reader/video_reader.cpp -------------------------------------------------------------------------------- /torchvision/csrc/io/video_reader/video_reader.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/io/video_reader/video_reader.h -------------------------------------------------------------------------------- /torchvision/csrc/macros.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/macros.h -------------------------------------------------------------------------------- /torchvision/csrc/ops/autocast/deform_conv2d_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/autocast/deform_conv2d_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/autocast/nms_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/autocast/nms_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/autocast/ps_roi_align_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/autocast/ps_roi_align_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/autocast/ps_roi_pool_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/autocast/ps_roi_pool_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/autocast/roi_align_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/autocast/roi_align_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/autocast/roi_pool_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/autocast/roi_pool_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/autograd/deform_conv2d_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/autograd/deform_conv2d_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/autograd/ps_roi_align_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/autograd/ps_roi_align_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/autograd/ps_roi_pool_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/autograd/ps_roi_pool_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/autograd/roi_align_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/autograd/roi_align_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/autograd/roi_pool_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/autograd/roi_pool_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/cpu/deform_conv2d_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/cpu/deform_conv2d_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/cpu/nms_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/cpu/nms_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/cpu/ps_roi_align_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/cpu/ps_roi_align_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/cpu/ps_roi_pool_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/cpu/ps_roi_pool_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/cpu/roi_align_common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/cpu/roi_align_common.h -------------------------------------------------------------------------------- /torchvision/csrc/ops/cpu/roi_align_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/cpu/roi_align_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/cpu/roi_pool_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/cpu/roi_pool_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/cuda/cuda_helpers.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/cuda/cuda_helpers.h -------------------------------------------------------------------------------- /torchvision/csrc/ops/cuda/deform_conv2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/cuda/deform_conv2d_kernel.cu -------------------------------------------------------------------------------- /torchvision/csrc/ops/cuda/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/cuda/nms_kernel.cu -------------------------------------------------------------------------------- /torchvision/csrc/ops/cuda/ps_roi_align_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/cuda/ps_roi_align_kernel.cu -------------------------------------------------------------------------------- /torchvision/csrc/ops/cuda/ps_roi_pool_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/cuda/ps_roi_pool_kernel.cu -------------------------------------------------------------------------------- /torchvision/csrc/ops/cuda/roi_align_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/cuda/roi_align_kernel.cu -------------------------------------------------------------------------------- /torchvision/csrc/ops/cuda/roi_pool_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/cuda/roi_pool_kernel.cu -------------------------------------------------------------------------------- /torchvision/csrc/ops/deform_conv2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/deform_conv2d.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/deform_conv2d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/deform_conv2d.h -------------------------------------------------------------------------------- /torchvision/csrc/ops/mps/deform_conv2d_kernel.mm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/mps/deform_conv2d_kernel.mm -------------------------------------------------------------------------------- /torchvision/csrc/ops/mps/mps_helpers.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/mps/mps_helpers.h -------------------------------------------------------------------------------- /torchvision/csrc/ops/mps/mps_kernels.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/mps/mps_kernels.h -------------------------------------------------------------------------------- /torchvision/csrc/ops/mps/nms_kernel.mm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/mps/nms_kernel.mm -------------------------------------------------------------------------------- /torchvision/csrc/ops/mps/ps_roi_align_kernel.mm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/mps/ps_roi_align_kernel.mm -------------------------------------------------------------------------------- /torchvision/csrc/ops/mps/ps_roi_pool_kernel.mm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/mps/ps_roi_pool_kernel.mm -------------------------------------------------------------------------------- /torchvision/csrc/ops/mps/roi_align_kernel.mm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/mps/roi_align_kernel.mm -------------------------------------------------------------------------------- /torchvision/csrc/ops/mps/roi_pool_kernel.mm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/mps/roi_pool_kernel.mm -------------------------------------------------------------------------------- /torchvision/csrc/ops/nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/nms.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/nms.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/nms.h -------------------------------------------------------------------------------- /torchvision/csrc/ops/ops.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/ops.h -------------------------------------------------------------------------------- /torchvision/csrc/ops/ps_roi_align.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/ps_roi_align.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/ps_roi_align.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/ps_roi_align.h -------------------------------------------------------------------------------- /torchvision/csrc/ops/ps_roi_pool.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/ps_roi_pool.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/ps_roi_pool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/ps_roi_pool.h -------------------------------------------------------------------------------- /torchvision/csrc/ops/quantized/cpu/qnms_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/quantized/cpu/qnms_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/quantized/cpu/qroi_align_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/quantized/cpu/qroi_align_kernel.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/roi_align.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/roi_align.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/roi_align.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/roi_align.h -------------------------------------------------------------------------------- /torchvision/csrc/ops/roi_pool.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/roi_pool.cpp -------------------------------------------------------------------------------- /torchvision/csrc/ops/roi_pool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/ops/roi_pool.h -------------------------------------------------------------------------------- /torchvision/csrc/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/vision.cpp -------------------------------------------------------------------------------- /torchvision/csrc/vision.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/csrc/vision.h -------------------------------------------------------------------------------- /torchvision/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/__init__.py -------------------------------------------------------------------------------- /torchvision/datasets/_optical_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/_optical_flow.py -------------------------------------------------------------------------------- /torchvision/datasets/_stereo_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/_stereo_matching.py -------------------------------------------------------------------------------- /torchvision/datasets/caltech.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/caltech.py -------------------------------------------------------------------------------- /torchvision/datasets/celeba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/celeba.py -------------------------------------------------------------------------------- /torchvision/datasets/cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/cifar.py -------------------------------------------------------------------------------- /torchvision/datasets/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/cityscapes.py -------------------------------------------------------------------------------- /torchvision/datasets/clevr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/clevr.py -------------------------------------------------------------------------------- /torchvision/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/coco.py -------------------------------------------------------------------------------- /torchvision/datasets/country211.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/country211.py -------------------------------------------------------------------------------- /torchvision/datasets/dtd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/dtd.py -------------------------------------------------------------------------------- /torchvision/datasets/eurosat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/eurosat.py -------------------------------------------------------------------------------- /torchvision/datasets/fakedata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/fakedata.py -------------------------------------------------------------------------------- /torchvision/datasets/fer2013.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/fer2013.py -------------------------------------------------------------------------------- /torchvision/datasets/fgvc_aircraft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/fgvc_aircraft.py -------------------------------------------------------------------------------- /torchvision/datasets/flickr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/flickr.py -------------------------------------------------------------------------------- /torchvision/datasets/flowers102.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/flowers102.py -------------------------------------------------------------------------------- /torchvision/datasets/folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/folder.py -------------------------------------------------------------------------------- /torchvision/datasets/food101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/food101.py -------------------------------------------------------------------------------- /torchvision/datasets/gtsrb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/gtsrb.py -------------------------------------------------------------------------------- /torchvision/datasets/hmdb51.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/hmdb51.py -------------------------------------------------------------------------------- /torchvision/datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/imagenet.py -------------------------------------------------------------------------------- /torchvision/datasets/imagenette.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/imagenette.py -------------------------------------------------------------------------------- /torchvision/datasets/inaturalist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/inaturalist.py -------------------------------------------------------------------------------- /torchvision/datasets/kinetics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/kinetics.py -------------------------------------------------------------------------------- /torchvision/datasets/kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/kitti.py -------------------------------------------------------------------------------- /torchvision/datasets/lfw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/lfw.py -------------------------------------------------------------------------------- /torchvision/datasets/lsun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/lsun.py -------------------------------------------------------------------------------- /torchvision/datasets/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/mnist.py -------------------------------------------------------------------------------- /torchvision/datasets/moving_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/moving_mnist.py -------------------------------------------------------------------------------- /torchvision/datasets/omniglot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/omniglot.py -------------------------------------------------------------------------------- /torchvision/datasets/oxford_iiit_pet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/oxford_iiit_pet.py -------------------------------------------------------------------------------- /torchvision/datasets/pcam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/pcam.py -------------------------------------------------------------------------------- /torchvision/datasets/phototour.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/phototour.py -------------------------------------------------------------------------------- /torchvision/datasets/places365.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/places365.py -------------------------------------------------------------------------------- /torchvision/datasets/rendered_sst2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/rendered_sst2.py -------------------------------------------------------------------------------- /torchvision/datasets/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/samplers/__init__.py -------------------------------------------------------------------------------- /torchvision/datasets/samplers/clip_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/samplers/clip_sampler.py -------------------------------------------------------------------------------- /torchvision/datasets/sbd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/sbd.py -------------------------------------------------------------------------------- /torchvision/datasets/sbu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/sbu.py -------------------------------------------------------------------------------- /torchvision/datasets/semeion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/semeion.py -------------------------------------------------------------------------------- /torchvision/datasets/stanford_cars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/stanford_cars.py -------------------------------------------------------------------------------- /torchvision/datasets/stl10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/stl10.py -------------------------------------------------------------------------------- /torchvision/datasets/sun397.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/sun397.py -------------------------------------------------------------------------------- /torchvision/datasets/svhn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/svhn.py -------------------------------------------------------------------------------- /torchvision/datasets/ucf101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/ucf101.py -------------------------------------------------------------------------------- /torchvision/datasets/usps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/usps.py -------------------------------------------------------------------------------- /torchvision/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/utils.py -------------------------------------------------------------------------------- /torchvision/datasets/video_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/video_utils.py -------------------------------------------------------------------------------- /torchvision/datasets/vision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/vision.py -------------------------------------------------------------------------------- /torchvision/datasets/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/voc.py -------------------------------------------------------------------------------- /torchvision/datasets/widerface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/datasets/widerface.py -------------------------------------------------------------------------------- /torchvision/extension.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/extension.py -------------------------------------------------------------------------------- /torchvision/io/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/io/__init__.py -------------------------------------------------------------------------------- /torchvision/io/_load_gpu_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/io/_load_gpu_decoder.py -------------------------------------------------------------------------------- /torchvision/io/_video_deprecation_warning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/io/_video_deprecation_warning.py -------------------------------------------------------------------------------- /torchvision/io/_video_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/io/_video_opt.py -------------------------------------------------------------------------------- /torchvision/io/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/io/image.py -------------------------------------------------------------------------------- /torchvision/io/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/io/video.py -------------------------------------------------------------------------------- /torchvision/io/video_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/io/video_reader.py -------------------------------------------------------------------------------- /torchvision/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/__init__.py -------------------------------------------------------------------------------- /torchvision/models/_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/_api.py -------------------------------------------------------------------------------- /torchvision/models/_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/_meta.py -------------------------------------------------------------------------------- /torchvision/models/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/_utils.py -------------------------------------------------------------------------------- /torchvision/models/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/alexnet.py -------------------------------------------------------------------------------- /torchvision/models/convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/convnext.py -------------------------------------------------------------------------------- /torchvision/models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/densenet.py -------------------------------------------------------------------------------- /torchvision/models/detection/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/detection/__init__.py -------------------------------------------------------------------------------- /torchvision/models/detection/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/detection/_utils.py -------------------------------------------------------------------------------- /torchvision/models/detection/anchor_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/detection/anchor_utils.py -------------------------------------------------------------------------------- /torchvision/models/detection/backbone_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/detection/backbone_utils.py -------------------------------------------------------------------------------- /torchvision/models/detection/faster_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/detection/faster_rcnn.py -------------------------------------------------------------------------------- /torchvision/models/detection/fcos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/detection/fcos.py -------------------------------------------------------------------------------- /torchvision/models/detection/generalized_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/detection/generalized_rcnn.py -------------------------------------------------------------------------------- /torchvision/models/detection/image_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/detection/image_list.py -------------------------------------------------------------------------------- /torchvision/models/detection/keypoint_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/detection/keypoint_rcnn.py -------------------------------------------------------------------------------- /torchvision/models/detection/mask_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/detection/mask_rcnn.py -------------------------------------------------------------------------------- /torchvision/models/detection/retinanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/detection/retinanet.py -------------------------------------------------------------------------------- /torchvision/models/detection/roi_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/detection/roi_heads.py -------------------------------------------------------------------------------- /torchvision/models/detection/rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/detection/rpn.py -------------------------------------------------------------------------------- /torchvision/models/detection/ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/detection/ssd.py -------------------------------------------------------------------------------- /torchvision/models/detection/ssdlite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/detection/ssdlite.py -------------------------------------------------------------------------------- /torchvision/models/detection/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/detection/transform.py -------------------------------------------------------------------------------- /torchvision/models/efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/efficientnet.py -------------------------------------------------------------------------------- /torchvision/models/feature_extraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/feature_extraction.py -------------------------------------------------------------------------------- /torchvision/models/googlenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/googlenet.py -------------------------------------------------------------------------------- /torchvision/models/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/inception.py -------------------------------------------------------------------------------- /torchvision/models/maxvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/maxvit.py -------------------------------------------------------------------------------- /torchvision/models/mnasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/mnasnet.py -------------------------------------------------------------------------------- /torchvision/models/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/mobilenet.py -------------------------------------------------------------------------------- /torchvision/models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/mobilenetv2.py -------------------------------------------------------------------------------- /torchvision/models/mobilenetv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/mobilenetv3.py -------------------------------------------------------------------------------- /torchvision/models/optical_flow/__init__.py: -------------------------------------------------------------------------------- 1 | from .raft import * 2 | -------------------------------------------------------------------------------- /torchvision/models/optical_flow/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/optical_flow/_utils.py -------------------------------------------------------------------------------- /torchvision/models/optical_flow/raft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/optical_flow/raft.py -------------------------------------------------------------------------------- /torchvision/models/quantization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/quantization/__init__.py -------------------------------------------------------------------------------- /torchvision/models/quantization/googlenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/quantization/googlenet.py -------------------------------------------------------------------------------- /torchvision/models/quantization/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/quantization/inception.py -------------------------------------------------------------------------------- /torchvision/models/quantization/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/quantization/mobilenet.py -------------------------------------------------------------------------------- /torchvision/models/quantization/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/quantization/mobilenetv2.py -------------------------------------------------------------------------------- /torchvision/models/quantization/mobilenetv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/quantization/mobilenetv3.py -------------------------------------------------------------------------------- /torchvision/models/quantization/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/quantization/resnet.py -------------------------------------------------------------------------------- /torchvision/models/quantization/shufflenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/quantization/shufflenetv2.py -------------------------------------------------------------------------------- /torchvision/models/quantization/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/quantization/utils.py -------------------------------------------------------------------------------- /torchvision/models/regnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/regnet.py -------------------------------------------------------------------------------- /torchvision/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/resnet.py -------------------------------------------------------------------------------- /torchvision/models/segmentation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/segmentation/__init__.py -------------------------------------------------------------------------------- /torchvision/models/segmentation/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/segmentation/_utils.py -------------------------------------------------------------------------------- /torchvision/models/segmentation/deeplabv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/segmentation/deeplabv3.py -------------------------------------------------------------------------------- /torchvision/models/segmentation/fcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/segmentation/fcn.py -------------------------------------------------------------------------------- /torchvision/models/segmentation/lraspp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/segmentation/lraspp.py -------------------------------------------------------------------------------- /torchvision/models/shufflenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/shufflenetv2.py -------------------------------------------------------------------------------- /torchvision/models/squeezenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/squeezenet.py -------------------------------------------------------------------------------- /torchvision/models/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/swin_transformer.py -------------------------------------------------------------------------------- /torchvision/models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/vgg.py -------------------------------------------------------------------------------- /torchvision/models/video/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/video/__init__.py -------------------------------------------------------------------------------- /torchvision/models/video/mvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/video/mvit.py -------------------------------------------------------------------------------- /torchvision/models/video/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/video/resnet.py -------------------------------------------------------------------------------- /torchvision/models/video/s3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/video/s3d.py -------------------------------------------------------------------------------- /torchvision/models/video/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/video/swin_transformer.py -------------------------------------------------------------------------------- /torchvision/models/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/models/vision_transformer.py -------------------------------------------------------------------------------- /torchvision/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/__init__.py -------------------------------------------------------------------------------- /torchvision/ops/_box_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/_box_convert.py -------------------------------------------------------------------------------- /torchvision/ops/_register_onnx_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/_register_onnx_ops.py -------------------------------------------------------------------------------- /torchvision/ops/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/_utils.py -------------------------------------------------------------------------------- /torchvision/ops/boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/boxes.py -------------------------------------------------------------------------------- /torchvision/ops/ciou_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/ciou_loss.py -------------------------------------------------------------------------------- /torchvision/ops/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/deform_conv.py -------------------------------------------------------------------------------- /torchvision/ops/diou_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/diou_loss.py -------------------------------------------------------------------------------- /torchvision/ops/drop_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/drop_block.py -------------------------------------------------------------------------------- /torchvision/ops/feature_pyramid_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/feature_pyramid_network.py -------------------------------------------------------------------------------- /torchvision/ops/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/focal_loss.py -------------------------------------------------------------------------------- /torchvision/ops/giou_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/giou_loss.py -------------------------------------------------------------------------------- /torchvision/ops/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/misc.py -------------------------------------------------------------------------------- /torchvision/ops/poolers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/poolers.py -------------------------------------------------------------------------------- /torchvision/ops/ps_roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/ps_roi_align.py -------------------------------------------------------------------------------- /torchvision/ops/ps_roi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/ps_roi_pool.py -------------------------------------------------------------------------------- /torchvision/ops/roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/roi_align.py -------------------------------------------------------------------------------- /torchvision/ops/roi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/roi_pool.py -------------------------------------------------------------------------------- /torchvision/ops/stochastic_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/ops/stochastic_depth.py -------------------------------------------------------------------------------- /torchvision/prototype/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/__init__.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/README.md -------------------------------------------------------------------------------- /torchvision/prototype/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/__init__.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_api.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/README.md -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/__init__.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/caltech.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/caltech.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/celeba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/celeba.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/cifar.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/clevr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/clevr.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/coco.categories: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/coco.categories -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/coco.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/country211.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/country211.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/cub200.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/cub200.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/dtd.categories: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/dtd.categories -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/dtd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/dtd.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/eurosat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/eurosat.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/fer2013.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/fer2013.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/food101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/food101.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/gtsrb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/gtsrb.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/imagenet.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/mnist.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/pcam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/pcam.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/sbd.categories: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/sbd.categories -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/sbd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/sbd.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/semeion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/semeion.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/stanford_cars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/stanford_cars.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/svhn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/svhn.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/usps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/usps.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/voc.categories: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/voc.categories -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_builtin/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_builtin/voc.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_folder.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/_home.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/_home.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/benchmark.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/utils/__init__.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/utils/_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/utils/_dataset.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/utils/_encoded.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/utils/_encoded.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/utils/_internal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/utils/_internal.py -------------------------------------------------------------------------------- /torchvision/prototype/datasets/utils/_resource.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/datasets/utils/_resource.py -------------------------------------------------------------------------------- /torchvision/prototype/models/__init__.py: -------------------------------------------------------------------------------- 1 | from . import depth 2 | -------------------------------------------------------------------------------- /torchvision/prototype/models/depth/__init__.py: -------------------------------------------------------------------------------- 1 | from . import stereo 2 | -------------------------------------------------------------------------------- /torchvision/prototype/models/depth/stereo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/models/depth/stereo/__init__.py -------------------------------------------------------------------------------- /torchvision/prototype/models/depth/stereo/crestereo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/models/depth/stereo/crestereo.py -------------------------------------------------------------------------------- /torchvision/prototype/models/depth/stereo/raft_stereo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/models/depth/stereo/raft_stereo.py -------------------------------------------------------------------------------- /torchvision/prototype/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/transforms/__init__.py -------------------------------------------------------------------------------- /torchvision/prototype/transforms/_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/transforms/_augment.py -------------------------------------------------------------------------------- /torchvision/prototype/transforms/_geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/transforms/_geometry.py -------------------------------------------------------------------------------- /torchvision/prototype/transforms/_misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/transforms/_misc.py -------------------------------------------------------------------------------- /torchvision/prototype/transforms/_presets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/transforms/_presets.py -------------------------------------------------------------------------------- /torchvision/prototype/transforms/_type_conversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/transforms/_type_conversion.py -------------------------------------------------------------------------------- /torchvision/prototype/tv_tensors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/tv_tensors/__init__.py -------------------------------------------------------------------------------- /torchvision/prototype/tv_tensors/_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/tv_tensors/_label.py -------------------------------------------------------------------------------- /torchvision/prototype/utils/__init__.py: -------------------------------------------------------------------------------- 1 | from . import _internal 2 | -------------------------------------------------------------------------------- /torchvision/prototype/utils/_internal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/prototype/utils/_internal.py -------------------------------------------------------------------------------- /torchvision/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/__init__.py -------------------------------------------------------------------------------- /torchvision/transforms/_functional_pil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/_functional_pil.py -------------------------------------------------------------------------------- /torchvision/transforms/_functional_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/_functional_tensor.py -------------------------------------------------------------------------------- /torchvision/transforms/_functional_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/_functional_video.py -------------------------------------------------------------------------------- /torchvision/transforms/_presets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/_presets.py -------------------------------------------------------------------------------- /torchvision/transforms/_transforms_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/_transforms_video.py -------------------------------------------------------------------------------- /torchvision/transforms/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/autoaugment.py -------------------------------------------------------------------------------- /torchvision/transforms/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/functional.py -------------------------------------------------------------------------------- /torchvision/transforms/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/transforms.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/__init__.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/_augment.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/_auto_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/_auto_augment.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/_color.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/_color.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/_container.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/_container.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/_deprecated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/_deprecated.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/_geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/_geometry.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/_meta.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/_misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/_misc.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/_temporal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/_temporal.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/_transform.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/_type_conversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/_type_conversion.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/_utils.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/functional/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/functional/__init__.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/functional/_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/functional/_augment.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/functional/_color.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/functional/_color.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/functional/_deprecated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/functional/_deprecated.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/functional/_geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/functional/_geometry.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/functional/_meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/functional/_meta.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/functional/_misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/functional/_misc.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/functional/_temporal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/functional/_temporal.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/functional/_type_conversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/functional/_type_conversion.py -------------------------------------------------------------------------------- /torchvision/transforms/v2/functional/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/transforms/v2/functional/_utils.py -------------------------------------------------------------------------------- /torchvision/tv_tensors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/tv_tensors/__init__.py -------------------------------------------------------------------------------- /torchvision/tv_tensors/_bounding_boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/tv_tensors/_bounding_boxes.py -------------------------------------------------------------------------------- /torchvision/tv_tensors/_dataset_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/tv_tensors/_dataset_wrapper.py -------------------------------------------------------------------------------- /torchvision/tv_tensors/_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/tv_tensors/_image.py -------------------------------------------------------------------------------- /torchvision/tv_tensors/_keypoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/tv_tensors/_keypoints.py -------------------------------------------------------------------------------- /torchvision/tv_tensors/_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/tv_tensors/_mask.py -------------------------------------------------------------------------------- /torchvision/tv_tensors/_torch_function_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/tv_tensors/_torch_function_helpers.py -------------------------------------------------------------------------------- /torchvision/tv_tensors/_tv_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/tv_tensors/_tv_tensor.py -------------------------------------------------------------------------------- /torchvision/tv_tensors/_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/tv_tensors/_video.py -------------------------------------------------------------------------------- /torchvision/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/vision/HEAD/torchvision/utils.py -------------------------------------------------------------------------------- /version.txt: -------------------------------------------------------------------------------- 1 | 0.25.0a0 2 | --------------------------------------------------------------------------------