Skip to content

GitLab

Switch branch/tag
  1. 05 Oct, 2021 5 commits
  3. 04 Oct, 2021 3 commits
    • Nicolas Hug's avatar
      Fix all outstanding flake8 issues (#4535) · b81d189d
      Nicolas Hug authored
      b81d189d
    • Philip Meier's avatar
      Add ufmt (usort + black) as code formatter (#4384) · 5f0edb97
      Philip Meier authored 
      

* add ufmt as code formatter

* cleanup

* quote ufmt requirement

* split imports into more groups

* regenerate circleci config

* fix CI

* clarify local testing utils section

* use ufmt pre-commit hook

* split relative imports into local category

* Revert "split relative imports into local category"

This reverts commit f2e224cde2008c56c9347c1f69746d39065cdd51.

* pin black and usort dependencies

* fix local test utils detection

* fix ufmt rev

* add reference utils to local category

* fix usort config

* remove custom categories sorting

* Run pre-commit without fixing flake8

* got a double import in merge
Co-authored-by: default avatarNicolas Hug <nicolashug@fb.com>
      5f0edb97
    • Nicolas Hug's avatar
      Relative import for roi_align (#4533) · e45489b1
      Nicolas Hug authored
      e45489b1
  5. 01 Oct, 2021 4 commits
  7. 30 Sep, 2021 3 commits
  9. 29 Sep, 2021 3 commits
    • Kai Zhang's avatar
      Add RegNet Architecture in TorchVision (#4403) · 194a0846
      Kai Zhang authored 
      * initial code

* add SqueezeExcitation

* initial code

* add SqueezeExcitation

* add SqueezeExcitation

* regnet blocks, stems and model definition

* nit

* add fc layer

* use Callable instead of Enum for block, stem and activation

* add regnet_x and regnet_y model build functions, add docs

* remove unused depth

* use BN/activation constructor and ConvBNActivation

* add expected test pkl files

* allow custom activation in SqueezeExcitation

* use ReLU as the default activation

* initial code

* add SqueezeExcitation

* initial code

* add SqueezeExcitation

* add SqueezeExcitation

* regnet blocks, stems and model definition

* nit

* add fc layer

* use Callable instead of Enum for block, stem and activation

* add regnet_x and regnet_y model build functions, add docs

* remove unused depth

* use BN/activation constructor and ConvBNActivation

* reuse SqueezeExcitation from efficientnet

* refactor RegNetParams into BlockParams

* use nn.init, replace np with torch

* update README

* construct model with stem, block, classifier instances

* Revert "construct model with stem, block, classifier instances"

This reverts commit 850f5f3ed01a2a9b36fcbf8405afd6e41d2e58ef.

* remove unused blocks

* support scaled model

* fuse into ConvBNActivation

* make reset_parameters private

* fix type errors

* fix for unit test

* add pretrained weights for 6 variant models, update docs
      194a0846
    • Vasilis Vryniotis's avatar
      Replace MobileNetV3's SqueezeExcitation with EfficientNet's one (#4487) · ff126ae2
      Vasilis Vryniotis authored 
      * Reuse EfficientNet SE layer.

* Deprecating the mobilenetv3.SqueezeExcitation layer.

* Passing the right activation on quantization.

* Making strict named param.

* Set default params if missing.

* Fixing typos.
      ff126ae2
    • Philip Meier's avatar
      cleanup prototype datasets (#4471) · a068602e
      Philip Meier authored 
      

* cleanup image folder

* make shuffling mandatory

* rename parameter in home() function

* don't show builtin list

* make categories optional in dataset info

* use pseudo-infinite buffer size for shuffler
Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      a068602e
  11. 28 Sep, 2021 1 commit
  13. 27 Sep, 2021 2 commits
    • Loi Ly's avatar
      Added gray image support to `adjust_saturation` function (#4480) · f483e71b
      Loi Ly authored 
      * update channels parameter to every calling to check_functional_vs_PIL_vs_scripted

* update adjust_saturation

* update docstrings for functional transformations

* parametrize channels

* update docstring of ColorJitter class

* move channels to class's parameter

* remove testing channels for geometric transforms

* revert redundant changes

* revert redundant changes

* update grayscale test cases for randaugment, autoaugment, trivialaugment

* update docstrings of randaugment, autoaugment, trivialaugment

* update docstring of ColorJitter

* fix adjust_hue's docstring

* change test equal tolerance

* refactor grayscale tests

* make get_grayscale_test_image private
      f483e71b
    • peterbell10's avatar
      Update gpu atomics include path (#4478) · 3e27eb21
      peterbell10 authored 
      

* Update gpu atomics include path

THC is being removed, and this header now lives in ATen.
See pytorch/pytorch#65470.

* Empty commit to trigger build
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      3e27eb21
  15. 24 Sep, 2021 3 commits
  17. 23 Sep, 2021 1 commit
  19. 22 Sep, 2021 1 commit
  21. 21 Sep, 2021 3 commits
  23. 17 Sep, 2021 1 commit
  25. 16 Sep, 2021 1 commit
  27. 15 Sep, 2021 2 commits
    • Vasilis Vryniotis's avatar
      Adding Mixup and Cutmix (#4379) · c8e3b2a5
      Vasilis Vryniotis authored 
      * Add RandomMixupCutmix.

* Add test with real data.

* Use dataloader and collate in the test.

* Making RandomMixupCutmix JIT scriptable.

* Move out label_smoothing and try roll instead of flip

* Adding mixup/cutmix in references script.

* Handle one-hot encoded target in accuracy.

* Add support of devices on tests.

* Separate Mixup from Cutmix.

* Add check for floats.

* Adding device on expect value.

* Remove hardcoded weights.

* One-hot only when necessary.

* Fix linter.

* Moving mixup and cutmix to references.

* Final code clean up.
      c8e3b2a5
    • Prabhat Roy's avatar
      Updated annotation to be a Union of Tensor and List (#4416) · b0962719
      Prabhat Roy authored 
      * Updated annotation to be a Union of Tensor and List

* Updated check_roi_boxes_shape.
      b0962719
  29. 14 Sep, 2021 1 commit
    • Muhammed Abdullah's avatar
      Added LFW Dataset (#4255) · d85aa6d3
      Muhammed Abdullah authored 
      

* Added LFW Dataset

* Added dataset to list in __init__.py

* Updated lfw.py
* Created a common superclass for people and pairs type datatsets
* corrected the .download() method

* Added docstrings and updated datasets.rst

* Wrote tests for LFWPeople and LFWPairs

* Resolved mypy error: Need type annotation for "data"

* Updated inject_fake_data method for LFWPeople

* Updated tests for LFW

* Updated LFW tests and minor changes in lfw.py

* Updated LFW
* Added functionality for 10-fold validation view
* Optimized the code so to replace repeated lines by method in super
  class
* Updated LFWPeople to get classes from all lfw-names.txt rather than
  just the classes fron trainset

* Updated lfw.py and tests
* Updated inject_fake_data method to create 10fold fake data
* Minor changes in docstring and extra_repr

* resolved py lint errors

* Added checksums for annotation files

* Minor changes in test

* Updated docstrings, defaults and minor changes in test

* Removed 'os.path.exists' check
Co-authored-by: default avatarABD-01 <myac931@gmai.com>
Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>
Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      d85aa6d3
  31. 10 Sep, 2021 1 commit
  33. 08 Sep, 2021 2 commits
  35. 06 Sep, 2021 3 commits