SYMBOL INDEX (37 symbols across 9 files)

FILE: evaluate.py
  function evaluate (line 9) | def evaluate(net, dataloader, device, amp):

FILE: hubconf.py
  function unet_carvana (line 4) | def unet_carvana(pretrained=False, scale=0.5):

FILE: predict.py
  function predict_img (line 15) | def predict_img(net,
  function get_args (line 36) | def get_args():
  function get_output_filenames (line 55) | def get_output_filenames(args):
  function mask_to_image (line 62) | def mask_to_image(mask: np.ndarray, mask_values):

FILE: train.py
  function train_model (line 27) | def train_model(
  function get_args (line 170) | def get_args():

FILE: unet/unet_model.py
  class UNet (line 6) | class UNet(nn.Module):
    method __init__ (line 7) | def __init__(self, n_channels, n_classes, bilinear=False):
    method forward (line 25) | def forward(self, x):
    method use_checkpointing (line 38) | def use_checkpointing(self):

FILE: unet/unet_parts.py
  class DoubleConv (line 8) | class DoubleConv(nn.Module):
    method __init__ (line 11) | def __init__(self, in_channels, out_channels, mid_channels=None):
    method forward (line 24) | def forward(self, x):
  class Down (line 28) | class Down(nn.Module):
    method __init__ (line 31) | def __init__(self, in_channels, out_channels):
    method forward (line 38) | def forward(self, x):
  class Up (line 42) | class Up(nn.Module):
    method __init__ (line 45) | def __init__(self, in_channels, out_channels, bilinear=True):
    method forward (line 56) | def forward(self, x1, x2):
  class OutConv (line 71) | class OutConv(nn.Module):
    method __init__ (line 72) | def __init__(self, in_channels, out_channels):
    method forward (line 76) | def forward(self, x):

FILE: utils/data_loading.py
  function load_image (line 16) | def load_image(filename):
  function unique_mask_values (line 26) | def unique_mask_values(idx, mask_dir, mask_suffix):
  class BasicDataset (line 38) | class BasicDataset(Dataset):
    method __init__ (line 39) | def __init__(self, images_dir: str, mask_dir: str, scale: float = 1.0,...
    method __len__ (line 61) | def __len__(self):
    method preprocess (line 65) | def preprocess(mask_values, pil_img, scale, is_mask):
    method __getitem__ (line 93) | def __getitem__(self, idx):
  class CarvanaDataset (line 115) | class CarvanaDataset(BasicDataset):
    method __init__ (line 116) | def __init__(self, images_dir, mask_dir, scale=1):

FILE: utils/dice_score.py
  function dice_coeff (line 5) | def dice_coeff(input: Tensor, target: Tensor, reduce_batch_first: bool =...
  function multiclass_dice_coeff (line 20) | def multiclass_dice_coeff(input: Tensor, target: Tensor, reduce_batch_fi...
  function dice_loss (line 25) | def dice_loss(input: Tensor, target: Tensor, multiclass: bool = False):

FILE: utils/utils.py
  function plot_img_and_mask (line 4) | def plot_img_and_mask(img, mask):