simpleig / Geo-PIFu

This repository is the official implementation of Geo-PIFu: Geometry and Pixel Aligned Implicit Functions for Single-view Human Reconstruction.
112 stars 17 forks source link

Ram memory out while creating query points. (on Colab PRO High Ram) #24

Closed LazerLikeFocus closed 3 years ago

LazerLikeFocus commented 3 years ago

I am trying for the past 3 days to create query points on colab pro but its crashing. I did try on my pc (high end gaming pc) but it still crashed.

Problem is that the code gets stuck on : https://github.com/simpleig/Geo-PIFu/blob/0b570802d6c46d7230bb505e2a8c3d5a1cf7086a/geopifu/apps/prepare_shape_query.py#L225

My solution I noticed you did 3 things:

  1. Random rotation
  2. Normalization (X~[+-0.333], Y~[+-0.5], Z~[+-0.333])
  3. Changed the space by changing BMIN BMAx to [+-0.33333333 +-0.5 +-0.33333333]

So i tested the on only one of them at a time (replacing the other 2 with the org PIFu defaults). Then i tested 2 of them at a time. The problem still continued.

Then I took the mesh(deephuman dataset #10198) to load inside points using PIFu script:

It worked after i changed https://github.com/simpleig/Geo-PIFu/blob/0b570802d6c46d7230bb505e2a8c3d5a1cf7086a/geopifu/apps/prepare_shape_query.py#L216

to

sample_points = surface_points + np.random.normal(scale=5.0, size=surface_points.shape) #(as given in PIFu for Gaussian noise)

1.Can anyone tell me why this happened ?

  1. I know this is wrong according to geopifu. But still. And can i train geopifu on these query points ?

PLz tell me how can i proceed futher.