Esempio n. 1
0
MURAUER is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with MURAUER.  If not, see <http://www.gnu.org/licenses/>.
"""

# Project specific
import data.basetypes as basetypes
# Other libs
import numpy as np

args = basetypes.Arguments()

# How many (CPU) workers for loading data
args.num_loader_workers = 5

# Model architecture specific parameters
# Number of (base-)feature channels (usually increased after downsampling)
args.num_features = 32
# Number of latent representation/embedding dimensions
args.num_bottleneck_dim = 1024
# Number of additional input dimensions for the view prediction decoder
args.num_cond_dims = 3
# Architecture of the discriminator
args.discriminator_type = basetypes.DiscriminatorNetType.RESNET

# Learning-rate decay
Esempio n. 2
0
PreView is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with PreView.  If not, see <http://www.gnu.org/licenses/>.
"""

# Project specific
import data.basetypes as basetypes
from data.LoaderFactory import DatasetType
from data.NyuHandPoseDataset import NyuAnnoType

args_data = basetypes.Arguments()

args_data.dataset_type = DatasetType.NYU

# Change to point to the original NYU dataset
args_data.nyu_data_basepath = "/path/to/NyuDataset/original_data"

# If a "cache" should be used (=> faster loading/training), change the path
args_data.use_pickled_cache = False
args_data.nyu_data_basepath_pickled = "/path/to/NyuDataset/original_data_pickled"

args_data.input_cam_id_train = 1
args_data.input_cam_id_test = 1

# frame IDs
args_data.id_start_train, args_data.id_end_train = 29116, 72756  # 0-based IDs (largest part with approx. same setup)