In [1]:
%pylab inline
%load_ext autoreload
%autoreload 2
%run fix_paths.ipy


Populating the interactive namespace from numpy and matplotlib
/home/femianjc/Projects/srp-boxes

In [2]:
import srp.model.arch

In [11]:
a = srp.model.arch.Architecture(rgb_shape=(3, 64, 64),
                                lidar_shape=(6, 64, 64),
                                fusion=srp.model.arch.FusionOptions.LATE_CAT)

In [12]:
a.fusion


Out[12]:
'late_cat'

In [13]:
a.combined_features

In [16]:
a.fusion_layer


Out[16]:
Linear(in_features=1024, out_features=512, bias=True)

In [ ]: