SYMBOL INDEX (109 symbols across 18 files)

FILE: dataset/augmentation.py
  class PairCompose (line 18) | class PairCompose(T.Compose):
    method __call__ (line 19) | def __call__(self, *x):
  class PairApply (line 25) | class PairApply:
    method __init__ (line 26) | def __init__(self, transforms):
    method __call__ (line 29) | def __call__(self, *x):
  class PairApplyOnlyAtIndices (line 33) | class PairApplyOnlyAtIndices:
    method __init__ (line 34) | def __init__(self, indices, transforms):
    method __call__ (line 38) | def __call__(self, *x):
  class PairRandomAffine (line 42) | class PairRandomAffine(T.RandomAffine):
    method __init__ (line 43) | def __init__(self, degrees, translate=None, scale=None, shear=None, re...
    method __call__ (line 47) | def __call__(self, *x):
  class PairRandomHorizontalFlip (line 55) | class PairRandomHorizontalFlip(T.RandomHorizontalFlip):
    method __call__ (line 56) | def __call__(self, *x):
  class RandomBoxBlur (line 62) | class RandomBoxBlur:
    method __init__ (line 63) | def __init__(self, prob, max_radius):
    method __call__ (line 67) | def __call__(self, img):
  class PairRandomBoxBlur (line 74) | class PairRandomBoxBlur(RandomBoxBlur):
    method __call__ (line 75) | def __call__(self, *x):
  class RandomSharpen (line 82) | class RandomSharpen:
    method __init__ (line 83) | def __init__(self, prob):
    method __call__ (line 87) | def __call__(self, img):
  class PairRandomSharpen (line 93) | class PairRandomSharpen(RandomSharpen):
    method __call__ (line 94) | def __call__(self, *x):
  class PairRandomAffineAndResize (line 100) | class PairRandomAffineAndResize:
    method __init__ (line 101) | def __init__(self, size, degrees, translate, scale, shear, ratio=(3./4...
    method __call__ (line 111) | def __call__(self, *x):
  class RandomAffineAndResize (line 139) | class RandomAffineAndResize(PairRandomAffineAndResize):
    method __call__ (line 140) | def __call__(self, img):

FILE: dataset/images.py
  class ImagesDataset (line 6) | class ImagesDataset(Dataset):
    method __init__ (line 7) | def __init__(self, root, mode='RGB', transforms=None):
    method __len__ (line 13) | def __len__(self):
    method __getitem__ (line 16) | def __getitem__(self, idx):

FILE: dataset/sample.py
  class SampleDataset (line 4) | class SampleDataset(Dataset):
    method __init__ (line 5) | def __init__(self, dataset, samples):
    method __len__ (line 10) | def __len__(self):
    method __getitem__ (line 13) | def __getitem__(self, idx):

FILE: dataset/video.py
  class VideoDataset (line 6) | class VideoDataset(Dataset):
    method __init__ (line 7) | def __init__(self, path: str, transforms: any = None):
    method __len__ (line 16) | def __len__(self):
    method __getitem__ (line 19) | def __getitem__(self, idx):
    method __enter__ (line 34) | def __enter__(self):
    method __exit__ (line 37) | def __exit__(self, exc_type, exc_value, exc_traceback):

FILE: dataset/zip.py
  class ZipDataset (line 4) | class ZipDataset(Dataset):
    method __init__ (line 5) | def __init__(self, datasets: List[Dataset], transforms=None, assert_eq...
    method __len__ (line 13) | def __len__(self):
    method __getitem__ (line 16) | def __getitem__(self, idx):

FILE: export_torchscript.py
  class MattingRefine_TorchScriptWrapper (line 33) | class MattingRefine_TorchScriptWrapper(nn.Module):
    method __init__ (line 46) | def __init__(self, *args, **kwargs):
    method forward (line 57) | def forward(self, src, bgr):
    method load_state_dict (line 67) | def load_state_dict(self, *args, **kwargs):

FILE: inference_images.py
  function writer (line 118) | def writer(img, path):

FILE: inference_utils.py
  class HomographicAlignment (line 6) | class HomographicAlignment:
    method __init__ (line 11) | def __init__(self):
    method __call__ (line 15) | def __call__(self, src, bgr):

FILE: inference_video.py
  class VideoWriter (line 80) | class VideoWriter:
    method __init__ (line 81) | def __init__(self, path, frame_rate, width, height):
    method add_batch (line 84) | def add_batch(self, frames):
  class ImageSequenceWriter (line 93) | class ImageSequenceWriter:
    method __init__ (line 94) | def __init__(self, path, extension):
    method add_batch (line 100) | def add_batch(self, frames):
    method _add_batch (line 104) | def _add_batch(self, frames, index):

FILE: inference_webcam.py
  class Camera (line 57) | class Camera:
    method __init__ (line 58) | def __init__(self, device_id=0, width=1280, height=720):
    method __update (line 71) | def __update(self):
    method read (line 78) | def read(self):
    method __exit__ (line 82) | def __exit__(self, exec_type, exc_value, traceback):
  class FPSTracker (line 86) | class FPSTracker:
    method __init__ (line 87) | def __init__(self, ratio=0.5):
    method tick (line 91) | def tick(self):
    method get (line 100) | def get(self):
  class Displayer (line 105) | class Displayer:
    method __init__ (line 106) | def __init__(self, title, width=None, height=None, show_info=True):
    method step (line 114) | def step(self, image):
  function cv2_frame_to_cuda (line 145) | def cv2_frame_to_cuda(frame):

FILE: model/decoder.py
  class Decoder (line 6) | class Decoder(nn.Module):
    method __init__ (line 21) | def __init__(self, channels, feature_channels):
    method forward (line 32) | def forward(self, x4, x3, x2, x1, x0):

FILE: model/mobilenet.py
  class MobileNetV2Encoder (line 5) | class MobileNetV2Encoder(MobileNetV2):
    method __init__ (line 13) | def __init__(self, in_channels, norm_layer=None):
    method forward (line 32) | def forward(self, x):

FILE: model/model.py
  class Base (line 13) | class Base(nn.Module):
    method __init__ (line 19) | def __init__(self, backbone: str, in_channels: int, out_channels: int):
    method forward (line 31) | def forward(self, x):
    method load_pretrained_deeplabv3_state_dict (line 37) | def load_pretrained_deeplabv3_state_dict(self, state_dict, print_stats...
  class MattingBase (line 61) | class MattingBase(Base):
    method __init__ (line 86) | def __init__(self, backbone: str):
    method forward (line 89) | def forward(self, src, bgr):
  class MattingRefine (line 101) | class MattingRefine(MattingBase):
    method __init__ (line 140) | def __init__(self,
    method forward (line 161) | def forward(self, src, bgr):

FILE: model/refiner.py
  class Refiner (line 8) | class Refiner(nn.Module):
    method __init__ (line 48) | def __init__(self,
    method forward (line 80) | def forward(self,
    method select_refinement_regions (line 163) | def select_refinement_regions(self, err: torch.Tensor):
    method crop_patch (line 186) | def crop_patch(self,
    method replace_patch (line 227) | def replace_patch(self,
    method compute_pixel_indices (line 255) | def compute_pixel_indices(self,

FILE: model/resnet.py
  class ResNetEncoder (line 5) | class ResNetEncoder(ResNet):
    method __init__ (line 19) | def __init__(self, in_channels, variant='resnet101', norm_layer=None):
    method forward (line 34) | def forward(self, x):

FILE: model/utils.py
  function load_matched_state_dict (line 1) | def load_matched_state_dict(model, state_dict, print_stats=True):

FILE: train_base.py
  function train (line 70) | def train():
  function compute_loss (line 219) | def compute_loss(pred_pha, pred_fgr, pred_err, true_pha, true_fgr):
  function random_crop (line 228) | def random_crop(*imgs):
  function valid (line 239) | def valid(model, dataloader, writer, step):

FILE: train_refine.py
  function train_worker (line 80) | def train_worker(rank, addr, port):
  function compute_loss (line 250) | def compute_loss(pred_pha_lg, pred_fgr_lg, pred_pha_sm, pred_fgr_sm, pre...
  function random_crop (line 265) | def random_crop(*imgs):
  function valid (line 278) | def valid(model, dataloader, writer, step):