- 30 Mar, 2020 5 commits
-
-
Yuwen Xiong authored
* fix shape error for deform conv gpu op recover shape of columns for next iteration in for loops, previous version will cause error when batch_sz / n_parallel_imgs > 1 * fix shape error for deform conv cpu op recover shape of columns for next iteration in for loops, previous version will cause error when batch_sz / n_parallel_imgs > 1
-
PatrickBue authored
-
Mikhail Lobanov authored
-
theonekeyg authored
-
Fernando Pérez-García authored
* Improve code readability and docstring * Remove unused argument * Improve make_dataset() readability
-
- 24 Mar, 2020 3 commits
-
-
Francisco Massa authored
* Fix C++ lint * More fixes
-
Francisco Massa authored
Summary: Currently the scale argument can only be of type tuple or integer, this diff allows feeding the input argument `scale` with a list. Pull Request resolved: https://github.com/pytorch/vision/pull/1997 Test Plan: Without this diff, launching the following classy vision task causes error: https://our.intern.facebook.com/intern/fblearner/details/175876950/ With this diff, everything works fine: https://our.intern.facebook.com/intern/fblearner/details/175913768/ Reviewed By: resonatevision Differential Revision: D20544904 Pulled By: ymao1993 fbshipit-source-id: a95a2e9ceadec77fffe234756fb3b38b1b9c9cb1 Co-authored-by:
Yu Mao <ymao1@fb.com>
-
peterjc123 authored
Signed-off-by:peter <peterghost86@gmail.com>
-
- 23 Mar, 2020 3 commits
-
-
bmanga authored
* Add hello world example for C++ torchvision * README updates and mention the hello_world example
-
Willie Maddox authored
* Add fill option to RandomPerspective #1972 * Minor fix to docstring syntax * Add _parse_fill() to get fillcolor (#1972) * Minor refactoring as per comments. * Added test for RandomPerspective with fillcolor. * Force perspective transform in test.
-
Danylo Ulianych authored
* F.normalize unsqueeze mean & std if necessary * added tests to F.normalize for 3d mean & std tensors
-
- 20 Mar, 2020 2 commits
-
-
Philip Meier authored
* add default parameters to README * fix vgg_*_bn
-
Monica Alfaro authored
* modified FasterRCNN to accept negative samples * remove debug lines * Change torch.zeros_like to torch.zerros * Add unit tests * take the `device` into account Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
- 19 Mar, 2020 1 commit
-
-
Eli Uriegas authored
* ci: Add cu102 to CI and packaging Adds cu102 to our testing matrix as well as change the default manywheel version to be cu102. Also adds a function to get manylinux images instead of it being a conditional block. Signed-off-by:
Eli Uriegas <eliuriegas@fb.com> * .circleci: Remove cu100, no longer built We removed the build for cu100 in upstream pytorch as well since we have begun to support cu102 Signed-off-by:
Eli Uriegas <eliuriegas@fb.com> * Fix lint * More lint fixes Co-authored-by:
Francisco Massa <fvsmassa@gmail.com>
-
- 17 Mar, 2020 4 commits
-
-
Tee Jung authored
-
NVS Abhilash authored
Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
Francisco Massa authored
* Base decoder for video. (#1747) Summary: Pull Request resolved: https://github.com/pytorch/vision/pull/1747 Pull Request resolved: https://github.com/pytorch/vision/pull/1746 Added the implementation of ffmpeg based decoder with functionality that can be used in VUE and TorchVision. Reviewed By: fmassa Differential Revision: D19358914 fbshipit-source-id: abb672f89bfaca6351dda2354f0d35cf8e47fa0f * Integrated base decoder into VideoReader class and video_utils.py (#1766) Summary: Pull Request resolved: https://github.com/pytorch/vision/pull/1766 Replaced FfmpegDecoder (incompativle with VUE) by base decoder (compatible with VUE). Modified python utilities video_utils.py for internal simplification. Public interface got preserved. Reviewed By: fmassa Differential Revision: D19415903 fbshipit-source-id: 4d7a0158bd77bac0a18732fe4183fdd9a57f6402 * Optimizating base decoder performance. (#1852) Summary: Pull Request resolved: https://github.com/pytorch/vision/pull/1852 Changed base decoder internals for a faster clip processing. Reviewed By: stephenyan1231 Differential Revision: D19748379 fbshipit-source-id: 58a435f0a0b25545e7bd1a3edb0b1d558176a806 * Minor fix and decoder class members access. Summary: Found and fix a bug in cropping algorithm (simple mistyping). Also derived classes need access to some decoder class members, like initialization parameters - make it protected. Reviewed By: stephenyan1231, fmassa Differential Revision: D19895076 fbshipit-source-id: 691336c8e18526b085ae5792ac3546bc387a6db9 * Added missing header for less dependencies. (#1898) Summary: Pull Request resolved: https://github.com/pytorch/vision/pull/1898 Include streams/samplers shouldn't depend on decoder headers. Add dependencies directly to the place where they are required. Reviewed By: stephenyan1231 Differential Revision: D19911404 fbshipit-source-id: ef322a053708405c02cee4562b456b1602fb12fc * Implemented VUE Asynchronous Decoder Summary: For Mothership we have found that asynchronous decoder provides a better performance. Differential Revision: D20026194 fbshipit-source-id: 627b91844b4e3f917002031dd32cb19c239f4ba8 * fix a bug in API read_video_from_memory (#1942) Summary: Pull Request resolved: https://github.com/pytorch/vision/pull/1942 In D18720474, it introduces a bug in `read_video_from_memory` API. Thank weiyaowang for reporting it. Reviewed By: weiyaowang Differential Revision: D20270179 fbshipit-source-id: 66348c99a5ad1f9129b90e934524ddfaad59de03 * extend decoder to support new video_max_dimension argument (#1924) Summary: Pull Request resolved: https://github.com/pytorch/vision/pull/1924 Extend `video reader` decoder python API in Torchvision to support a new argument `video_max_dimension`. This enables the new video decoding use cases. When setting `video_width=0`, `video_height=0`, `video_min_dimension != 0`, and `video_max_dimension != 0`, we can rescale the video clips so that its spatial resolution (height, width) becomes - (video_min_dimension, video_max_dimension) if original height < original width - (video_max_dimension, video_min_dimension) if original height >= original width This is useful at video model testing stage, where we perform fully convolution evaluation and take entire video frames without cropping as input. Previously, for instance we can only set `video_width=0`, `video_height=0`, `video_min_dimension = 128`, which will preserve aspect ratio. In production dataset, there are a small number of videos where aspect ratio is either extremely large or small, and when the shorter edge is rescaled to 128, the longer edge is still large. This will easily cause GPU memory OOM when we sample multiple video clips, and put them in a single minibatch. Now, we can set (for instance) `video_width=0`, `video_height=0`, `video_min_dimension = 128` and `video_max_dimension = 171` so that the rescale resolution is either (128, 171) or (171, 128) depending on whether original height is larger than original width. Thus, we are less likely to have gpu OOM because the spatial size of video clips is determined. Reviewed By: putivsky Differential Revision: D20182529 fbshipit-source-id: f9c40afb7590e7c45e6908946597141efa35f57c * Fixing samplers initialization (#1967) Summary: Pull Request resolved: https://github.com/pytorch/vision/pull/1967 No-ops for torchvision diff, which fixes samplers. Differential Revision: D20397218 fbshipit-source-id: 6dc4d04364f305fbda7ca4f67a25ceecd73d0f20 * Exclude C++ test files Co-authored-by:
Yuri Putivsky <yuri@fb.com> Co-authored-by:
Zhicheng Yan <zyan3@fb.com>
-
Negin Raoof authored
* Fix for roi_align export * Disable interpolate script module tests Disable test until export of interpolate script module to ONNX is fixed
-
- 16 Mar, 2020 4 commits
-
-
Negin Raoof authored
-
eellison authored
Co-authored-by:eellison <eellison@fb.com>
-
Charles Pao authored
Co-authored-by:Charles Pao <dirtybluer@gmail.com>
-
NVS Abhilash authored
-
- 13 Mar, 2020 4 commits
-
-
Guanheng George Zhang authored
* add checkout/assert in roi_pool * add checkout/assert in roi_align * move check_roi_boxes_shape func to ops/_utils.py * add tests * fix CI * fix CI Co-authored-by:Guanheng Zhang <zhangguanheng@devfair0197.h2.fair>
-
Ailing authored
-
Jerry Zhang authored
https://github.com/pytorch/vision/pull/1949 seems to forget fixing quantized googlenet
-
hx89 authored
-
- 12 Mar, 2020 3 commits
-
-
Francisco Massa authored
-
hx89 authored
* update model path * remove autologits before loading quantized model
-
NVS Abhilash authored
-
- 11 Mar, 2020 4 commits
-
-
Lutz Roeder authored
-
Ashish Farmer authored
* Added code to support creating extension on ROCm * max -> fmaxf conversion for hipification * added WITH_HIP flag for hipExtension * added appropriate headers for HIP build * use USE_ROCM in condition to build * change fmaxf and fminf calls * fminf -> min * fix the check for ROCM_HOME * more robust checking for rocm pytorch * add check for pytorch version before using HIP extensions * conditional reading of ROCM_HOME
-
peterjc123 authored
* Update Windows CI for nightly builds * Add CUDA 10.2 jobs * Update vs install scripts * Add switch case for cuda 10.2 * more changes for cuda 10.2 * Revert upload jobs
-
Eli Uriegas authored
pjh5 is a former employee so we should switch over to using a bot token. Should resolve nightly pipeline failing on upload. Signed-off-by:Eli Uriegas <eliuriegas@fb.com>
-
- 10 Mar, 2020 4 commits
-
-
Soham Tamba authored
-
Kentaro Yoshioka authored
usage and performance are from the vision0.5 release notes.
-
hx89 authored
-
eellison authored
* fix googlenet no aux logits * small fix Co-authored-by:eellison <eellison@fb.com>
-
- 04 Mar, 2020 3 commits
-
-
Francis Charette Migneault authored
-
Tongzhou Wang authored
-
Philip Meier authored
-