- 25 Mar, 2022 3 commits
-
-
Philip Meier authored
* fix PCAM prototype dataset * update Zip and Tar archive loader datapipes * only fail on warnings from the pytorch ecosystem * Revert "only fail on warnings from the pytorch ecosystem" This reverts commit 2bf3aa6f2d875a4055f7f3ed0b468316fc60d4f4.
-
vfdev authored
* Fixed rotate with expand inconsistency between torch vs PIL on odd-sized images * Update functional_tensor.py
-
Lezwon Castelino authored
* added usps dataset * fixed type issues * fix mobilnet norm layer test (#5643) * xfail mobilnet norm layer test * fix test * More robust check in tests for 16 bits images (#5652) * Prefer nvidia channel for conda builds (#5648) To mitigate missing `libcupti.so` dependency * fix torchdata CI installation (#5657) * update urls for kinetics dataset (#5578) * update urls for kinetics dataset * update urls for kinetics dataset * remove errors * update the changes and add test option to split * added test to valid values for split arg * change .txt to .csv for annotation url of k600 Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> * Port Multi-weight support from prototype to main (#5618) * 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:
Vasilis Vryniotis <datumbox@users.noreply.github.com> * pre-download model weights in CI docs build (#5625) * pre-download model weights in CI docs build * move changes into template * change docs image * Regenerated config.yml Co-authored-by:
Philip Meier <github.pmeier@posteo.de> Co-authored-by:
Anton Thomma <11010310+thommaa@users.noreply.github.com> Co-authored-by:
Nikita Shulga <nshulga@fb.com> Co-authored-by:
Sahil Goyal <sahilgoyal6801@gmail.com> Co-authored-by:
-
- 24 Mar, 2022 1 commit
-
-
Joao Gomes authored
* set default tracer kwargs always * simplify code * torchvision/models/feature_extraction.py * Adress PR comments * fix doc format * fix formatting * fix doc error
-
- 23 Mar, 2022 5 commits
-
-
Philip Meier authored
* fix HttpResource.resolve() with preprocess set * fix README * add safe guard for invalid str inputs
-
vfdev authored
* Added functional affine_bounding_box op with tests * Updated comments and added another test case * Update _geometry.py * Added affine_segmentation_mask with tests * Fixed device mismatch issue Added a cude/cpu test Reduced the number of test samples * Added test_correctness_affine_segmentation_mask_on_fixed_input * Updates according to the review * Replaced [None, ...] by [None, :] * Adressed review comments * Fixed formatting and more updates according to the review * Fixed bad merge
-
Philip Meier authored
-
vfdev authored
* [proto] Added functional rotate_bounding_box op * Fix mypy * Apply suggestions from code review Co-authored-by:
-
Philip Meier authored
* port ColorJitter to prototype transforms * make color module private * address review comments
-
- 22 Mar, 2022 2 commits
-
-
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:
-
Sahil Goyal authored
* update urls for kinetics dataset * update urls for kinetics dataset * remove errors * update the changes and add test option to split * added test to valid values for split arg * change .txt to .csv for annotation url of k600 Co-authored-by:
-
- 18 Mar, 2022 1 commit
-
-
Philip Meier authored
-
- 17 Mar, 2022 2 commits
-
-
Philip Meier authored
Co-authored-by: -
Federico Pozzi authored
Co-authored-by:Federico Pozzi <federico.pozzi@argo.vision>
-
- 16 Mar, 2022 1 commit
-
-
Philip Meier authored
* port Pad to prototype transforms * use literal
-
- 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 4 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:
-
Georg Grab authored
Co-authored-by: -
Vasilis Vryniotis authored
* Moving `sequence_to_str` to `torchvision._utils` * Fix linter * Rename test_prototype_utils test to test_internal_utils
-
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 2 commits
-
-
vfdev authored
Co-authored-by: -
Vasilis Vryniotis 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 3 commits
-
-
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
-
Philip Meier 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 5 commits
-
-
Vasilis Vryniotis authored
* Refactor preset transforms * Making presets public.
-
Vasilis Vryniotis authored
* Adding improved MobileNetV2 weights * Fix eval sizes. * Update accuracy with batch-size 1 run
-
kaijieshi authored
Co-authored-by: -
Philip Meier authored
* port FiveCrop and TenCrop to prototype API * fix ten crop for pil * Update torchvision/prototype/transforms/_geometry.py Co-authored-by:
-
Vasilis Vryniotis authored
-
- 05 Mar, 2022 1 commit
-
-
Philip Meier authored
* fix vanilla tensor image detection * fix naming Co-authored-by:
-
- 04 Mar, 2022 3 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.
-
Xu Zhao authored
Co-authored-by:
-
- 03 Mar, 2022 3 commits
-
-
Philip Meier authored
* simplify Feature implementation * fix mypy
-
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:
-
