- 28 Jul, 2022 1 commit
-
-
vfdev authored
* [proto] Added few transforms tests, part 1 (#6262) * Added supported/unsupported data checks in the tests for cutmix/mixup * Added RandomRotation, RandomAffine transforms tests * Added tests for RandomZoomOut, Pad * Update test_prototype_transforms.py * Added RandomCrop transform and tests (#6271) * [proto] Added GaussianBlur transform and tests (#6273) * Added GaussianBlur transform and tests * Fixing code format * Copied correctness test * [proto] Added random color transforms and tests (#6275) * Added random color transforms and tests * Disable smoke test for RandomSolarize, RandomAdjustSharpness * Added RandomPerspective and tests (#6284) - replaced real image creation by mocks for other tests * Added more functional tests (#6285) * [proto] Added elastic transform and tests (#6295) * WIP [proto] Added functional elastic transform with tests * Added more functional tests * WIP on elastic op * Added elastic transform and tests * Added tests * Added tests for ElasticTransform * Try to format code as in https://github.com/pytorch/vision/pull/5106 * Fixed bug in affine get_params test * Implemented RandomErase on PIL input as fallback to tensors (#6309) Added tests * Added image_size computation for BoundingBox.rotate if expand (#6319) * Added image_size computation for BoundingBox.rotate if expand * Added tests * Added erase_image_pil and eager/jit erase_image_tensor test (#6320) * Updates according to the review Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 22 Jul, 2022 1 commit
-
-
Philip Meier authored
* upgrade usort to * Also update black * Actually use 1.0.2 * Apply pre-commit Co-authored-by:Nicolas Hug <contact@nicolas-hug.com>
-
- 21 Jul, 2022 1 commit
-
-
Zachariah Carmichael authored
-
- 11 Jul, 2022 1 commit
-
-
vfdev authored
* Another attempt to add transforms * Fixed padding type hint * Fixed fill arg for pad and rotate, affine * code formatting and type hints for affine transformation * Fixed flake8 * Updated tests to save and load transforms * Fixed code formatting issue * Fixed jit loading issue * Restored fill default value to None Updated code according to the review * Added tests for rotation, affine and zoom transforms * Put back commented code * Random erase bypass boxes and masks Go back with if-return/elif-return/else-return * Fixed acceptable and non-acceptable types for Cutmix/Mixup * Updated conditions for _BaseMixupCutmix
-
- 06 Jul, 2022 1 commit
-
-
vfdev authored
* Added mid-level ops and feature-based ops * Fixing deadlock in dataloader with circular imports * Added non-scalar fill support workaround for pad * Removed comments * int/float support for fill in pad op * Updated type hints and removed bypass option from mid-level methods * Minor nit fixes
-
- 01 Jul, 2022 1 commit
-
-
vfdev authored
-
- 27 Jun, 2022 1 commit
-
-
YosuaMichael authored
-
- 23 Jun, 2022 2 commits
- 13 Jun, 2022 1 commit
-
-
Lenz authored
* Added elastic augment * ufmt formatting * updated comments * fixed circular dependency issue and bare except error * Fixed three type checking errors in functional_tensor.py * ufmt formatted * changed elastic_deformation to a more common implementation Implementation uses alpha and sigma to control strength and smoothness of the displacement vectors in elastic_deformation instead of control_point_spacings and sigma. * ufmt formatting * Some performance updates Put random offset vectors to device before gaussian_blur is applied speeds it up 3-fold. * fixed type error * fixed again a type error * Update torchvision/transforms/functional_tensor.py Co-authored-by:
vfdev <vfdev.5@gmail.com> * Added some requested changes - pil image support similar to GaussianBlur - changed interpolation arg to InterpolationMode - added a wrapper in torchvision.transforms.functional.py that gets called by the class in transforms.py -renamed it to ElasticTransform - handled sigma = 0 case * added img docstring * added some tests * Updated tests and the code * Added the requested changes to the arguments of F.elastic_transform Added random_state and displacement as arguments to F.elastic_transform * fixed the type error * Fixed tests and docs * implemented requested changes Changes: 1) alpha AND sigma OR displacement must be given as arguments to transforms.functional_tensor.elastic_transform instead of alpha AND sigma AND displacement 2) displacements are accepted in transforms.functional.elastic_transform as np.array and torch.Tensor instead of only accepting torch.Tensor * ufmt formatting * trochscript error resolved replaced torch.from_numpy() to torch.Tensor() to make it compatible to torchscript * revert to torch.from_numpy() * updated argument checks and errors - In F.elastic_transform added check to see if both user inputs img and displacement are either of type PIL Image and ndarray or both of type tensor. - In F_t.elastic_transform added check if alpha and sigma are None if displacement is given or vice versa. * fixed seed error changed torch.seed to torch.manual_seed in F_t.elastic_transform * Reverted displacement type and other cosmetics * Other minor improvements * changed gaussian_blur filter size changed gaussian_blur filter size from 4 * int(sigma) + 1 to int(8 * sigma + 1) to make it consistent with ernestums implementation * resolved merge error * Revert "resolved merge error" This reverts commit 6a4a4e74ff4d078e2c2753d359185f9a81c415d0. * resolve merge error * ufmt formatted * ufmt formated once again.. * fixed unsupported operand error * Update API and removed random_state from functional part * Added default values * Added ElasticTransform to gallery and updated the docstring * Updated gallery and added _log_api_usage_once BTW, matplotlib.pylab is deprecated * Updated gallery transforms code * Updates according to review Co-authored-by:
-
- 11 Jun, 2022 1 commit
-
-
vfdev authored
* Update _pil_constants.py * Update _pil_constants.py * Fix flake8 * Fixed two related warnings in tests * switch dir with hasattr Co-authored-by:
-
- 23 May, 2022 1 commit
-
-
Vasilis Vryniotis authored
* Remove `(N, T, H, W, C) => (N, T, C, H, W)` conversion on presets * Update docs. * Fix the tests * Use `output_format` for `read_video()` * Use `output_format` for `Kinetics()` * Adding input descriptions on presets
-
- 16 May, 2022 2 commits
-
-
Vasilis Vryniotis authored
* Minor updates on model examples. * Improving wording of auto-generated docs. * Add general info for pre-trained weights. * Updating torch hub * Minor updates * Make lengthy meta-data partially visible * Adding meta-data and reference info. * Minor corrections * Update docs/source/models_new.rst Co-authored-by:
-
kylematoba authored
* Requested here https://github.com/pytorch/vision/pull/5898#discussion_r864765799 . * Fix tests * ufmt, not black Co-authored-by:
Philip Meier <github.pmeier@posteo.de> Co-authored-by:
-
- 13 May, 2022 1 commit
-
-
Vasilis Vryniotis authored
* Adding `__repr__` in presets * Adds `describe()` methods to all presets. * Adding transform descriptions in the documentation. * Change "preprocessing" to "inference"
-
- 09 May, 2022 1 commit
-
-
kylematoba authored
* Update transforms for PIL deprecation * Changes agreed at pytorch/vision#5898 * black, sort constants, version check * Format tests * Square brackets * Update torchvision/transforms/_pil_constants.py Co-authored-by:
-
- 26 Apr, 2022 2 commits
-
-
vfdev authored
* Fixed issue with padding on CI * Disabled failing tests with color_jitter * Remove Jitter workaround Co-authored-by: -
Philip Meier authored
-
- 04 Apr, 2022 1 commit
-
-
Philip Meier authored
* port image type conversion transforms to prototype API * implement proposal for image type conversion * add deprecation warnings * appease mypy Co-authored-by:
-
- 29 Mar, 2022 1 commit
-
-
Behrooz authored
* Update docstring of adjust_hue Signed-off-by:
Behrooz <3968947+drbeh@users.noreply.github.com> * Update docstring of ColorJitter Signed-off-by:
-
- 25 Mar, 2022 1 commit
-
-
vfdev authored
* Fixed rotate with expand inconsistency between torch vs PIL on odd-sized images * Update functional_tensor.py
-
- 22 Mar, 2022 1 commit
-
-
Vasilis Vryniotis authored
* Moving basefiles outside of prototype and porting Alexnet, ConvNext, Densenet and EfficientNet. * Porting googlenet * Porting inception * Porting mnasnet * Porting mobilenetv2 * Porting mobilenetv3 * Porting regnet * Porting resnet * Porting shufflenetv2 * Porting squeezenet * Porting vgg * Porting vit * Fix docstrings * Fixing imports * Adding missing import * Fix mobilenet imports * Fix tests * Fix prototype tests * Exclude get_weight from models on test * Fix init files * Porting googlenet * Porting inception * porting mobilenetv2 * porting mobilenetv3 * porting resnet * porting shufflenetv2 * Fix test and linter * Fixing docs. * Porting Detection models (#5617) * fix inits * fix docs * Port faster_rcnn * Port fcos * Port keypoint_rcnn * Port mask_rcnn * Port retinanet * Port ssd * Port ssdlite * Fix linter * Fixing tests * Fixing tests * Fixing vgg test * Porting Optical Flow, Segmentation, Video models (#5619) * Porting raft * Porting video resnet * Porting deeplabv3 * Porting fcn and lraspp * Fixing the tests and linter * Porting docs, examples, tutorials and galleries (#5620) * Fix examples, tutorials and gallery * Update gallery/plot_optical_flow.py Co-authored-by:
Anton Thomma <anton@pri.co.nz> Co-authored-by:
Anton Thomma <11010310+thommaa@users.noreply.github.com> Co-authored-by:
-
- 17 Mar, 2022 1 commit
-
-
Philip Meier authored
Co-authored-by:
-
- 15 Mar, 2022 3 commits
-
-
Vasilis Vryniotis authored
-
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:
-
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 2 commits
-
-
Georg Grab authored
Co-authored-by: -
Philip Meier authored
* port RandomZoomOut from detection references to prototype transforms * copy by default * move padding parsing to stable API * merge zoom out kernels into transform * lint * fix filling if no padding was done * resolve todo
-
- 11 Mar, 2022 1 commit
-
-
Vasilis Vryniotis authored
-
- 09 Mar, 2022 1 commit
-
-
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
-
- 04 Mar, 2022 2 commits
-
-
Nicolas Hug authored
* Fix accimage tests * Adding workaround for accimage * Refactoring * restore channels Co-authored-by: -
Vasilis Vryniotis authored
* Remove from models and references. * Adding most tests and docs. * Adding transforms tests. * Remove unnecesary ipython notebook. * Simplify tests. * Addressing comments.
-
- 01 Mar, 2022 1 commit
-
-
vfdev authored
-
- 28 Feb, 2022 1 commit
-
-
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.
-
- 23 Feb, 2022 1 commit
-
-
Philip Meier authored
* add tensor kernels for normalize and erase * add image tensor assertion
-
- 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:
-
- 16 Feb, 2022 1 commit
-
-
Nikita Shulga authored
-
- 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:
-
- 10 Feb, 2022 1 commit
-
-
Vasilis Vryniotis authored
-
