PawanratRung commited on
Commit
e08ca65
·
verified ·
1 Parent(s): 32122ff

Delete to_mask.py

Browse files
Files changed (1) hide show
  1. to_mask.py +0 -49
to_mask.py DELETED
@@ -1,49 +0,0 @@
1
- # Copyright (c) Facebook, Inc. and its affiliates.
2
-
3
- from typing import Any, Tuple
4
-
5
- from detectron2.structures import BitMasks, Boxes
6
-
7
- from .base import BaseConverter
8
-
9
- ImageSizeType = Tuple[int, int]
10
-
11
-
12
- class ToMaskConverter(BaseConverter):
13
- """
14
- Converts various DensePose predictor outputs to masks
15
- in bit mask format (see `BitMasks`). Each DensePose predictor output type
16
- has to register its convertion strategy.
17
- """
18
-
19
- registry = {}
20
- dst_type = BitMasks
21
-
22
- @classmethod
23
- # pyre-fixme[14]: `convert` overrides method defined in `BaseConverter`
24
- # inconsistently.
25
- def convert(
26
- cls,
27
- densepose_predictor_outputs: Any,
28
- boxes: Boxes,
29
- image_size_hw: ImageSizeType,
30
- *args,
31
- **kwargs
32
- ) -> BitMasks:
33
- """
34
- Convert DensePose predictor outputs to BitMasks using some registered
35
- converter. Does recursive lookup for base classes, so there's no need
36
- for explicit registration for derived classes.
37
-
38
- Args:
39
- densepose_predictor_outputs: DensePose predictor output to be
40
- converted to BitMasks
41
- boxes (Boxes): bounding boxes that correspond to the DensePose
42
- predictor outputs
43
- image_size_hw (tuple [int, int]): image height and width
44
- Return:
45
- An instance of `BitMasks`. If no suitable converter was found, raises KeyError
46
- """
47
- return super(ToMaskConverter, cls).convert(
48
- densepose_predictor_outputs, boxes, image_size_hw, *args, **kwargs
49
- )