- 22 Feb, 2021 8 commits
-
-
Nicolas Hug authored
* Specify coordinate constraints * some more * flake8
-
Nicolas Hug authored
* removed call to optimze_for_mobile * removed unrelated changes
-
Philip Meier authored
Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
Philip Meier authored
Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
Philip Meier authored
* use common download utils in VOC and SBDataset * add tests for VOC * use common base class for VOC datasets * remove old voc test and fake data generation Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
Philip Meier authored
Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
Philip Meier authored
Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
Francisco Massa authored
This reverts commit 74de51d6.
-
- 19 Feb, 2021 1 commit
-
-
peterjc123 authored
-
- 18 Feb, 2021 1 commit
-
-
Philip Meier authored
* add base class for datasets tests * add better type hints * add documentation to subclasses * add utility functions to create files / folders of random images and videos * fix imports * remove class properties * fix smoke test * fix type hints * fix random size generation * add Caltech256 as example * add utility function to create grid of combinations * add CIFAR100? as example * lint * add missing import * improve documentation * create 1 frame videos by default * remove obsolete check * return path of files created with utility functions * [test] close PIL file handles before deletion * fix video folder creation * generalize file handle closing * fix lazy imports * add test for transforms * fix explanation comment * lint * force load opened PIL images * lint * copy default config to avoid inplace modification * enable additional arg forwarding
-
- 16 Feb, 2021 4 commits
-
-
Matti Picus authored
-
Matti Picus authored
Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
Matti Picus authored
Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
Nicolas Hug authored
* restore cache * Trigger CI again to see if cache is hit
-
- 15 Feb, 2021 4 commits
-
-
vfdev authored
- Added conda-forge channel
-
Vasilis Vryniotis authored
* Avoid freezing bn1 if all layers are trainable. * Remove misleading comments.
-
vfdev authored
Fixes #3393
-
Vasilis Vryniotis authored
* Replace type T with accumulator. * Upcast tensors of box ops to avoid overflow in multiplications.
-
- 12 Feb, 2021 5 commits
-
-
Vasilis Vryniotis authored
Summary: In opt mode, the linker could strip out the symbols related to the torchvision ops if they had not been explicitly requested. Adding link-whole is one way of fixing this. Reviewed By: datumbox, ppwwyyxx Differential Revision: D26249713 fbshipit-source-id: 4e55e2ebb483922fe308fd8b83456439bde728c9
-
Vasilis Vryniotis authored
-
Vasilis Vryniotis authored
* Fix test flakiness caused by rounding. * Update test/test_transforms.py * Styles Co-authored-by:Nicolas Hug <contact@nicolas-hug.com>
-
Vasilis Vryniotis authored
Summary: Pull Request resolved: https://github.com/pytorch/vision/pull/3370 Pull Request resolved: https://github.com/pytorch/vision/pull/3361 Convert pytorch_gpu to use GPU/RE Reviewed By: datumbox Differential Revision: D26264102 fbshipit-source-id: b45179215ecb1142f6468386c4fee1759dd82f0c
-
Eli Uriegas authored
(cherry picked from commit 2f40a483d73018ae6e1488a484c5927f2b309969) Signed-off-by:Eli Uriegas <eliuriegas@fb.com>
-
- 11 Feb, 2021 6 commits
-
-
bmanga authored
Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
bmanga authored
* On MSVC, instruct the linker not to drop the _register_ops symbol * Remove workaround for ops registration on windows Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
Vasilis Vryniotis authored
-
Vasilis Vryniotis authored
Reviewed By: zertosh Differential Revision: D26396892 fbshipit-source-id: be0bd7fb54cb2f7bd0e0fb5222cc25b4930679a6
-
Philip Meier authored
* remove generated dataset folders after download tests * revert unnecessary style changes
-
Vasilis Vryniotis authored
Summary: Noticed this copy-paste error while reading D26026259 (https://github.com/pytorch/vision/commit/f16322b596c7dc9e9d67d3b40907694f29e16357) for inspiration to solve a similar problem Created from Diffusion's 'Open in Editor' feature. Reviewed By: datumbox Differential Revision: D26373056 fbshipit-source-id: d67f46243a280e8a59d12bad9dd730c1a3d512c9
-
- 10 Feb, 2021 2 commits
-
-
Vasilis Vryniotis authored
* Adding the average_checkpoints() method. * Adding the store_model_weights() method.
-
Vasilis Vryniotis authored
-
- 09 Feb, 2021 3 commits
-
-
Vasilis Vryniotis authored
-
Vasilis Vryniotis authored
* Adding TODO placeholders. * More placeholders. * Add MobileNetV3 small pre-trained weights. * Remove placeholders.
-
clint (woonhyuk baek) authored
* aspect ratio must be a sampling from log scale. reference from: https://github.com/pytorch/vision/blob/8317295c1d272e0ba7b2ce31e3fd2c048235fc73/torchvision/transforms/transforms.py#L833-L836 * add random erasing unittest code * Increased threshold for difference in sampling rate * move outside of the loop * log_ratio move outside of the loop. RandomResizedCrop also Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 08 Feb, 2021 1 commit
-
-
Vasilis Vryniotis authored
-
- 05 Feb, 2021 1 commit
-
-
Philip Meier authored
* fix STL10 folds * use np.int64 over np.long
-
- 04 Feb, 2021 4 commits
-
-
Ivan Kobzarev authored
* [android] android gradle project for ops * Change CMakeLists to latest PyTorch * Use mobilenet_v3 models for detection Don't need to have two variants of the model anymore, but I'm not removing it for now * Fix orientation when angle = 0 * [android][test_app] Fix YUV decoding * Use smaller version of mobilenet model * Divide inputs by 255 again * [android] assets mobilenetv3 Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
Nicolas Hug authored
Summary: This diff adds a new target to torchvision which enables users to use torchvision ops from C++. For now, the `cpp_library` is not used by the `python_cpp_library`. We should instead refactor the logic in torchvision to directly use `cpp_library` instead. There is currently an inconsistency between fbcode and OSS users. OSS users can import torchvision via ``` #include <torchvision/vision.h> ``` while fbcode users need to do ``` #include <torchvision/csrc/vision.h> ``` It would be good to fix this discrepancy in the future. I didn't directly use `test_frcnn_tracing.cpp` due to complications for getting the `.pt` file in a way that works for both OSS and fbcode, so instead we added a self-contained test that should validate that the torchvision ops are properly registered and visible to JIT Reviewed By: datumbox Differential Revision: D26225669 fbshipit-source-id: 5dd9fb98dd58e854f95806e4860d02f54fc04ea4 Co-authored-by:Francisco Massa <fmassa@fb.com>
-
Francisco Massa authored
-
Philip Meier authored
* extract some functionality from places365 fakedata for common use * add a common DatasetTestcase * add fakedata generation and tests for STL10 * lint Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-