- 19 May, 2022 1 commit
-
-
Nicolas Hug authored
* Remove models.rst * Remove '- New' * Put back torchhub section where it originally was
-
- 18 May, 2022 4 commits
-
-
Nicolas Hug authored
* Doc revamp for optical flow models * Some more
-
Nicolas Hug authored
* Classif models * Detection * Segmentation * quantization * Video * optical flow * tests * Fix docs * Fix Video dataset * Consistency for RAFT dataset names * use ImageNet-1K * Use COCO-val2017-VOC-labels for segmentation * formatting
-
Vasilis Vryniotis authored
* Adding docs for quantized models. * Adding docs for video models. * Adding docs for segmentation models. * Adding docs for optical flow models. * Adding docs for detection models. * Fix typo. * Make changes from code-review.
-
Aditya Oke authored
* try re arranging docs * rewrite docs a bit * Apply suggestions from code review Co-authored-by:
Vasilis Vryniotis <datumbox@users.noreply.github.com> Co-authored-by:
-
- 17 May, 2022 2 commits
-
-
Vasilis Vryniotis authored
* Improving the auto-gen doc. * Adding details for AlexNet, ConvNext, DenseNet, EfficientNets, GoogLeNet and InceptionV3. * Fixing location of `_docs` * Adding docs in the remaining classification models. * Fix linter
-
F-G Fernandez authored
* docs: Added quantized ResNeXt to new docs * docs: Fixed docstring Co-authored-by:
-
- 16 May, 2022 3 commits
-
-
Zhiqiang Wang authored
-
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:
Nicolas Hug <contact@nicolas-hug.com> * Moving Torch hub section at the end Co-authored-by:
-
Abhijit Deo authored
* added note * quantize = True higlighted in the note. * Keep "Large" in docstring Co-authored-by:
-
- 13 May, 2022 4 commits
-
-
Hu Ye authored
* Create resnet_quant.rst * add resnet quant * refactor docs * Minor fix * Nit Co-authored-by: -
Yassine Alouini authored
* [DOC] Add quantized inception updated documentation. * Add missing file. * Apply suggestions from code review [ENH] Improve doc thanks to various code review comments. Co-authored-by:
Aditya Oke <47158509+oke-aditya@users.noreply.github.com> Co-authored-by:
-
Vasilis Vryniotis authored
* Adding code examples for image classification + quant * Adding code example detection * Adding code example segmentation * Adding code example for video classification * Adding information on how to use the new API. * Putting back the comma. * Apply suggestions from code review Co-authored-by:
-
Vasilis Vryniotis authored
* Adding `__repr__` in presets * Adds `describe()` methods to all presets. * Adding transform descriptions in the documentation. * Change "preprocessing" to "inference"
-
- 12 May, 2022 6 commits
-
-
Vasilis Vryniotis authored
* Remove old "minimum input size" from docstrings. * Remove "currently only XYZ weights available" * Fix description of wide_resnet101_2 * Make display license URLs as links. * Clarify the order of dims of min_size. * Remove lengthy keypoint_names from meta-table.
-
Lezwon Castelino authored
* added quantized mobilenetv2 docs * remove quotes Co-authored-by: -
Nicolas Hug authored
-
Vasilis Vryniotis authored
-
Aditya Oke authored
Co-authored-by: -
Aditya Oke authored
-
- 11 May, 2022 1 commit
-
-
Nicolas Hug authored
* POC * Update torchvision/models/resnet.py Co-authored-by:
-
- 10 May, 2022 1 commit
-
-
Nicolas Hug authored
* Add Docs for QuantizedGoogleNet * ufmt * Use explicit _QuantizedWeights or _Weights suffix
-
- 09 May, 2022 4 commits
-
-
Yassine Alouini authored
* [FEAT] Add distance IoU and distance IoU loss + some tests (WIP for tests). * [FIX] Remove URL from docstring + remove assert since it causes a big performance drop. * [FIX] eps isn't None. * [TEST] Update existing box dIoU test + add dIoU loss tests (inspired from cIoU ones). * [ENH] Some pre-commit fixes + remove print + mypy. * [ENH] Pass the device in the assertion for the dIoU loss test. * [FIX] Remove type hints from the dIoU box test. * [ENH] Refactor box and loss for dIoU functions + fix half tests. * [FIX] Precommits fix. * [ENH] Some improvement for the distance IoU tests thanks to code review. * [ENH] Upcast in distance boxes computation to avoid overflow. * [ENH] Revert the refactor of distance IoU loss back since it introduced a bug and can be slow. * Precommit fix. * [FIX] Few changes introduced by merge conflict. * Add code reference * Fix test Co-authored-by: -
Nicolas Hug authored
-
Nicolas Hug authored
* Fixing AlexNet references * Fix doc rendering
-
YosuaMichael authored
* Add resnext101_64x4d model definition * Add test for resnext101_64x4d * Add resnext101_64x4d weight * Update checkpoint to use EMA weigth * Add quantization model signature for resnext101_64x4d * Fix class name and update accuracy using 1 gpu and batch_size=1 * Apply ufmt * Update the quantized weight and accuracy that we still keep the training log * Add quantized expect file * Update docs and fix acc1 * Add recipe for quantized to PR * Update models.rst
-
- 05 May, 2022 1 commit
-
-
Aditya Oke authored
* Add mask rcnn to instance segmentation * Add docs a use nicolas suggestion * remov param * remov docsstring, edit docs * Add one section level Co-authored-by:
-
- 04 May, 2022 2 commits
-
-
Nicolas Hug authored
* Document Keypoint RCNN separately * Move Keypoint detection into its own section * ufmt
-
Joao Gomes authored
* Remove references to non-existing weights * Update torchvision/models/mnasnet.py Co-authored-by:
-
- 03 May, 2022 8 commits
-
-
puhuk authored
* docs: Added SSD300 to the new doc To resolve #5897 * Minor fixes * ufmt Co-authored-by: -
Joao Gomes authored
* updating mnasnet docs * fix bug * fix lint * fix end of file * fix linting * Add missing weights Co-authored-by: -
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:
-
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:
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:
-
- 28 Apr, 2022 3 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
-
Aditya Oke authored
* Start adding * Finish adding docs * Fix other bugs * Update torchvision/models/mobilenetv3.py * Update torchvision/models/mobilenetv3.py Co-authored-by:
-
