- 03 May, 2022 9 commits
-
-
Joao Gomes authored
* updating mnasnet docs * fix bug * fix lint * fix end of file * fix linting * Add missing weights Co-authored-by:Nicolas Hug <contact@nicolas-hug.com>
-
Zhiqiang Wang authored
* Update ShuffleNetV2 docstring * Add entry for ShuffleNet V2 in new doc * Minor fixes Co-authored-by: -
Abhijit Deo authored
* init * init * Update torchvision/models/detection/faster_rcnn.py Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> * Update torchvision/models/detection/faster_rcnn.py Co-authored-by:
-
Aditya Oke authored
* Add inception * Fix googlenet bug * Fix bug add file * Update torchvision/models/inception.py Co-authored-by: -
Aditya Oke authored
* add docs * Small fix * Update docs/source/models/lraspp.rst Co-authored-by: -
F-G Fernandez authored
* docs: Added FCN to new doc * docs: Updated docstring * docs: Fixed docstring * Update docs/source/models/fcn.rst Co-authored-by:
Aditya Oke <47158509+oke-aditya@users.noreply.github.com> * Apply suggestions from code review Co-authored-by:
FG Fernandez <26927750+frgfm@users.noreply.github.com> Co-authored-by:
-
F-G Fernandez authored
* docs: Added SSDLite to new doc * docs: Updated docstring * docs: Fixed docstring * Apply suggestions from code review Co-authored-by:
-
Jeffery (Zeyu) Zhao authored
* Add *_FOUND macros to CMakeLists.txt * Remove NVJPEG as it fails CI Co-authored-by: -
Daniel Angelov authored
-
- 02 May, 2022 1 commit
-
-
Federico Pozzi authored
* feat: add functional pad on segmentation mask * test: add basic correctness test with random masks * test: add all padding options * fix: pr comments * fix: tests * refactor: reshape tensor in 4d, then pad Co-authored-by:Federico Pozzi <federico.pozzi@argo.vision>
-
- 28 Apr, 2022 10 commits
-
-
Abhijit Deo authored
* added ciou loss * "formatting with flake8 and ufmt" * formatting with ufmt and flake8 * minor changes * changes as per the suggestions * added reference in torchvision/ops/__init__.py * sample test * tests formatted * added description * formatting * edited tests * changes in tests * added tests for multiple boxes * minor edits * minor edit * doc added * minor edits * Update test_ops.py * formatting test file * changes as per the suggestions * formatting and adding some more tests * bounding box added * removed unnecessary comment * added docstring * added type annotations * removed potential bug * Update torchvision/ops/boxes.py Co-authored-by:
Philip Meier <github.pmeier@posteo.de>
-
YosuaMichael authored
* Add shufflenetv2 1.5 and 2.0 weights * Update recipe * Add to docs * Use resize_size=232 for eval and update the result * Add quantized shufflenetv2 large * Update docs and readme * Format with ufmt * Add to hubconf.py * Update readme for classification reference * Fix reference classification readme * Fix typo on readme * Update reference/classification/readme
-
Philip Meier authored
* add Windows and macOS CI jobs for prototype tests * fix CircleCI config * cleanup * use 3.8 as base to surface errors * try using bash explicitly * cleanup * try test reports * debug * disable CircleCI * add write permissions for write * expand permissions * try move permissions on job rather than global * debug * always debug * maximum permissions * cleanup * cleanup
-
Philip Meier authored
-
Philip Meier authored
* fix test_data_loader on Windows and macOS * Update test/test_prototype_builtin_datasets.py Co-authored-by:
-
Aditya Oke authored
* Start adding * Finish adding docs * Fix other bugs * Update torchvision/models/mobilenetv3.py * Update torchvision/models/mobilenetv3.py Co-authored-by: -
Abhijit Deo authored
* init * minor change * typo Co-authored-by: -
Nicolas Hug authored
* Custom class for weights table * Proper CSS
-
YosuaMichael authored
* Skip big model in test * Let get_models_from_module func to read skipped_models from global directly instead of function param to reduce changes * Also skip regnet_y_128gf * Only skip test for test_classification_model and create toggle using env var * Remove unnecessary comment * Fix comparison of device to use str * Add logic to test_classification_model directly * Add kprcnn in autocast flaky list Co-authored-by: -
Vasilis Vryniotis authored
-
- 27 Apr, 2022 12 commits
-
-
Vasilis Vryniotis authored
-
Nicolas Hug authored
Co-authored-by: -
Joao Gomes authored
* Add swin_t to slow list * apply ufmt * update expect for swin_t Co-authored-by: -
Hu Ye authored
* refactor doc * refactor code * add fcos.rst * add fcos Co-authored-by: -
F-G Fernandez authored
* docs: Added MaskRCNN to new doc * docs: Updated docstring * ufmt Co-authored-by:
-
Nicolas Hug authored
* Add revamped docs for video classification models * EOL
-
F-G Fernandez authored
* docs: Added entry for MobileNet V2 in new doc * docs: Updated MobileNetV2 docstring * docs: Fixed docstring Co-authored-by: -
Hu Ye authored
* add swin transformer * Update swin_transformer.py * Update swin_transformer.py * fix lint * fix lint * refactor code * add swin_transformer * Update swin_transformer.py * fix bug * refactor code * fix lint * update init_weights * move shift_window into attention * refactor code * fix bug * Update swin_transformer.py * Update swin_transformer.py * fix lint * add patch_merge * fix bug * Update swin_transformer.py * Update swin_transformer.py * Update swin_transformer.py * refactor code * Update swin_transformer.py * refactor code * fix lint * refactor code * add swin_tiny * add swin_tiny.pkl * fix lint * Delete ModelTester.test_swin_tiny_expect.pkl * add swin_tiny * add * add Optional to bias * update init weights * update init_weights and add no weight decay * add no weight decay * add set_weight_decay * add set_weight_decay * fix lint * fix lint * add lr_cos_min * add other swin models * Update torchvision/models/swin_transformer.py Co-authored-by:
Joao Gomes <jdsgomes@fb.com>
-
Vasilis Vryniotis authored
-
Vasilis Vryniotis authored
* Document MobileNetV3 quantization approach * Change the URL.
-
Yassine Alouini authored
* [DOC] Add alexnet new documentation. * [FIX] Fix wrong alexnet header. * [FIX] Fix the AlexNet paper link. Co-authored-by: -
Nicolas Hug authored
-
- 26 Apr, 2022 5 commits
-
-
Yassine Alouini authored
-
Nicolas Hug authored
* Start doc revamp for detection models * Minor cleanup * Use list of tuples for metrics
-
vfdev authored
* Fixed issue with padding on CI * Disabled failing tests with color_jitter * Remove Jitter workaround Co-authored-by: -
Philip Meier authored
-
Lezwon Castelino authored
* added dense net docs * fix models_new.rst
-
- 25 Apr, 2022 3 commits
-
-
vfdev authored
* [proto] Added crop_bounding_box op * Added `crop_segmentation_mask` op * Fixed failed mypy * Added tests for resized_crop_bounding_box * Fixed code formatting * Added resized_crop_segmentation_mask op * Added tests
-
Nicolas Hug authored
-
Abhijit Deo authored
* initial commit * formatting * minor change * replaced pdf link with webpage link * replaced pdf link with webpage link * minor * minor Co-authored-by:
-
