FilippoAleotti / mobilePydnet

Pydnet on mobile devices
Apache License 2.0
254 stars 40 forks source link

The new mlmodel #10

Closed dfrumkin closed 4 years ago

dfrumkin commented 4 years ago

Hello! Great work! I have a few questions: Is the new mlmodel the one trained using MiDaS outputs on 450K internet images? What is its output? Looks like an 8-bit image that would be scaled and shifted disparity map. Is that right? What if I want to get the original disparity map (or at least floats and not uint), is there an easy way to modify the mlmodel?

FilippoAleotti commented 4 years ago

Hi, Yes, in the conversion the output type is set to imageType.colorSpace(“GRAYSCALE”), to visualize depth maps in the application as images. If you need float values you can use the json serialized tensorflow model available here

dfrumkin commented 4 years ago

Great, thank you, Filippo!