Skip to content

Releases: hmorimitsu/ptlflow

v0.3.2

12 Jun 15:01
542899c
Compare
Choose a tag to compare

Added

  • New models:
    • MemFlow
    • NeuFlow
    • SEA-RAFT
    • SplatFlow
  • New datasets
    • Kubric
    • TartanAir
  • ONNX and TensorRT conversion scripts to RAPIDFlow

Fixed

  • LR scheduler when accumulating gradients

v0.3.1

07 Mar 14:11
c69eeb2
Compare
Choose a tag to compare

Added

  • New models:
    • CCMR
    • LLA-Flow
    • RAPIDFlow
  • FP16 inference in most models
  • Script to plot results

v0.3.0

13 Jan 15:44
185f402
Compare
Choose a tag to compare

This is a major update and introduces breaking changes to v0.2.
The list of changes below is not comprehensive, there may be other changes that are not listed.

Added

  • New models:
    • DIP
    • Flow1D
    • FlowFormer++
    • GMFlow+
    • MatchFlow
    • MS-RAFT+
    • RPKNet
      -SeparableFlow
    • SKFlow
    • VideoFlow
  • New datasets:
    • Middlebury
    • Monkaa
    • Spring
  • Option to use RAFT's alt_cuda_corr for supported models
    • Also added a pure PyTorch version of alt_cuda_corr, which is slower but does not need to be compiled

Fixed

  • Compatibility with PyTorch 2.0, 2.1
  • Compatibility issues with PyTorch Lightning 1.9
  • Resizing augmentation when the valid mask is sparse
  • Models should produce results more similar to the paper
    • HOWEVER, we do not guarantee our results are correct. Use the official implementations for the most accurate results.

Changed

  • Each model now handle its own IO reshaping, instead of using padding for all models
  • speed_benchmark.py becomes model_benchmark.py and records more metrics
  • Renamed model: pwcdcnet -> pwcnet, pwcnet -> pwcnet_nodc
  • Updated requirements, support for many old versions are dropped
    • Important requirements:
      • python>=3.8,<3.12
      • torch>=1.13,<2.2
      • lightning>=1.9.0,<2

v0.2.7

08 Aug 15:14
d5ec04e
Compare
Choose a tag to compare

Fixed

  • Memory leak in FlowMetrics caused when setting full_state_update=False

v0.2.6

01 Aug 02:54
44d3b8a
Compare
Choose a tag to compare

Added

  • Support for AutoFlow dataset
  • New models:
    • CRAFT
    • CSFlow
    • FlowFormer
    • GMFlow
    • GMFlowNet

Fixed

  • Compatibility issues with PyTorch Lightning 1.6

v0.2.5

03 Jan 03:34
26f7533
Compare
Choose a tag to compare

Added

  • FastFlowNet model

Fixed

  • Check if max_steps has negative value (the new standard value in newer PyTorch Lightning versions)

0.2.4

13 Sep 12:16
08a3c9e
Compare
Choose a tag to compare

Added

  • GMA model
  • SCV model

0.2.3

03 Aug 16:28
3fbb0b8
Compare
Choose a tag to compare

Added

  • DICL model
  • LCV with RAFT model

Fixed

  • Check if an input is an array before showing as an image during validation

0.2.2

01 Aug 11:26
Compare
Choose a tag to compare

Added

  • MaskFlownet model
  • Script for generating predictions for the test split of some datasets (currently Sintel and KITTI)

Fixed

  • Inference uses no_grad annotation
  • Threshold values for computing outlier metric and update table of results
  • Some utility functions did not pass through all the keys from the input dicts

0.2.1

12 Jul 16:45
a068a50
Compare
Choose a tag to compare

Added

  • STaRFlow model
  • Interactive plots with Plotly

Fixed

  • Flow values when input images were resized during validation