Skip to content

Commit 34deced

Browse files
committed
export preprocess_canny
1 parent bd56734 commit 34deced

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

preprocessing.hpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,7 @@ void threshold_hystersis(struct ggml_tensor* img, float highThreshold, float low
162162
}
163163
}
164164

165-
uint8_t* preprocess_canny(uint8_t* img, int width, int height, float highThreshold = 0.08f, float lowThreshold = 0.08f, float weak = 0.8f, float strong = 1.0f, bool inverse = false) {
165+
uint8_t* preprocess_canny(uint8_t* img, int width, int height, float highThreshold, float lowThreshold, float weak, float strong, bool inverse) {
166166
struct ggml_init_params params;
167167
params.mem_size = static_cast<size_t>(10 * 1024 * 1024); // 10
168168
params.mem_buffer = NULL;

stable-diffusion.h

+2
Original file line numberDiff line numberDiff line change
@@ -162,6 +162,8 @@ SD_API void log_printf(sd_log_level_t level, const char* file, int line, const c
162162

163163
SD_API std::string sd_basename(const std::string& path);
164164

165+
SD_API uint8_t* preprocess_canny(uint8_t* img, int width, int height, float highThreshold = 0.08f, float lowThreshold = 0.08f, float weak = 0.8f, float strong = 1.0f, bool inverse = false);
166+
165167
#ifdef __cplusplus
166168
}
167169
#endif

0 commit comments

Comments
 (0)