bert/research-project
Archived
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Set defaults for parameters in launcher.

Browse Source 
The launcher assumes it is running in a subfolder of the project root,
and then checks some default paths for the parameters of the launcher.

Currently, it loads up the ilsvrc12 configuration in conjunction with
Alexnet. Maube this can change in the future to sommething more
interesting?
This commit is contained in:
Bert Peters
2018-04-09 13:38:42 +02:00
parent 3f3c95fd13
commit 2c373fbd1e
1 changed files with 23 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/tools/launcher.cpp
View File

@@ -5,6 +5,11 @@
using namespace std::string_literals;
bool file_exists(const char* path, unsigned int mode = F_OK)
{
return access(path, mode) == 0;
}
bool executable_exists(std::string_view dir, std::string_view executable)
{
if (dir.size() + executable.size() + 1 >= PATH_MAX) {
@@ -17,7 +22,7 @@ bool executable_exists(std::string_view dir, std::string_view executable)
ptr = std::copy(executable.begin(), executable.end(), ptr + 1);
*ptr = '\0';
return access(path_buf, F_OK | X_OK) == 0;
return file_exists(path_buf, F_OK | X_OK);
}
auto file_filter_for_extension(std::string_view extension)
@@ -70,6 +75,13 @@ void float_parameter(std::vector<char *> &argv, std::string_view flag, double va
argv.push_back(wrap_string(buffer));
}
void set_default_path(Gtk::FileChooserButton& chooser, const char* path)
{
if (file_exists(path)) {
chooser.set_filename(path);
}
}
class Launcher : public Gtk::Window {
public:
Launcher();
@@ -119,13 +131,20 @@ Launcher::Launcher()
// Configure all widgets
fmriChooser.set_hexpand(true);
findExecutable();
modelChooser.add_filter(file_filter_for_extension("prototxt"));
weightsChooser.add_filter(file_filter_for_extension("caffemodel"));
labelChooser.add_filter(file_filter_for_extension("txt"));
meansChooser.add_filter(file_filter_for_extension("binaryproto"));
pathColor.set_use_alpha(true);
// Set the default paths if called from the expected place
findExecutable();
set_default_path(modelChooser, "../data/models/alexnet/model-dedup.prototxt");
set_default_path(weightsChooser, "../data/models/alexnet/bvlc_alexnet.caffemodel");
set_default_path(labelChooser, "../data/ilsvrc12/synset_words.txt");
set_default_path(meansChooser, "../data/ilsvrc12/imagenet_mean.binaryproto");
set_default_path(inputChooser, "../data/samples");
// Configure grid display options
grid.set_row_spacing(2);
grid.set_column_spacing(2);
@@ -176,8 +195,8 @@ void Launcher::start()
color_string(pathColor),
};
float_parameter(argv, "--layer-opacity", layerTransparancy.get_adjustment()->get_value());
float_parameter(argv, "--interaction-opacity", interactionTransparancy.get_adjustment()->get_value());
float_parameter(argv, "--layer-opacity", layerTransparancy.get_value());
float_parameter(argv, "--interaction-opacity", interactionTransparancy.get_value());
if (labelChooser.get_file()) {
argv.push_back(wrap_string("-l"));