2 ms·
I built a new Optical Flow AI. Instead of being a cranky old man complaining about other people's mistakes, I decided to just fix things myself. Before Corona
by fxtentacle 6y ago
I built a new Optical Flow AI. Instead of being a cranky old man complaining about other people's mistakes, I decided to just fix things myself.
Before Corona hit, I had created a new data set and given a talk about all the stuff that current state of the art AIs for Optical Flow get wrong. I find the topic interesting, because those bugs directly lead to drones like Skydio crashing into thin wires or branches without leaves.
Plus, I needed optical flow to work for a project, but the state of the art didn't work well enough.
Then I thought "ah f* it" and so I built a new architecture that avoids the issues that I had been whining about.
Now, 36 days of training later, it looks like I might have succeeded :)
https://www.dropbox.com/s/3tfrs9sjm9heypk/example_selflow_hajo.jpg?dl=0 https://www.dropbox.com/s/3tfrs9sjm9heypk/example_selflow_ha...
Top left and right are the two input images. Bottom center is the "ground truth", meaning the result that a perfect AI would produce. Bottom left is one of the State of the Art, according to the Sintel benchmark. Bottom right is my new architecture.
You can see that the State of the Art tends to blur the two persons together and blur them with the background between them and the frame border. My approach uses monocular depth cues to correctly segment them into multiple movement layers.
- fpgaminer 6y agoI'm not familiar with Optical Flow research, and I'm sure you already know, but are you familiar with the U^2-Net architecture (a new variation of U-Nets)? I only bother mentioning it at the risk of being redundant because it's new and I know how easy it is to miss stuff in the fire hose that is machine learning research papers. Its results for their specific task, salience mapping, are incredible for such a simple and lightweight architecture. And from a skim of Optical Flow and your example it seems like it might apply there as well. It should be able to grab the long-ranging features necessary for Optical Flow. Do you have a public dataset that you can share or only a private one? Kinda looks like you built a dataset using Skyrim which ... is brilliant.