Skip to content

GitLab

Switch branch/tag
  1. 12 Feb, 2021 2 commits
  3. 11 Feb, 2021 2 commits
  5. 10 Feb, 2021 1 commit
  7. 09 Feb, 2021 1 commit
  9. 04 Feb, 2021 3 commits
    • Nicolas Hug's avatar
      Add ops-cpp target to torchvision (#3350) · aa264980
      Nicolas Hug authored 
      

Summary:
This diff adds a new target to torchvision which enables users to use torchvision ops from C++.

For now, the `cpp_library` is not used by the `python_cpp_library`. We should instead refactor the logic in torchvision to directly use `cpp_library` instead.

There is currently an inconsistency between fbcode and OSS users. OSS users can import torchvision via
```
#include <torchvision/vision.h>
```
while fbcode users need to do
```
#include <torchvision/csrc/vision.h>
```
It would be good to fix this discrepancy in the future.

I didn't directly use `test_frcnn_tracing.cpp` due to complications for getting the `.pt` file in a way that works for both OSS and fbcode, so instead we added a self-contained test that should validate that the torchvision ops are properly registered and visible to JIT

Reviewed By: datumbox

Differential Revision: D26225669

fbshipit-source-id: 5dd9fb98dd58e854f95806e4860d02f54fc04ea4
Co-authored-by: default avatarFrancisco Massa <fmassa@fb.com>
      aa264980
    • Philip Meier's avatar
      Add tests for the STL10 dataset (#3345) · bda17ddf
      Philip Meier authored 
      

* extract some functionality from places365 fakedata for common use

* add a common DatasetTestcase

* add fakedata generation and tests for STL10

* lint
Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      bda17ddf
    • Philip Meier's avatar
      Improve download tests (#3346) · c645f9d2
      Philip Meier authored 
      

* make timeout a parameter

* check redirects
Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      c645f9d2
  11. 03 Feb, 2021 1 commit
    • Philip Meier's avatar
      Add download tests for remaining datasets (#3338) · c5d6f1f3
      Philip Meier authored 
      

* kmnist

* emnist

* qmnist

* omniglot

* phototour

* sbdataset

* sbu

* semeion

* stl10

* svhn

* usps

* cifar100

* enable download logging for google drive

* celeba

* widerface

* lint

* add timeout logic

* lint

* debug CI connection to problematic server

* set timeout for ping

* [ci skip] remove ping

* revert debugging

* disable requests to problematic server

* re-enable all other tests
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      c5d6f1f3
  13. 02 Feb, 2021 4 commits
  15. 01 Feb, 2021 1 commit
  17. 28 Jan, 2021 3 commits
  19. 27 Jan, 2021 2 commits
  21. 25 Jan, 2021 1 commit
  23. 20 Jan, 2021 1 commit
  25. 19 Jan, 2021 1 commit
  27. 18 Jan, 2021 1 commit
    • Vasilis Vryniotis's avatar
      Add MobileNetV3 architecture for Detection (#3253) · bf211dac
      Vasilis Vryniotis authored 
      * Minor refactoring of a private method to make it reusuable.

* Adding a FasterRCNN + MobileNetV3 with & w/o FPN models.

* Reducing Resolution to 320-640 and anchor sizes to 16-256.

* Increase anchor sizes.

* Adding rpn score threshold param on the train script.

* Adding trainable_backbone_layers param on the train script.

* Adding rpn_score_thresh param directly in fasterrcnn_mobilenet_v3_large_fpn.

* Remove fasterrcnn_mobilenet_v3_large prototype and update expected file.

* Update documentation and adding weights.

* Use buildin Identity.

* Fix spelling.
      bf211dac
  29. 15 Jan, 2021 1 commit
  31. 14 Jan, 2021 2 commits
  33. 11 Jan, 2021 1 commit
    • Josh Bradley's avatar
      Add widerface dataset (#2883) · d0063f3d
      Josh Bradley authored 
      

* initial commit of widerface dataset

* comment out old code

* improve parsing of annotation files

* code cleanup and fix docstring comments

* speed up check for quota exceeded

* cleanup print statements

* reformat code and remove print statements

* minor code cleanup and reformatting

* add more comments

* reuse variable

* reverse formatting changes

* fix flake8 errors

* add type annotations

* fix mypy errors

* add a base_folder to root directory

* some formatting fixes

* GDrive threshold does not throw 403 error

* testing new download logic

* cleanup logic for download and integrity check

* use a better variable name

* format fix

* reorder list in docstring

* initial widerface unit test - fails on MD5 check

* use list of dictionaries to store dataset

* fix docstring formatting

* remove unnecessary error checking

* fix type checker error

* revert typo fix

* rename var constants, use file context manager, verify str args

* fix flake8 error

* fix checking target_type argument values

* create uncompressed dataset folders

* cleanup unit tests for widerface

* use correct os function

* add more info to docstring

* disable unittests for windows

* fix _check_integrity logic

* update docstring

* remove citation

* remove target_type option

* fix formatting issue
Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>

* remove comment and add more info to docstring

* update type annotations

* restart CI jobs
Co-authored-by: default avatarJoshua Bradley <jgbrad3@evoforge.org>
Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>
Co-authored-by: default avatarvfdev <vfdev.5@gmail.com>
      d0063f3d
  35. 04 Jan, 2021 1 commit
    • Vasilis Vryniotis's avatar
      Speed up CI runtime (#3189) · 4d2d8bb0
      Vasilis Vryniotis authored 
      * Speedup test_ucf101 (#2623 

* Speedup Cmake builds (#3186)

* Speedup test_autoaugment (#3190)

* Speedup DeformConvTester (#3191)

* Speedup InceptionV3 and GoogleNet on Windows (#3196)
      4d2d8bb0
  37. 15 Dec, 2020 2 commits
  39. 14 Dec, 2020 1 commit
    • Vasilis Vryniotis's avatar
      Implement all AutoAugment transforms + Policies (#3123) · 83171d6a
      Vasilis Vryniotis authored 
      

* Invert Transform (#3104)

* Adding invert operator.

* Make use of the _assert_channels().

* Update upper bound value.

* Remove private doc from invert, create or reuse generic testing methods to avoid duplication of code in the tests. (#3106)

* Create posterize transformation and refactor common methods to assist reuse. (#3108)

* Implement the solarize transform. (#3112)

* Implement the adjust_sharpness transform (#3114)

* Adding functional operator for sharpness.

* Adding transforms for sharpness.

* Handling tiny images and adding a test.

* Implement the autocontrast transform. (#3117)

* Implement the equalize transform (#3119)

* Implement the equalize transform.

* Turn off deterministic for histogram.

* Fixing test. (#3126)

* Force ratio to be float to avoid numeric overflows on blend. (#3127)

* Separate the tests of Adjust Sharpness from ColorJitter. (#3128)

* Add AutoAugment Policies and main Transform (#3142)

* Separate the tests of Adjust Sharpness from ColorJitter.

* Initial implementation, not-jitable.

* AutoAugment passing JIT.

* Adding tests/docs, changing formatting.

* Update test.

* Fix formats

* Fix documentation and imports.

* Apply changes from code review:
- Move the transformations outside of AutoAugment on a separate method.
- Renamed degenerate method for sharpness for better clarity.

* Update torchvision/transforms/functional.py
Co-authored-by: default avatarvfdev <vfdev.5@gmail.com>

* Apply more changes from code review:
- Add InterpolationMode parameter.
- Move all declarations away from AutoAugment constructor and into the private method.

* Update documentation.

* Apply suggestions from code review
Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>

* Apply changes from code review:
- Refactor code to eliminate as any to() and clamp() as possible.
- Reuse methods where possible.
- Apply speed ups.

* Replacing pad.
Co-authored-by: default avatarvfdev <vfdev.5@gmail.com>
Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      83171d6a
  41. 12 Dec, 2020 2 commits
  43. 11 Dec, 2020 1 commit
  45. 10 Dec, 2020 2 commits
    • David Fan's avatar
      Enable ONNX test in circle CI (#3144) · e337103f
      David Fan authored 
      * Enable ONNX test in circle CI
      e337103f
    • Vasilis Vryniotis's avatar
      Restructuring C++ project: (#3146) · 7d831a2f
      Vasilis Vryniotis authored 
      Summary:
* Reduce unnecessary header inclusions in models and io.

* Move autocast to separate folder and hide autograd implementation in an anonymous namespace.

* Moving files in subfolders.

Reviewed By: fmassa

Differential Revision: D25461523

fbshipit-source-id: 756eeb6848aacaa474de4825ed4c1045d17e2cea
      7d831a2f
  47. 08 Dec, 2020 2 commits
  49. 04 Dec, 2020 1 commit