Commits · 375e4ab21ea3bab29d8c22d5afb1744f84e7acf0 · OpenDAS / vision

15 Mar, 2022 1 commit

Replace asserts with exceptions (#5587) · 289fce29

Joao Gomes authored Mar 15, 2022



* 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: Nicolas Hug <contact@nicolas-hug.com>

* Update references/classification/transforms.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update references/optical_flow/transforms.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update references/optical_flow/transforms.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update references/optical_flow/transforms.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* make value checks more pythonic:

* Update references/optical_flow/transforms.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* make value checks more pythonic

* make more checks pythonic

* fix bug

* appy ufmt

* fix tracing issues

* fib typos

* fix lint

* remove unecessary f-strings

* fix bug

* Update torchvision/datasets/mnist.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update torchvision/datasets/mnist.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update torchvision/ops/boxes.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update torchvision/ops/poolers.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update torchvision/utils.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* address PR comments

* Update torchvision/io/_video_opt.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update torchvision/models/detection/generalized_rcnn.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update torchvision/models/feature_extraction.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* Update torchvision/models/optical_flow/raft.py
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

* address PR comments

* addressing further pr comments

* fix bug

* remove unecessary else

* apply ufmt

* last pr comment

* replace RuntimeErrors
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>

289fce29

25 Feb, 2022 1 commit

Add Conv2dNormActivation and Conv3dNormActivation Blocks (#5445) · f15ba56f

Aditya Oke authored Feb 25, 2022



* Add ops.conv3d

* Refactor for conv2d and 3d

* Refactor

* Fix bug

* Addres review

* Fix bug

* nit fix

* Fix flake

* Final fix

* remove documentation

* fix linter

* Update all the implementations to use new Conv

* Small doc fix
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: Joao Gomes <jdsgomes@fb.com>

f15ba56f

24 Feb, 2022 1 commit
- Revert exposing convNormActivation in ops (#5472) · 55a8300b
  Joao Gomes authored Feb 24, 2022
  
  55a8300b
09 Feb, 2022 1 commit

Add example for optical flow visualizaition and RAFT (#5316) · c39c23ed

Aditya Oke authored Feb 09, 2022



* Start adding example

* Add thumbnail and text

* Replace video

* Improve

* Change default weights of RAFT model builders

* WIP

* WIP

* update handle_legacy_interface input

* lots of stuff

* Oops, wrong default

* Typo

* NITs

* Reduce image size

* Update gallery/plot_optical_flow.py
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Remove link to profile

* Update gallery/plot_optical_flow.py
Co-authored-by: Aditya Oke <47158509+oke-aditya@users.noreply.github.com>

* Address comments

* Nits

* Revert "Remove link to profile"

This reverts commit 2c7a468c2bb0e14d4f39070524d216cbc153bf1d.
Co-authored-by: Nicolas Hug <nicolashug@fb.com>
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

c39c23ed

07 Feb, 2022 1 commit

Change default weights of RAFT model builders (#5381) · 97eddc5d

Nicolas Hug authored Feb 07, 2022

* Change default weights of RAFT model builders

* update handle_legacy_interface input

* Oops, wrong default

97eddc5d

02 Feb, 2022 1 commit
- Move all weight initializations from private methods to constructors (#5331) · 4ae20e53
  Vasilis Vryniotis authored Feb 02, 2022
```
* Move weight initialization in constructors.

* Fixing mypy for ViT.

* remove unnecessary import
```
  4ae20e53
27 Dec, 2021 1 commit
- Update code to Python 3.7 compliance (#5125) · 922db308
  Aditya Oke authored Dec 27, 2021
```
* Upgrade to py37

* Update
```
  922db308
16 Dec, 2021 1 commit

Simplified usage log API (#5095) · eac3dc7b

Kai Zhang authored Dec 16, 2021



* log API v3

* make torchscript happy

* make torchscript happy

* add missing logs to constructor

* log ops C++ API as well

* fix type hint

* check function with isinstance
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

eac3dc7b

09 Dec, 2021 3 commits
- revamp log api usage method (#5072) · d716c426
  Kai Zhang authored Dec 09, 2021
```
* revamp log api usage method
```
  d716c426
- Add our weights for raft_small on C+T (#5076) · 2508a020
  Nicolas Hug authored Dec 09, 2021
```
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
```
  2508a020
- Add pretrained weights for raft_small from original paper (#5070) · 48e2f23c
  Nicolas Hug authored Dec 09, 2021
  
  48e2f23c
08 Dec, 2021 2 commits
- Add pretrained weights on Chairs and Things for raft_large (#5060) · 849d02bc
  Nicolas Hug authored Dec 08, 2021
  
  849d02bc
- Add raft builders and presets in prototypes (#5043) · 5dc1e20b
  Nicolas Hug authored Dec 08, 2021
```
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
```
  5dc1e20b
07 Dec, 2021 1 commit
- Adding logging calls for raft and vit (#5044) · 47bd9620
  Vasilis Vryniotis authored Dec 07, 2021
```
* Adding logging calls for raft and vit

* Linter fix
```
  47bd9620
06 Dec, 2021 1 commit
- Add RAFT model for optical flow (#5022) · 01ffb3ae
  Nicolas Hug authored Dec 06, 2021
  
  01ffb3ae