patches()

std::shared_ptr< Core::VKImage > MayaFlux::Yantra::VisionExtractor::patches	(	const std::shared_ptr< Core::VKImage > &	source,
		const Kinesis::Vision::VisionAnalysis &	analysis,
		uint32_t	patch_w,
		uint32_t	patch_h,
		uint32_t	source_w,
		uint32_t	source_h
	)

One fixed-size crop per keypoint, else per track, side by side in one atlas.

Definition at line 569 of file VisionExtractor.cpp.

{
    const auto centers = points_for_analysis(analysis);
    if (centers.empty())
        return nullptr;
 
    const auto count = static_cast<uint32_t>(centers.size());
 
    auto& ctx = image_context();
    use_shader(ctx, "vision_patch_extract.comp.spv", k_wg_image, sizeof(PatchExtractPC));
    ctx.stage_image(source);
    ctx.set_binding_data(3, std::span<const float>(
                                reinterpret_cast<const float*>(centers.data()), centers.size() * 2));
    ctx.set_push_constants(PatchExtractPC {
        .patch_w = patch_w, .patch_h = patch_h, .count = count,
        .src_w = source_w, .src_h = source_h });
    return dispatch_image(ctx, patch_w * count, patch_h);
}

References count, image_context(), patch_h, patch_w, and points_for_analysis().

Referenced by run_operation().

Here is the call graph for this function:

Here is the caller graph for this function: