1. 15 Apr, 2021 2 commits
  2. 14 Apr, 2021 2 commits
  3. 13 Apr, 2021 2 commits
    • Yanghan Wang's avatar
      support rebuild train loader when data has expired · 754a95fa
      Yanghan Wang authored
      Summary:
      - store expiration in meta data when loading dat
      
      - use before_train_hook to rebuild data loader when expiration condition is met.
      
      Reviewed By: zisting
      
      Differential Revision: D27683164
      
      fbshipit-source-id: e3e3c6c15eee7c02c7a1bfed5f4d4d0e67d61a4f
      754a95fa
    • Sam Tsai's avatar
      update for supporting variation of coco-text like datasets · 2caa6646
      Sam Tsai authored
      Summary:
      1. Add changes to support variation of datasets
      2. Fix runner to support torchscript export
      
      Reviewed By: wat3rBro
      
      Differential Revision: D26871461
      
      fbshipit-source-id: ec46f7e0d2c14c9b802aec22d78b2a089e962a2f
      2caa6646
  4. 09 Apr, 2021 2 commits
  5. 08 Apr, 2021 1 commit
    • Yanghan Wang's avatar
      remove unused dataset_mapper.py · e47d6a24
      Yanghan Wang authored
      Summary: fbgs register_uri_image_loader, UniversalResourceLoader, _IMAGE_LOADER_REGISTRY returns no results other than this file
      
      Reviewed By: newstzpz
      
      Differential Revision: D27639902
      
      fbshipit-source-id: 52e3bb77dbb547334938b8537d6e1c173405d12d
      e47d6a24
  6. 06 Apr, 2021 2 commits
  7. 05 Apr, 2021 2 commits
    • Owen Wang's avatar
      make prediction count evaluation stable with DDP · cbdd46ab
      Owen Wang authored
      Summary: Prediction count evaluator needs to gather it's state before computing metrics, otherwise when parallelized across N GPUs, we only get metrics computed from 1/N of the dataset, increasing our eval signal's variance.
      
      Reviewed By: wat3rBro
      
      Differential Revision: D27416864
      
      fbshipit-source-id: b2c5334cd5a38bebcd06c6ace1627a6b71645fdd
      cbdd46ab
    • Sam Tsai's avatar
      add typing to transforms · 82f17be0
      Sam Tsai authored
      Summary: Add typing to transform.
      
      Reviewed By: wat3rBro
      
      Differential Revision: D27145140
      
      fbshipit-source-id: 8556427b421bf91a05692a590db175c68c4d6890
      82f17be0
  8. 03 Apr, 2021 2 commits
    • Peizhao Zhang's avatar
      Make data and evaluation visualization optional. · daf7f294
      Peizhao Zhang authored
      Summary:
      Make data and evaluation visualization optional.
      * could return None.
      
      Reviewed By: zhanghang1989, wat3rBro
      
      Differential Revision: D27316632
      
      fbshipit-source-id: 2a85db4815cbf3407a20a74c125dcd52d75167fa
      daf7f294
    • Peizhao Zhang's avatar
      Format changes. · 66df06ef
      Peizhao Zhang authored
      Summary:
      Format changes.
      * [Option] + [Shift] + [F]
      
      Reviewed By: mattcyu1, zhanghang1989, wat3rBro
      
      Differential Revision: D27316555
      
      fbshipit-source-id: 0fc3396eb34d964478cb3551dc73b47412089ccb
      66df06ef
  9. 02 Apr, 2021 1 commit
    • Yanghan Wang's avatar
      new interface for build_d2go_train_loader · 4bae056b
      Yanghan Wang authored
      Summary:
      #Facebook:
      
      `build_d2go_train_loader` will replace `runner.build_detection_train_loader`, currently we call `build_d2go_train_loader` from `runner.build_detection_train_loader` since some runner has there own implementation, we will solve those cases and remove the `runner.build_detection_train_loader` API.
      
      Currently `build_d2go_train_loader` uses `_MAPPED_TRAIN_LOADER_BUILDER_REGISTRY` to support different versions between OSS and FB, not sure if this is a good pattern or not, please comment in the diff if you have better idea.
      
      Reviewed By: zhanghang1989
      
      Differential Revision: D27505681
      
      fbshipit-source-id: b5caf7280a88c2ebccb498097c0b7af51c966fc6
      4bae056b
  10. 31 Mar, 2021 3 commits
    • Kai Zhang's avatar
      Sync quantization callback changes · 1850a632
      Kai Zhang authored
      Reviewed By: newstzpz
      
      Differential Revision: D27255960
      
      fbshipit-source-id: 1699ff23d2bc610dffc0215a90a7c1c17e3783c3
      1850a632
    • Sam Tsai's avatar
      fix bypassed rebase error · 1027896a
      Sam Tsai authored
      Summary: Fixing unit test that was not listed due to rebase error.
      
      Reviewed By: newstzpz, wat3rBro
      
      Differential Revision: D27456322
      
      fbshipit-source-id: 519c5c086adfb19104ed99234f4f476eb34a79bc
      1027896a
    • Tao Xu's avatar
      Train a pix2pix model · bb0ea3d5
      Tao Xu authored
      Summary: Train a pix2pix model on the paired dataset. During inference, it can transfer an source image to the target image.
      
      Reviewed By: newstzpz
      
      Differential Revision: D27371290
      
      fbshipit-source-id: 3141bc6d9e4fe0013f6ea3de3cf998163d286168
      bb0ea3d5
  11. 30 Mar, 2021 3 commits
  12. 29 Mar, 2021 5 commits
    • TannerGilbert's avatar
      updated test utils import in d2go_beginner.ipynb · 4b997f09
      TannerGilbert authored
      Summary:
      In 9d238344, the test utils were moved to the core library, but the import for the create_fake_detection_data_loader inside the d2go_beginner.ipynb wasn't updated.
      
      Pull Request resolved: https://github.com/facebookresearch/d2go/pull/29
      
      Reviewed By: newstzpz
      
      Differential Revision: D27239846
      
      Pulled By: zhanghang1989
      
      fbshipit-source-id: e39df32746b1d1081026f9969bda84e73ac7df55
      4b997f09
    • Yanghan Wang's avatar
      remove d2go.tests package from setup · bcc90f54
      Yanghan Wang authored
      Summary: all utils code are moved to d2go.utils.testing
      
      Reviewed By: newstzpz
      
      Differential Revision: D27209943
      
      fbshipit-source-id: 6c5cb14858155a8ed13478d65ee8e02ef74616d7
      bcc90f54
    • Sanjeev Kumar's avatar
      Dataset to model subclass mapping handling in hands eval pipeline · 1d609cd5
      Sanjeev Kumar authored
      Summary: - Added support for running evaluation for models where the number of subclasses in the model output is less than the number of subclases in the annotated dataset
      
      Reviewed By: vivekn
      
      Differential Revision: D27090466
      
      fbshipit-source-id: 704c438c1bbca333648c0477c412bf3ed79f04e7
      1d609cd5
    • Yanghan Wang's avatar
      initial implementation of build_auto_stream_train_loader · f9c62b77
      Yanghan Wang authored
      Summary: Add `build_auto_stream_train_loader`
      
      Reviewed By: newstzpz
      
      Differential Revision: D27343030
      
      fbshipit-source-id: a79d3ed1ac41fc159d10bb6ff1db74549b645a1c
      f9c62b77
    • Yanghan Wang's avatar
      support using prefetched files in the mapper · bf8d84b2
      Yanghan Wang authored
      Summary: The default mapper may load "file_name" and "sem_seg_file_name" from `dataset_dict`, when prefetching them from manifold, we no longer need to load them because they're already fetched. This diff adds two more fields for holding those pre-fetched data, and make the mapper work in both cases.
      
      Reviewed By: newstzpz
      
      Differential Revision: D26972340
      
      fbshipit-source-id: 63f6dc809d321e149aa5adf9f92c3ace07cbf2a7
      bf8d84b2
  13. 24 Mar, 2021 3 commits
    • Kai Zhang's avatar
      Support evaluate predictor · 6aec097e
      Kai Zhang authored
      Summary:
      Evaluate the predictor generated by previous step.
      This diff modify the lightning_train_net to reuse the evaluation logic by adding a `predictor_path` param.
      This diff also makes Lightning training backend depends on `cfg.MODEL.DEVICE` so that in evaluate_predictor step, user could set backend by changing model device. This is useful for evaluating int8 quantized model.
      
      Reviewed By: newstzpz
      
      Differential Revision: D27150609
      
      fbshipit-source-id: fb72da3e81db932c0fa479350150720143e09a3e
      6aec097e
    • Kai Zhang's avatar
      Support export_predictors · 242b2d37
      Kai Zhang authored
      Summary: As titled.
      
      Reviewed By: newstzpz
      
      Differential Revision: D27074737
      
      fbshipit-source-id: 72f2535fc730a37f5ea8f58aaff88005c28ffc5b
      242b2d37
    • Kai Zhang's avatar
      Simplify Lightning task and model creation · 9051f71a
      Kai Zhang authored
      Summary:
      Given that the way to create D2 (https://github.com/facebookresearch/d2go/commit/465cdb842513eb910aa20fcedea1d2edd15dc7b7)go runner and Lightning task are different, get_class was introduced so that in application we could do:
      ```
      if is Lightning:
          task_cls = get_class(classname)
          task = task_cls(cfg)
      else:
          runner = create_runner(classname)
      ```
      It turns out that we could need to do that in many places: workflow, binaries.
      This diff revert `get_class` and return class in `create_runner` if the class is a Lightning module.
      
      Reviewed By: newstzpz
      
      Differential Revision: D26676595
      
      fbshipit-source-id: c3ce2016d09fe073af4c2dd9f98eea4e59ca621b
      9051f71a
  14. 21 Mar, 2021 1 commit
    • Tao Xu's avatar
      register new dataset for GANs on the fly · 498cd31b
      Tao Xu authored
      Summary: Prepare the launch script for IG, which support registering new dataset for GANs on the fly
      
      Reviewed By: newstzpz
      
      Differential Revision: D27211763
      
      fbshipit-source-id: f79978ceae246ab4f27a8083d25dd50c62dcefab
      498cd31b
  15. 20 Mar, 2021 1 commit
    • Yanghan Wang's avatar
      move test utils to core library · 9d238344
      Yanghan Wang authored
      Summary: Not d2go.tests is not a library for oss, move utils code to d2go.utils.testing
      
      Reviewed By: zhanghang1989
      
      Differential Revision: D26706933
      
      fbshipit-source-id: 85767b66bbb6c67db05e11823beb4840220b2aa3
      9d238344
  16. 18 Mar, 2021 2 commits
  17. 17 Mar, 2021 2 commits
  18. 16 Mar, 2021 3 commits
    • Sam Tsai's avatar
      extend cocotext conversion · b80f2ff5
      Sam Tsai authored
      Summary: Extend conversion to support ids beyond cocotext format where ids are strings.
      
      Reviewed By: newstzpz
      
      Differential Revision: D27018211
      
      fbshipit-source-id: 7282fd4b9a7e9cd19323235ed1a3c3e7b33cb6b4
      b80f2ff5
    • Xiaoliang Dai's avatar
      add more layers to the subclass head · d5c214b9
      Xiaoliang Dai authored
      Reviewed By: sanjeevk42
      
      Differential Revision: D26922320
      
      fbshipit-source-id: 8d6b4bc709d931020483284febede8cf8904d90f
      d5c214b9
    • Georgy Marrero's avatar
      Add d2go data registry support for v2 mask paths · 1c35e46c
      Georgy Marrero authored
      Summary:
      # context
      
      Right now, when we register Person or Hair data we have a mask directory with .PNGs inside and when we register MCS data we have multiple directories <class>_mask. These are 2 separate file structures to maintain.
      
      # this diff
      
      **This diff is to add support for the following file structure for *both* Multi-Class and Single-Class Semantic Segmentation:**
      
      ```
      mask/
        person/
        hair/
        skin/
        beard/
        clothes/
      ```
      
      **without breaking the previous format for Single-Class Semantic Segmentation (`mask/` with .PNGs).**
      
      # why?
      
      With this change, we'll be able to prepare data once (create the file structure and d2go coco jsons) and make it ready-to-be-registered for Single-Class Semantic Segmentation (Person or Hair) and Multi-Class Segmentation.
      
      We'll save on having to prepare the dataset 3 times and managing multiple copies of the same images and annotations. There will be one source of truth. For UGC, we'll be able to enforce privacy deletions with less complications.
      
      Also, this is the format used in the Video Eval tool. So data prepared for training can be used for video evaluation if needed and data prepared for video evaluation can easily be prepared for training.
      
      # to dos:
      
      - ~~Add the v2 path support on the Single-Class Segmentation's `SemSegEvaluator` so that the v2 format can be used for evaluation (sync with wat3rBro for best way to do this)~~
        - **Done.** This happens automatically given the the Single-Class case in the `MultiSemSegEvaluator` class doesn't re-register the datasets. See the first job for an example.
      
      Reviewed By: wat3rBro
      
      Differential Revision: D26795469
      
      fbshipit-source-id: 6ebbb1a430c049dad361d56842a587c76926ee23
      1c35e46c
  19. 12 Mar, 2021 1 commit