- 16 Mar, 2022 1 commit
-
-
Philip Meier authored
* port Pad to prototype transforms * use literal
-
- 15 Mar, 2022 2 commits
-
-
Joao Gomes authored
* replace most asserts with exceptions * fix formating issues * fix linting and remove more asserts * fix regresion * fix regresion * fix bug * apply ufmt * apply ufmt * fix tests * fix format * fix None check * fix detection models tests * non scriptable any * add more checks for None values * fix retinanet test * fix retinanet test * Update references/classification/transforms.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Update references/classification/transforms.py Co-authored-by:
-
vfdev authored
* Added functional affine_bounding_box op with tests * Updated comments and added another test case * Update _geometry.py * Fixed device mismatch issue Added a cude/cpu test Reduced the number of test samples
-
- 14 Mar, 2022 3 commits
-
-
Federico Pozzi authored
* refactor: port RandomHorizontalFlip to prototype API (#5523) * refactor: merge HorizontalFlip and RandomHorizontalFlip Add unit tests for RandomHorizontalFlip * test: RandomHorizontalFlip with p=0 * refactor: remove type annotations from tests * refactor: improve tests * Update test/test_prototype_transforms.py Co-authored-by:
Federico Pozzi <federico.pozzi@argo.vision> Co-authored-by:
Philip Meier <github.pmeier@posteo.de>
-
Vasilis Vryniotis authored
* Moving `sequence_to_str` to `torchvision._utils` * Fix linter * Rename test_prototype_utils test to test_internal_utils
-
Philip Meier authored
* [DEBUG] eurosat prototype on windows and macos * print paths * fix eurosat mock data setup * revert changes * minor cleanup
-
- 11 Mar, 2022 3 commits
-
-
Vasilis Vryniotis authored
-
Philip Meier authored
* revert changes to assert_equal * add tolerances for batch vs single test
-
talregev authored
-
- 10 Mar, 2022 1 commit
-
-
Abhijit Deo authored
* basic stanford cars structure added. * added class in the init. * how to decode the images * fix typo in init * works partially * formatting * formatting * formatting * formatting * formatting * minor changes * minor changes * issue resolved * both datapipes are in sync * checksum dict keys changed * formatting * "ufmt formatting" * "ufmt formatting" * formatting * minor change * minor change * test added * lint (end of line issue) * lint (end of line issue) * minor changes * mock dataset formatting * minor changes * hopefully final * formatting * prototype test * prototype test * prototype test * prototype test * prototype test * prototype test * prototype test * prototype test * Update builtin_dataset_mocks.py * Update stanford_cars.py * Update torchvision/prototype/datasets/_builtin/stanford_cars.py Co-authored-by:
-
- 09 Mar, 2022 4 commits
-
-
Vasilis Vryniotis authored
* Measuring execution times of models. * Speed up models by avoiding re-estimation of eager output * Fixing linter * Reduce input size for big models * Speed up jit check method. * Add simple jitscript fallback check for flaky models. * Restore pytest filtering * Fixing linter
-
Philip Meier authored
* support grayscale / RGB alpha conversions * use _max_valu from stable * remove extra copy for PIL conversion * simplify test image generation for color spaces with alpha channel * use common _max_value in tests * replace dynamically created dicts with if/else * make color space conversion more explicit * make even more explicit * simplify alpha image generation * fix if / elif * add error for unknown conversions * rename RGBA to RGB_ALPHA * cleanup * GRAYSCALE to GRAY
-
Vasilis Vryniotis authored
-
vfdev authored
* Removed _max_value method and added a dictionary Related to https://github.com/pytorch/vision/issues/5502 * Addressed failing tests and restored _max_value method * Added xfailing test to switch quicker * Switch to if/else impl
-
- 07 Mar, 2022 2 commits
-
-
Philip Meier authored
-
Philip Meier authored
-
- 04 Mar, 2022 1 commit
-
-
Vasilis Vryniotis authored
* Remove from models and references. * Adding most tests and docs. * Adding transforms tests. * Remove unnecesary ipython notebook. * Simplify tests. * Addressing comments.
-
- 03 Mar, 2022 2 commits
-
-
Drishti Bhasin authored
* add eurosat * revert formatting * port test and make style changes * add eurosat to __init__ * fix pathlib error * create dataset zipfile and revert pre commit changes * remove unecessary variable in resources * revert auto formatter changes and modify ufmt version * revert change to contributing guide Co-authored-by:
Dbhasin1 <drishti_b@me.iitr.c.in> Co-authored-by:
Nicolas Hug <nicolashug@fb.com>
-
Drishti Bhasin authored
* add country211 * remove unused import * map val to valid and use path comparator * remove unused import * resolve keyerror * map split names in dataset mock Co-authored-by:
-
- 02 Mar, 2022 2 commits
-
-
Philip Meier authored
* improve StrEnum * use StrEnum for model weights * fix test * migrate StrEnum to main area
-
Vasilis Vryniotis authored
* Extend the EfficientNet class to support v1 and v2. * Refactor config/builder methods and add prototype builders * Refactoring weight info. * Update dropouts based on TF config ref * Update BN eps on TF base_config * Use Conv2dNormActivation. * Adding pre-trained weights for EfficientNetV2-s * Add Medium and Large weights * Update stats with single batch run. * Add accuracies in the docs.
-
- 01 Mar, 2022 4 commits
-
-
Prabhat Roy authored
-
Prabhat Roy authored
* Improve test_video_reader * Fix linter error
-
Prabhat Roy authored
-
Philip Meier authored
* expand prototype functional scriptability tests * remove obsolete skips
-
- 28 Feb, 2022 2 commits
-
-
Philip Meier authored
* add prototype AugMix transform * cleanup * refactor auto augment subclasses to only trnasform a single image * address review comments
-
Vasilis Vryniotis authored
* Replace get_image_size/num_channels with get_image_dims * Reduce verbosity * Fix JIT-scriptability * Refactoring * More refactoring * Replace all _FP/_FT direct calls. * Remove usages of get_image_size and get_image_num_channels from code-base. * Fix JIT issues * Adding missing assertion.
-
- 27 Feb, 2022 1 commit
-
-
Hu Ye authored
* Create dropblock.py * add dropblock2d * fix pylint * refactor dropblock * add dropblock * Rename dropblock.py to drop_block.py * fix pylint * add dropblock * add dropblock3d * add drop_block3d * add dropblock * Update drop_block.py * Update torchvision/ops/drop_block.py Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> * Update torchvision/ops/drop_block.py Co-authored-by:
-
- 25 Feb, 2022 2 commits
-
-
Prabhat Roy authored
-
Philip Meier authored
* add prototype transforms that don't need dispatchers * cleanup * remove legacy_transform decorator * remove legacy classes * remove explicit param passing * streamline extra_repr * remove obsolete ._supports() method * cleanup * remove Query * cleanup * fix tests * kernels -> functional * move image size and num channels extraction to functional * extend legacy function to extract image size and num channels * implement dispatching for auto augment * fix auto augment dispatch * revert some naming changes * remove ability to pass params to autoaugment * fix legacy image size extraction * align prototype.transforms.functional with transforms.functional * cleanup * fix image size and channels extraction * fix affine and rotate * revert image size to (width, height) * Minor corrections Co-authored-by:
-
- 24 Feb, 2022 1 commit
-
-
Nikita Shulga authored
Also, remove reference to conda-forge, all CUDA toolchain should be available from NVIDIA channel Install h5py from pip on Windows and skip failing gaussian_blur tests if Win+CUDA11.3 is used
-
- 18 Feb, 2022 1 commit
-
-
Vasilis Vryniotis authored
* Adding basic augmix implementation. * Finish the implementation. * Add tests and documentation. * Fix tests. * Simplify code. * Speed optimizations. * Per image weights instead of per batch. * Fix tests. * Update torchvision/transforms/autoaugment.py Co-authored-by:
vfdev <vfdev.5@gmail.com> * Changing the default severity value to get by default the same strength as RandAugment. Co-authored-by:
-
- 16 Feb, 2022 1 commit
-
-
Philip Meier authored
* add prototype transforms that use the prototype dispatchers Conflicts: torchvision/prototype/transforms/__init__.py * simplify * add logger * remove legacy classes Conflicts: torchvision/prototype/transforms/_augment.py torchvision/prototype/transforms/_auto_augment.py torchvision/prototype/transforms/_geometry.py * make get_params private * remove randbool method * remove AutoAugmentDispatcher * add high level kernels for meta conversion * remove transforms meta abstraction from auto augment transforms * appease mypy * add smoke tests for transforms * remove Query object * remove extra_repr helper * fix tests * appease mypy * revert some changes on the kernel tests * fix dispatcher annotations * remove float cast for torch.rand * add helper to query image * fix imports * address auto augment comments * cleanup
-
- 15 Feb, 2022 2 commits
-
-
Nicolas Hug authored
* Fix resize when size == small_edge_size and max_size isn't None * Better test name
-
vfdev authored
* Removed custom ops for interp with AA * Fixed umft issues Co-authored-by:
-
- 11 Feb, 2022 1 commit
-
-
Philip Meier authored
* revamp prototype features (#5283) * remove decoding from prototype datasets (#5287) * remove decoder from prototype datasets * remove unused imports * cleanup * fix readme * use OneHotLabel in SEMEION * improve voc implementation * revert unrelated changes * fix semeion mock data * fix pcam * readd functional transforms API to prototype (#5295) * readd functional transforms * cleanup * add missing imports * remove __torch_function__ dispatch * readd repr * readd empty line * add test for scriptability * remove function copy * change import from functional tensor transforms to just functional * fix import * fix test * fix prototype features and functional transforms after review (#5377) * fix prototype functional transforms after review * address features review * make mypy more strict on prototype features * make mypy more strict for prototype transforms * fix annotation * fix kernel tests * add automatic feature type dispatch to functional transforms (#5323) * add auto dispatch * fix missing arguments error message * remove pil kernel for erase * automate feature specific parameter detection * fix typos * cleanup dispatcher call * remove __torch_function__ from transform dispatch * remove auto-generation * revert unrelated changes * remove implements decorator * change register parameter order * change order of transforms for readability * add documentation for __torch_function__ * fix mypy * inline check for support * refactor kernel registering process * refactor dispatch to be a regular decorator * split kernels and dispatchers * remove sentinels * replace pass with ... * appease mypy * make single kernel dispatchers more concise * make dispatcher signatures more generic * make kernel checking more strict * revert doc changes * address Franciscos comments * remove inplace * rename kernel test module * fix inplace * remove special casing for pil and vanilla tensors * address comments * update docs * cleanup features / transforms feature branch (#5406) * mark candidates for removal * align signature of resize_bounding_box with corresponding image kernel * fix documentation of Feature * remove interpolation mode and antialias option from resize_segmentation_mask * remove or privatize functionality in features / datasets / transforms
-
- 09 Feb, 2022 1 commit
-
-
Joao Gomes authored
* Consolidating __repr__ strings Co-authored-by:
-
- 08 Feb, 2022 3 commits
-
-
vfdev authored
* Added ref tests for shear X/Y * Added PIL tests and fixed tan(level) difference * Updated tests * Fixed reparam for shear X/Y in autoaugment * Fixed arc_level -> level as atan is applied internally * Fixed links
-
HeungwooLee authored
* Refactor BoxOps tests to use parameterize * Refactor BoxOps tests to use parameterize * Refactor BoxOps to use parameterize, addressed comments from PR#5380 * Refactor BoxOps to use parameterize, addressed minor styling comments from PR#5380 * Refactor BoxOps to use parameterize, addressed typing errorsfrom PR#5380 * Refactor BoxOps to use parameterize, addressed minor naming comments for PR#5380
-
Philip Meier authored
* properly deprecate legacy implementation * cleanup * use warning over deprecation directive * remove patch version * fix link in Kinetics docstring * Some more * fix affine functional tests Co-authored-by:
-
