def main(_):
from shapenet.core import to_cat_id
from shapenet.core.renderings.renderings_manager import get_base_manager
from shapenet.core.frustrum_voxels import create_frustrum_voxels
from shapenet.core.frustrum_voxels import create_temp_frustrum_voxels
from shapenet.core.voxels.config import get_config
if FLAGS.src_voxel_dim is None:
FLAGS.src_voxel_dim = FLAGS.voxel_dim
voxel_config = get_config(
FLAGS.src_voxel_dim, alt=False).filled(FLAGS.fill_alg)
manager = get_base_manager(
dim=FLAGS.dim, turntable=FLAGS.turntable,
n_views=FLAGS.n_views)
cats = FLAGS.cat
if cats is None:
from shapenet.r2n2 import get_cat_ids
cats = get_cat_ids()
for cat in cats:
cat_id = to_cat_id(cat)
args = manager, voxel_config, FLAGS.voxel_dim, cat_id
if FLAGS.temp_only:
create_temp_frustrum_voxels(*args, compression=FLAGS.compression)
else:
create_frustrum_voxels(*args, compression=FLAGS.compression)
def main(_):
from shapenet.core.voxels.config import get_config
from shapenet.core.voxels.datasets import get_manager, convert
from shapenet.core import to_cat_id
config = get_config(FLAGS.voxel_dim, alt=FLAGS.alt)
if FLAGS.cat is None:
# from shapenet.r2n2 import get_cat_ids
raise ValueError('Must provide at least one cat to convert.')
if FLAGS.fill is not None:
config = config.filled(FLAGS.fill)
kwargs = dict(config=config, key=FLAGS.format)
safe_update(kwargs, compression=FLAGS.compression, shape_key=FLAGS.shape)
src_kwargs = dict()
safe_update(src_kwargs,
key=FLAGS.src_format,
compression=FLAGS.src_compression,
shape_key=FLAGS.src_shape)
for cat in FLAGS.cat:
dst = get_manager(cat_id=to_cat_id(cat), **kwargs)
convert(dst,
overwrite=FLAGS.overwrite,
delete_src=FLAGS.delete_src,
**src_kwargs)
def main(_):
from shapenet.core.voxels.config import get_config
from shapenet.core import to_cat_id
from shapenet.r2n2 import get_cat_ids
config = get_config(FLAGS.voxel_dim, alt=FLAGS.alt)
fill = FLAGS.fill
if fill is not None:
config = config.filled(fill)
if FLAGS.cat is None:
cat_ids = get_cat_ids()
else:
cat_ids = [to_cat_id(c) for c in FLAGS.cat]
if FLAGS.fill is not None:
config = config.filled(FLAGS.fill)
for cat_id in cat_ids:
config.create_voxel_data(cat_id)
def main(_):
from shapenet.core.voxels.config import get_config
from shapenet.core import to_cat_id
from shapenet.core import get_example_ids
config = get_config(FLAGS.voxel_dim, alt=FLAGS.alt)
fill = FLAGS.fill
if fill is not None:
config = config.filled(fill)
if FLAGS.cat is None:
raise ValueError('Must provide at least one cat to convert.')
if FLAGS.fill is not None:
config = config.filled(FLAGS.fill)
cat_id = to_cat_id(FLAGS.cat)
example_ids = FLAGS.example_id
if example_ids is None:
example_ids = get_example_ids(cat_id)
config.create_voxel_data(cat_id, example_ids)
def main(_):
from progress.bar import IncrementalBar
import numpy as np
from shapenet.core import to_cat_id
from shapenet.core.renderings.renderings_manager import get_base_manager
from shapenet.core.frustrum_voxels import get_frustrum_voxels_data
from shapenet.core.frustrum_voxels import GROUP_KEY
from shapenet.core.voxels.config import get_config
from util3d.voxel import rle
voxel_config = get_config(FLAGS.src_voxel_dim,
alt=False).filled('orthographic')
manager = get_base_manager(dim=FLAGS.dim,
turntable=FLAGS.turntable,
n_views=FLAGS.n_views)
cats = FLAGS.cat
if cats is None:
from shapenet.r2n2 import get_cat_ids
cats = get_cat_ids()
expected_length = FLAGS.voxel_dim**3
for ci, cat in enumerate(cats):
# if ci >= 3:
# continue
cat_id = to_cat_id(cat)
print('Checking cat %s: %d / %d' % (cat_id, ci + 1, len(cats)))
with get_frustrum_voxels_data(manager.root_dir, voxel_config,
FLAGS.voxel_dim, cat_id) as root:
data = root[GROUP_KEY]
ne, nr = data.shape[:2]
bar = IncrementalBar(max=ne)
for i in range(ne):
bar.next()
di = np.array(data[i])
for j in range(nr):
actual_length = rle.length(di[j])
# actual_length = len(rle.rle_to_dense(di[j]))
if actual_length != expected_length:
raise ValueError(
'Incorrect length at %s, %d, %d\n'
'Expected %d, got %d' %
(cat_id, i, j, expected_length, actual_length))
bar.finish()
from util3d.transform.nonhom import get_eye_to_world_transform
from util3d.voxel.binvox import DenseVoxels
from shapenet.core import get_example_ids, to_cat_id
from shapenet.core.renderings.renderings_manager import get_base_manager
from shapenet.core.voxels.config import get_config
from shapenet.core.voxels.datasets import get_dataset as get_voxel_dataset
import time
cat = 'plane'
voxel_dim = 64
ray_shape = (32,)*3
view_index = 0
cat_id = to_cat_id(cat)
config = get_config(voxel_dim, alt=False).filled('orthographic')
voxel_dataset = get_voxel_dataset(
config, cat_id, id_keys=True, key='rle', compression='lzf')
image_manager = get_base_manager(dim=256)
n_renderings = image_manager.get_render_params()['n_renderings']
f = 32 / 35
example_ids = get_example_ids(cat_id)
with voxel_dataset:
for example_id in example_ids:
start = time.time()
dense_data = voxel_dataset[example_id].dense_data()
dense_data = dense_data[:, -1::-1]
key = (cat_id, example_id)
cat = 'plane'
voxel_dim = 128
alt = False
fill = 'orthographic'
ds_kwargs = dict(key='rle', compression='lzf', shape_key='pad')
# cat = 'pistol'
# voxel_dim = 32
# alt = True
# fill = None
# ds_kwargs = dict(key='zip')
cat_id = to_cat_id(cat)
example_ids = get_example_ids(cat_id)
config = get_config(voxel_dim, alt=alt)
if fill is not None:
config = config.filled(fill)
with get_dataset(config, cat_id, **ds_kwargs) as dataset:
with RenderConfig(shape=(256, 256),
n_images=8).get_dataset(cat_id,
view_index=5) as render_ds:
for example_id in example_ids:
dense = dataset[example_id].dense_data()
render_ds[example_id].show()
mlab.figure()
vis_sliced(dense, axis_order='xyz')
mlab.figure()
vis_contours(dense, contours=[0.5])
mlab.show()