aboutsummaryrefslogtreecommitdiff
path: root/deeptagger
diff options
context:
space:
mode:
Diffstat (limited to 'deeptagger')
-rw-r--r--deeptagger/README.adoc3
-rw-r--r--deeptagger/deeptagger.cpp5
2 files changed, 4 insertions, 4 deletions
diff --git a/deeptagger/README.adoc b/deeptagger/README.adoc
index 3eb62cd..c42d3f9 100644
--- a/deeptagger/README.adoc
+++ b/deeptagger/README.adoc
@@ -29,8 +29,7 @@ if you plan on using the GPU-enabled options.
$ ./download.sh
$ build/deeptagger models/deepdanbooru-v3-20211112-sgd-e28.model image.jpg
-Very little effort is made to make the project compatible with non-POSIX
-systems.
+The project requires a POSIX-compatible system to build.
Options
-------
diff --git a/deeptagger/deeptagger.cpp b/deeptagger/deeptagger.cpp
index 27b0a91..729eeac 100644
--- a/deeptagger/deeptagger.cpp
+++ b/deeptagger/deeptagger.cpp
@@ -116,7 +116,8 @@ read_config(Config &config, const char *path)
}
read_tags(
- std::filesystem::path(path).replace_extension("tags"), config.tags);
+ std::filesystem::path(path).replace_extension("tags").string(),
+ config.tags);
}
// --- Data preparation --------------------------------------------------------
@@ -720,7 +721,7 @@ main(int argc, char *argv[])
// Load batched images in parallel (the first is for GM, the other for IM).
if (g.batch > 1) {
auto value = std::to_string(
- std::max(std::thread::hardware_concurrency() / g.batch, 1L));
+ std::max(long(std::thread::hardware_concurrency()) / g.batch, 1L));
setenv("OMP_NUM_THREADS", value.c_str(), true);
setenv("MAGICK_THREAD_LIMIT", value.c_str(), true);
}