Added app to generate random samples from the model and store them as objs

examples/CMakeLists.txt

@@ -37,7 +37,12 @@ endif()
 add_executable(fit-model fit-model.cpp)
 target_link_libraries(fit-model ${OpenCV_LIBS} ${Boost_LIBRARIES})
 
+# Generate random samples from the model:
+add_executable(generate-obj generate-obj.cpp)
+target_link_libraries(generate-obj ${OpenCV_LIBS} ${Boost_LIBRARIES})
+
 
 # install target:
 install(TARGETS fit-model DESTINATION bin)
+install(TARGETS generate-obj DESTINATION bin)
 install(DIRECTORY ${CMAKE_SOURCE_DIR}/examples/data DESTINATION bin)

examples/generate-obj.cpp

+/*
+ * Eos - A 3D Morphable Model fitting library written in modern C++11/14.
+ *
+ * File: examples/generate-obj.cpp
+ *
+ * Copyright 2016 Patrik Huber
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+#include "eos/morphablemodel/MorphableModel.hpp"
+#include "eos/render/render.hpp"
+#include "eos/fitting/nonlinear_camera_estimation.hpp"
+#include "glm/gtc/matrix_transform.hpp"
+
+#include "opencv2/core/core.hpp"
+#include "opencv2/highgui/highgui.hpp"
+
+#include "boost/program_options.hpp"
+#include "boost/filesystem/path.hpp"
+
+#include <iostream>
+
+using namespace eos;
+namespace po = boost::program_options;
+namespace fs = boost::filesystem;
+using std::vector;
+using std::cout;
+using std::endl;
+
+/**
+ * This app generates random samples from the model and stores them as obj file
+ * as well as outputs a frontal rendering of the sample.
+ *
+ * A list of shape and/or colour coefficients can be specified. Any coefficient
+ * not specified will be set to zero.
+ */
+int main(int argc, char *argv[])
+{
+	fs::path model_file, output_file;
+	vector<float> shape_coefficients, colour_coefficients;
+
+	try {
+		po::options_description desc("Allowed options");
+		desc.add_options()
+			("help",
+				"produce help message")
+			("model", po::value<fs::path>(&model_file)->required(),
+				"an eos .bin Morphable Model file")
+			("shape-coeffs", po::value<vector<float>>(&shape_coefficients)->multitoken(),
+				"optional parameter list of shape coefficients. All not specified will be set to zero. E.g.: '--shape-coeffs 0.0 1.5'. If omitted, the mean is used.")
+			("colour-coeffs", po::value<vector<float>>(&colour_coefficients)->multitoken(),
+				"optional parameter list of colour coefficients. All not specified will be set to zero. E.g.: '--colour-coeffs 0.0 1.5'. If omitted, the mean is used.")
+			("output", po::value<fs::path>(&output_file)->default_value("output.obj"),
+				"name of the output obj file (including .obj). Can be a full path.")
+			;
+
+		po::variables_map vm;
+		// disabling short options to allow negative values for the coefficients, e.g. '--shape-coeffs 0.0 -1.5'
+		po::store(po::parse_command_line(argc, argv, desc, po::command_line_style::unix_style ^ po::command_line_style::allow_short), vm);
+		if (vm.count("help")) {
+			cout << "Usage: generate-obj [options]" << endl;
+			cout << desc;
+			return EXIT_SUCCESS;
+		}
+		po::notify(vm);
+	}
+	catch (const po::error& e) {
+		cout << "Error while parsing command-line arguments: " << e.what() << endl;
+		cout << "Use --help to display a list of options." << endl;
+		return EXIT_SUCCESS;
+	}
+
+	morphablemodel::MorphableModel morphable_model = morphablemodel::load_model(model_file.string());
+
+	if (shape_coefficients.size() < morphable_model.get_shape_model().get_num_principal_components()) {
+		shape_coefficients.resize(morphable_model.get_shape_model().get_num_principal_components());
+	}
+
+	if (colour_coefficients.size() < morphable_model.get_color_model().get_num_principal_components()) {
+		colour_coefficients.resize(morphable_model.get_color_model().get_num_principal_components());
+	}
+
+	render::Mesh sample_mesh = morphable_model.draw_sample(shape_coefficients, colour_coefficients); // if one of the two vectors is empty, it uses get_mean()
+
+	render::write_obj(sample_mesh, output_file.string());
+	cv::Mat rendering;
+	std::tie(rendering, std::ignore) = render::render(sample_mesh, cv::Mat::eye(4, 4, CV_32FC1), fitting::to_mat(glm::ortho(-130.0f, 130.0f, -130.0f, 130.0f)), 512, 512, boost::none, true, false, false);
+	output_file.replace_extension(".png");
+	cv::imwrite(output_file.string(), rendering);
+
+	cout << "Wrote the generated obj and a rendering to files with basename " << output_file << "." << endl;
+
+	return EXIT_SUCCESS;
+}