vlkit package

Subpackages

• vlkit.datasets package
  • Submodules
  • vlkit.datasets.datasets module
  • vlkit.datasets.filelist module
  • Module contents
• vlkit.dense package
  • Submodules
  • vlkit.dense.dense module
  • Module contents
• vlkit.geometry package
  • Submodules
  • vlkit.geometry.distance_transform module
  • Module contents
• vlkit.image package
  • Submodules
  • vlkit.image.image module
  • Module contents
• vlkit.io package
  • Submodules
  • vlkit.io.bytes module
  • vlkit.io.pickle module
  • Module contents
• vlkit.lrscheduler package
  • Submodules
  • vlkit.lrscheduler.base module
  • vlkit.lrscheduler.cosine module
  • vlkit.lrscheduler.step module
  • Module contents
• vlkit.models package
  • Submodules
  • vlkit.models.transformer module
  • Module contents
• vlkit.nms package
  • Submodules
  • vlkit.nms.nms_ext module
  • Module contents
• vlkit.ops package
  • Submodules
  • vlkit.ops.blocks module
  • vlkit.ops.conv module
  • vlkit.ops.distributed module
  • vlkit.ops.divergence module
  • vlkit.ops.drop module
  • vlkit.ops.dual_bn module
  • vlkit.ops.functional module
  • vlkit.ops.nn module
  • vlkit.ops.non_local module
  • vlkit.ops.scale_grad module
  • vlkit.ops.squeeze_excite module
  • Module contents
• vlkit.plt package
  • Submodules
  • vlkit.plt.ticks module
  • Module contents
• vlkit.pytorch package
  • Submodules
  • vlkit.pytorch.loss module
  • vlkit.pytorch.wasserstein module
  • Module contents
• vlkit.transforms package
  • Submodules
  • vlkit.transforms.compose module
  • vlkit.transforms.interpolation module
  • vlkit.transforms.npr module
  • vlkit.transforms.resize module
  • vlkit.transforms.transforms module
  • Module contents

Submodules

vlkit.args module

vlkit.args.str2bool(v)

vlkit.detection module

vlkit.detection.bbox2mask(bbox, size)

convert bounding box to binary mask bbox: a bounding box represeting [x1, y1, x2, y2] size: size of the mask [H, W]

vlkit.detection.box_coverage(bbox, mask)

coverage of a box w.r.t the GT mask bbox: a bounding box represeting [x1, y1, x2, y2] mask: the GT binary mask

vlkit.detection.visualize_bboxes(image, bboxes, width=2, color='red')

vlkit.imagenet_labels module

vlkit.imagenet_labels.imagenet_labels()

vlkit.semseg module

vlkit.semseg.color_encode(labelmap, colors=None)

vlkit.semseg.rand_palette(num_classes=21)

vlkit.utils module

Module contents