Skip to content

GitLab

Switch branch/tag
  1. 12 Apr, 2021 3 commits
  3. 10 Apr, 2021 1 commit
  5. 09 Apr, 2021 2 commits
    • Prabhat Roy's avatar
      Added KITTI dataset (#3640) · 7da9afee
      Prabhat Roy authored 
      

* Added KITTI dataset

* Addressed review comments

* Changed type of target to List[Dict] and corrected the data types of the returned values.

* Updated unit test to rely on ImageDatasetTestCase

* Added kitti to dataset documentation

* Cleaned up test and some minor changes

* Made data_url a string instead of a list

* Removed unnecessary try and print
Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      7da9afee
    • Vasilis Vryniotis's avatar
      Unify onnx and JIT resize implementations (#3654) · 1d0b43ea
      Vasilis Vryniotis authored 
      * Make two methods as similar as possible.

* Introducing conditional fake casting.

* Change the casting mechanism.
      1d0b43ea
  7. 08 Apr, 2021 4 commits
    • Vasilis Vryniotis's avatar
      Add missing device info. (#3651) · 07fb8ba7
      Vasilis Vryniotis authored
      07fb8ba7
    • Nicolas Hug's avatar
      Add Quantized version of RoIAlign (#3624) · ad9cc62a
      Nicolas Hug authored 
      * WIP

* clang

* docs

* extracted out common utils

* Use better quantization function and pass tensors as parameters

* proper dequantization

* Some tests

* Dequantization optimization, seems to gain a few ms

* clang-format

* again

* more correct test. Had to remove optimization although it almost works

* Also test aligned=True

* remove useless part

* more docs and comments

* Put back optimization with more robust test

* Added check for index upper bound

* avoid possible overflow

* Move common function into common.h

* oops

* scale=1,zero_point=0 makes more sense

* Force batch size of 1 to prevent any indexingbug

* format

* format again

* updated docstring

* put back description comment for pre_calc_bilinear_interpolate

* revert most changes to docstring as it's taken care of in another PR
      ad9cc62a
    • Nicolas Hug's avatar
      Update to transforms docs (#3646) · 3a278d70
      Nicolas Hug authored 
      

* Fixed return docstrings

* Added some refs and corrected some parts

* more refs, and a note about dtypes
Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      3a278d70
    • Mustafa Bal's avatar
      Added defusedxml to parse untrusted XML data (#3636) · 7f4ae8c6
      Mustafa Bal authored 
      

* Added defusedxml to parse untrusted XML data

* Added typecheck disable for defusedxml
Co-authored-by: default avatarNicolas Hug <nicolashug@fb.com>
      7f4ae8c6
  9. 07 Apr, 2021 7 commits
  11. 06 Apr, 2021 2 commits
  13. 31 Mar, 2021 3 commits
  15. 30 Mar, 2021 11 commits
  17. 29 Mar, 2021 1 commit
  19. 25 Mar, 2021 1 commit
  21. 24 Mar, 2021 4 commits
    • Bruno Korbar's avatar
      Implementing multithreaded video decoding (#3389) · 3bfdb427
      Bruno Korbar authored 
      

* multithreading allowed in stream codec context

* numThreads is passed as a decoder parameter. At this stage code should be unchanged

* enabling multithreading in videoReader API

* moving defaults to header files

* replace long with int64_t because torchscript

* docstring for Num threads

* Enable codec-related heuristics as defaults

* Update torchvision/csrc/io/decoder/stream.cpp
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>

* Fixing build errors

* minor docs

* Linting

* updating defaults for the C++ function calls to be single threaded

* adding special case for single threaded stuff
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      3bfdb427
    • Francisco Massa's avatar
      Bump nightlies to 0.10.0 (#3599) · 38bb2704
      Francisco Massa authored
      38bb2704
    • Philip Meier's avatar
      add tests for MNIST and variants (#3423) · aa365993
      Philip Meier authored 
      * add tests for MNIST and variants

* remove old tests and fakedata generation

* fix default config detection for if dataset has variable keywords

* use split="mnist" as default for EMNIST

* fix QMNIST tests

* lint

* fix special kwargs detection

* Revert "use split="mnist" as default for EMNIST"

This reverts commit 62c9b23597f4a391cad409cbd93edac1b3474acf.

* fix tests

* fix QMNIST test case name

* remove dead code from test

* Revert "remove old tests and fakedata generation"

This reverts commit a285b97c4827566a5bc06c288f5bba8d614a4f7a.

* remove old tests

* readd removed import
      aa365993
    • Philip Meier's avatar
      Improve error handling in make_dataset (#3496) · 0818c682
      Philip Meier authored 
      * factor out find_classes

* use find_classes in video datasets

* adapt old tests
      0818c682
  23. 22 Mar, 2021 1 commit