Usage & Documentation

A user’s guide as well as full documentation of the library is available at ReadTheDocs. Some quick examples:

import oflibpytorch as of

shape = (300, 400)
transform = [['rotation', 200, 150, -30]]

# Make a flow field and display it
flow = of.Flow.from_transforms(transform, shape)
flow.show()

# Alternative option without using the custom flow class
flow = of.from_transforms(transform, shape, 't')
of.show_flow(flow)

Above: Visualisation of optical flow representing a rotation

# Combine sequentially with another flow field, display the result
flow_2 = of.Flow.from_transforms([['translation', 40, 0]], shape)
result = of.combine_flows(flow, flow_2, mode=3)
result.show(show_mask=True, show_mask_borders=True)

# Alternative option without using the custom flow class
flow_2 = of.from_transforms([['translation', 40, 0]], shape, 't')
result = of.combine_flows(flow, flow_2, mode=3, ref='t')
of.show_flow(result)  # Note: no way to show the valid flow area (see documentation)

Above: Visualisation of optical flow representing a rotation, translated to the right, using the custom flow class

result.show_arrows(show_mask=True, show_mask_borders=True)

# Alternative option without using the custom flow class
of.show_flow_arrows(result, 't')  # Note: again no way to show the valid flow area

Above: Visualisation of optical flow representing a rotation, translated to the right, using the custom flow class

Installation

In order for oflibpytorch to work, the python environment needs to contain a PyTorch installation. To enable GPU usage, the CUDA Toolkit is required as well. As it is difficult to guarantee an automatic installation via pip will use the correct versions and work on all operating systems, it is left to the user to install PyTorch and Cudatoolkit independently. The easiest route is a virtual conda environment and the recommended install command from the PyTorch website, configured for the user’s specific system. To install oflibpytorch itself, use the following command:

pip install oflibpytorch

Testing

Oflibpytorch contains a large number of tests to verify it is working as intended. Use the command line to navigate to oflibpytorch/test and run the following code:

python -m unittest discover .

The tests will take several minutes to run. Successful completion will be marked with OK.

Contribution & Support

• Source Code: https://github.com/RViMLab/oflibpytorch

• Issue Tracker: https://github.com/RViMLab/oflibpytorch/issues

License

Copyright (c) 2022 Claudio S. Ravasio, PhD student at University College London (UCL), research assistant at King’s College London (KCL), supervised by:

• Dr Christos Bergeles, PI of the Robotics and Vision in Medicine (RViM) lab in the School of Biomedical Engineering & Imaging Sciences (BMEIS) at King’s College London (KCL)

• Prof Lyndon Da Cruz, consultant ophthalmic surgeon, Moorfields Eye Hospital, London UK

This code is licensed under the MIT License.

If you use this code, please acknowledge us with the following citation:

@article{ravasio_oflib,
  title     = {oflibnumpy {\&} oflibpytorch: Optical Flow Handling and Manipulation in Python},
  author    = {Ravasio, Claudio S. and Da Cruz, Lyndon and Bergeles, Christos},
  journal   = {Journal of Open Research Software (JORS)},
  year      = {2021},
  volume    = {9},
  publisher = {Ubiquity Press, Ltd.},
  doi       = {10.5334/jors.380}
}