- 23 Jun, 2020 3 commits
-
-
Zhiqiang Wang authored
* Update docs * Update docs/source/ops.rst Co-authored-by:
Francisco Massa <fvsmassa@gmail.com> Co-authored-by:
Francisco Massa <fvsmassa@gmail.com>
-
joerg-de authored
-
Zhiqiang Wang authored
* Add Deformable Convolution in Doc * Add PSRoIPool and PSRoIAlign in Doc
-
- 22 Jun, 2020 2 commits
-
-
Quentin Duval authored
Refactoring to use contexts managers, list comprehensions when more idiomatic, and minor renaming to help reader clarity (#2335) * Refactoring to use contexts managers, list comprehensions when more idiomatic, and minor renaming to help reader clarity. * Fix flake8 warning in video_utils.py
-
Quentin Duval authored
Fix: PyAV does not support floating point numbers with decimals as FPS when writing and will throw in case this constraint is not satisfied. (#2334)
-
- 19 Jun, 2020 1 commit
-
-
Vitaliy Chiley authored
I was reading through the code and noticed a few spelling errors orignal -> original maintaing -> maintaining
-
- 17 Jun, 2020 1 commit
-
-
peterjc123 authored
* Pin ninja version for Windows jobs * Update build_wheel.sh * Pin for all
-
- 11 Jun, 2020 3 commits
-
-
Philip Meier authored
* add convert_image_dtype to functionals * add ConvertImageDtype transform * add test * remove underscores from numbers since they are not compatible with python<3.6 * address review comments 1/3 * fix torch.bool * use torch.iinfo in test * fix flake8 * remove double conversion * fix flake9 * bug fix * add error messages to test * disable torch.float16 and torch.half for now * add docstring * add test for consistency * move nested function to top * test in CI * dirty progress * add int to int and cleanup * lint Co-authored-by:Philip Meier <meier.philip@posteo.de>
-
vikramtankasali authored
* Adjust hue * Adjust hue acceps torch.tensor uint8 Co-authored-by:Vikram Mukunda Rao Tankasali <vikramtankasali@devvm765.lla0.facebook.com>
-
Ksenija Stanojevic authored
-
- 10 Jun, 2020 1 commit
-
-
Clement Joudet authored
* feat: torchscriptable adjusments * fix: tensor output type * feat: ColorJitter torchscriptable * fix: too many blank lines * fix: documentation spacing and torchscript annotation * refactor: list type for _check_input * refactor: reverting to original syntax Co-authored-by:clement.joudet <clement.joudet@inventia.life>
-
- 09 Jun, 2020 1 commit
-
-
Francisco Massa authored
Summary: Allow writes of >= 2^32 bytes. High-res video can cross this threshold sometimes. LHS is `size_t`, but RHS is all `int32`, and will overflow for output tensors >2Gb. Reviewed By: jsawruk Differential Revision: D21255664 fbshipit-source-id: 7b4c5da989777297a89e73615aaeee8c7a13186a Co-authored-by:Tilak Sharma <tilaksharma@fb.com>
-
- 05 Jun, 2020 1 commit
-
-
Shunta Saito authored
-
- 04 Jun, 2020 3 commits
-
-
Francisco Massa authored
* Make RandomHorizontalFlip torchscriptable * Make _is_tensor_a_torch_image more generic * Make RandomVerticalFlip torchscriptable (#2283) * Make RandomVerticalFlip torchscriptable * Fix lint
-
Ksenija Stanojevic authored
-
Francisco Massa authored
* Bugfix in pad * Address review comments * Fix lint
-
- 03 Jun, 2020 1 commit
-
-
Vasiliy Kuznetsov authored
Summary: We've made two recent changes to QAT in PyTorch core: 1. add support for SyncBatchNorm 2. make eager mode QAT prepare scripts respect device affinity This PR updates the torchvision QAT reference script to take advantage of both of these. This should be landed after https://github.com/pytorch/pytorch/pull/39337 (the last PT fix) to avoid compatibility issues. Test Plan: ``` python -m torch.distributed.launch --nproc_per_node 8 --use_env references/classification/train_quantization.py --data-path {imagenet1k_subset} --output-dir {tmp} --sync-bn ``` Reviewers: Subscribers: Tasks: Tags:
-
- 01 Jun, 2020 3 commits
-
-
Francisco Massa authored
* Add more tests to NMS * Fix lint
-
Francisco Massa authored
* Remove interpolate in favor of PyTorch's implementation * Bugfix * Bugfix
-
peterjc123 authored
-
- 31 May, 2020 1 commit
-
-
peterjc123 authored
-
- 30 May, 2020 1 commit
-
-
peterjc123 authored
* Fix binary builds with CUDA 9.2 on Windows * Remove packages according to VC_YEAR
-
- 29 May, 2020 4 commits
-
-
peterjc123 authored
-
Marc authored
* get pts directly instead of storing full frames to get pts later * fix linting * add initial pts value sort pts * catch decoding errors for read_video_timestamp
-
Michael Kösel authored
* Add norm_layer to MobileNetV2 * Add simple test case * Small fix
-
NVS Abhilash authored
-
- 26 May, 2020 3 commits
-
-
eellison authored
Co-authored-by:eellison <eellison@fb.com>
-
Philip Meier authored
* add compatibility table to README * Update README.rst * Update README.rst Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
Shawn Zhong authored
* Avoid `using` in header files * Fix clang_format * use clang-format-7 to reformat code
-
- 21 May, 2020 3 commits
-
-
Francisco Massa authored
-
Urwa Muaz authored
* add layer freezing param to maskrcnn_resnet50_fpn * freeze ayer param in keypointrcnn_resnet50_fpn * layer freeze tests for mask and keypoint rcnn * correct linting errors * correct linting errors. * correct linting errors
-
Ross Wightman authored
* Fix #2221, DenseNet issue with gradient checkpoints (memory_efficient=True) * Add grad/param count test for mem_efficient densenet
-
- 20 May, 2020 5 commits
-
-
Philip Meier authored
-
Francisco Massa authored
* Deprecate Conv2d, ConvTranspose2d and BatchNorm * Fix lint
-
Erik authored
* Update README.md added some clarity to get the examples executable. Waiting to hear back if instructions should mention to setup COCO dataset * Update README.md * Update README.md * Update README.md * Update README.md * Update README.md
-
Negin Raoof authored
* Fixing nms on boxes when no detection * test * Fix for scale_factor computation * remove newline * Fix for mask_rcnn dynanmic axes * Clean up * Update transform.py * Fix for torchscript * Fix scripting errors * Fix annotation * Fix lint * Fix annotation * Fix for interpolate scripting * Fix for scripting * refactoring * refactor the code * Fix annotation * Fixed annotations * Added test for resize * lint * format * bump ORT * ort-nightly version * Going to ort 1.1.0 * remove version * install typing-extension * Export model for images with no detection * Upgrade ort nightly * update ORT * Update test_onnx.py * updated tests * Updated tests * merge * Update transforms.py * Update cityscapes.py * Update celeba.py * Update caltech.py * Update pkg_helpers.bash * Clean up * Clean up for dynamic split * Remove extra casts * flake8 * Fix for mask rcnn no detection export * clean up * Enable mask rcnn tests * Added test * update ORT * Update .travis.yml * fix annotation * Clean up roi_heads * clean up * clean up misc ops
-
Mike Ruberry authored
Another instance of integer division using the division operator. In this case line 266 already shows the correct formulation, so line 185 only needs the update.
-
- 19 May, 2020 3 commits
-
-
Marc authored
* get pts directly instead of storing full frames to get pts later * fix linting * add initial pts value sort pts
-
Mike Ruberry authored
Integer division using the div operator is deprecated and will throw a RuntimeError in PyTorch 1.6 (and on PyTorch Master very soon). Running a test build with a recent Torchvision commit and integer division using div ('/') disabled revealed this integer division. I'll re-run the tests once this is fixed in case it's masking additional issues. -
Francisco Massa authored
* Make copy of targets in GeneralizedRCNNTransform * Fix flake8
-