/home/user/jupyter/env/lib/python3.7/site-packages/torch/cuda/__init__.py:52: UserWarning: CUDA initialization: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx (Triggered internally at  /pytorch/c10/cuda/CUDAFunctions.cpp:100.)
  return torch._C._cuda_getDeviceCount() > 0

Show method

show_binmask[source]

show_binmask(im, ax=None, figsize=None, title=None, ctx=None, cmap=None, norm=None, aspect=None, interpolation=None, alpha=None, vmin=None, vmax=None, origin=None, extent=None, filternorm=True, filterrad=4.0, resample=None, url=None, data=None, **kwargs)

Function to show binary masks with matplotlib

Transform Block

BinaryMasksBlock[source]

BinaryMasksBlock()

A TransformBlock for binary masks

class TensorBinMasks[source]

TensorBinMasks(x, **kwargs) :: TensorImageBase

Tensor class for binary mask representation

class TensorBinMasks2TensorMask[source]

TensorBinMasks2TensorMask(enc=None, dec=None, split_idx=None, order=None) :: Transform

Class to transform binary masks to fastai's TensorMask class to make fastai's transforms available