- 21 Apr, 2022 1 commit
-
-
YosuaMichael authored
* Remove publication_year and interpolation meta * Add type to _COMMON_META and _COMMON_SWAG_META to prevent error from mypy check * Remove test to check interpolation and publication_year meta Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
-
- 19 Apr, 2022 1 commit
-
-
Lezwon Castelino authored
* add warning if font is not set * added quotes * added changes after review * Apply suggestions from code review Co-authored-by:
Philip Meier <github.pmeier@posteo.de> * escaped warning string * Apply suggestions from code review Co-authored-by:
-
- 13 Apr, 2022 1 commit
-
-
YosuaMichael authored
* Add test for giou_loss * Use standard assert for integer comparison Co-authored-by:
-
- 08 Apr, 2022 2 commits
-
-
YosuaMichael authored
* Add test for sigmoid_focal_loss * Update test/test_ops.py Improve code by using torch.testing.make_tensor Co-authored-by:
-
vfdev authored
* [proto] Added crop_bounding_box op * Removed "pass" * Updated comment * Removed unused args from signature
-
- 07 Apr, 2022 1 commit
-
-
Philip Meier authored
* refactor prototype datasets to inherit from IterDataPipe (#5448) * refactor prototype datasets to inherit from IterDataPipe * depend on new architecture * fix missing file detection * remove unrelated file * reinstante decorator for mock registering * options -> config * remove passing of info to mock data functions * refactor categories file generation * fix imagenet * fix prototype datasets data loading tests (#5711) * reenable serialization test * cleanup * fix dill test * trigger CI * patch DILL_AVAILABLE for pickle serialization * revert CI changes * remove dill test and traversable test * add data loader test * parametrize over only_datapipe * draw one sample rather than exhaust data loader * cleanup * trigger CI * migrate VOC prototype dataset (#5743) * migrate VOC prototype dataset * cleanup * revert unrelated mock data changes * remove categories annotations * move properties to constructor * readd homepage * migrate CIFAR prototype datasets (#5751) * migrate country211 prototype dataset (#5753) * migrate CLEVR prototype datsaet (#5752) * migrate coco prototype (#5473) * migrate coco prototype * revert unrelated change * add kwargs to super constructor call * remove unneeded changes * fix docstring position * make kwargs explicit * add dependencies to docstring * fix missing dependency message * Migrate PCAM prototype dataset (#5745) * Port PCAM * skip_integrity_check * Update torchvision/prototype/datasets/_builtin/pcam.py Co-authored-by:
Nicolas Hug <contact@nicolas-hug.com> Co-authored-by:
-
- 05 Apr, 2022 2 commits
-
-
Vasilis Vryniotis authored
* Use frozen BN only if pre-trained. * Add LSJ and ability to from scratch training. * Fixing formatter * Adding `--opt` and `--norm-weight-decay` support in Detection. * Fix error message * Make ScaleJitter proportional. * Adding more norm layers in split_normalization_params. * Add FixedSizeCrop * Temporary fix for fill values on PIL * Fix the bug on fill. * Add RandomShortestSize. * Skip resize when an augmentation method is used. * multiscale in [480, 800] * Add missing star * Add new RetinaNet variant. * Add tests. * Update expected file for old retina * Fixing tests * Add FrozenBN to retinav2 * Fix network initialization issues * Adding BN support in MaskRCNNHeads and FPN * Adding support of FasterRCNNHeads * Introduce norm_layers in backbone utils. * Bigger RPN head + 2x rcnn v2 models. * Adding gIoU support to retinanet * Fix assert * Add back nesterov momentum * Rename and extend `FastRCNNConvFCHead` to support arbitrary FCs * Fix linter
-
YosuaMichael authored
* Add vit_b_16_swag * Better handling idiom for image_size, edit test_extended_model to handle case where number of param differ from default due to different image size input * Update the accuracy to the experiment result on torchvision model * Fix typo missing underscore * raise exception instead of torch._assert, add back publication year (accidentally deleted) * Add license information on meta and readme * Improve wording and fix typo for pretrained model license in readme * Add vit_l_16 weight * Update README.rst Co-authored-by:
-
- 04 Apr, 2022 3 commits
-
-
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: -
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 * WIP * Fixed tests * Updated warning message
-
Yassine Alouini authored
* [FEAT] Start implementing Food101 using the new datasets API. WIP. * [FEAT] Generate Food101 categories and start the test mock. * [FEAT] food101 dataset code seems to work now. * [TEST] food101 mock update. * [FIX] Some fixes thanks to running food101 tests. * [FIX] Fix mypy checks for the food101 file. * [FIX] Remove unused numpy. * [FIX] Some changes thanks to code review. * [ENH] More idomatic dataset code thanks to code review. * [FIX] Remove unused cast. * [ENH] Set decompress and extract to True for some performance gains. * [FEAT] Use the preprocess=decompress keyword. * [ENH] Use the train and test.txt file instead of the .json variants and simplify code + update mock data. * [ENH] Better food101 mock data generation. * [FIX] Remove a useless print. Co-authored-by:
-
- 01 Apr, 2022 3 commits
-
-
Prabhat Roy authored
* Fix shape mismatch error * Update offset Co-authored-by: -
YosuaMichael authored
* Add vit_b_16_swag * Better handling idiom for image_size, edit test_extended_model to handle case where number of param differ from default due to different image size input * Update the accuracy to the experiment result on torchvision model * Fix typo missing underscore * raise exception instead of torch._assert, add back publication year (accidentally deleted) * Add license information on meta and readme * Improve wording and fix typo for pretrained model license in readme * Add vit_l_16 weight * Update README.rst Co-authored-by:
-
Vasilis Vryniotis authored
* Detection recipe enhancements * Add back nesterov momentum
-
- 30 Mar, 2022 1 commit
-
-
Philip Meier authored
* add test to enforce infinite buffer size for all applicable datapipes * use utility function to extract datapipes * check for buffer_size attr rather than type * simplify
-
- 29 Mar, 2022 1 commit
-
-
Joao Gomes authored
* reverting some recently introduced exceptions * Update torchvision/ops/poolers.py Co-authored-by:
-
- 28 Mar, 2022 1 commit
-
-
Philip Meier authored
* enforce shuffling before sharding * revert test changes and add comment
-
- 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:
Anton Thomma <anton@pri.co.nz> 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:
-
- 23 Mar, 2022 4 commits
-
-
Philip Meier authored
-
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
-
vfdev authored
* [proto] Added functional rotate_bounding_box op * Fix mypy * Apply suggestions from code review Co-authored-by:
-
- 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:
-
- 21 Mar, 2022 1 commit
-
-
Nicolas Hug authored
-
- 20 Mar, 2022 1 commit
-
-
Philip Meier authored
* xfail mobilnet norm layer test * fix test
-
- 17 Mar, 2022 1 commit
-
-
Federico Pozzi authored
Co-authored-by:Federico Pozzi <federico.pozzi@argo.vision>
-
- 16 Mar, 2022 2 commits
-
-
Vasilis Vryniotis authored
* Remove unnecessary PYTORCH_TEST_WITH_SLOW env * Completely remove skip
-
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:
-
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:
-
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 1 commit
-
-
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
-
