crops()

std::shared_ptr< Core::VKImage > MayaFlux::Yantra::VisionExtractor::crops	(	const std::shared_ptr< Core::VKImage > &	source,
		const Kinesis::Vision::VisionAnalysis &	analysis,
		uint32_t	tile_w,
		uint32_t	tile_h
	)

Every find_elements box resampled to tile_w x tile_h, side by side in one atlas (tile_w * count, tile_h), in box order.

Definition at line 362 of file VisionExtractor.cpp.

{
    if (!analysis.find_elements || analysis.find_elements->boxes.empty() || tile_w == 0 || tile_h == 0)
        return nullptr;
 
    const auto& boxes = analysis.find_elements->boxes;
    const auto count = static_cast<uint32_t>(boxes.size());
 
    std::vector<float> rects;
    rects.reserve(boxes.size() * 4);
    for (const auto& b : boxes)
        rects.insert(rects.end(), { b.x, b.y, b.w, b.h });
 
    auto& ctx = image_context();
    use_shader(ctx, "vision_region_atlas.comp.spv", k_wg_image, sizeof(AtlasPC));
    ctx.stage_image(source);
    ctx.set_binding_data(3, std::span<const float>(rects));
    ctx.set_push_constants(AtlasPC { .tile_w = tile_w, .tile_h = tile_h, .count = count });
    return dispatch_image(ctx, tile_w * count, tile_h);
}

References b, count, MayaFlux::Kinesis::Vision::VisionAnalysis::find_elements, image_context(), tile_h, and tile_w.

Referenced by run_operation().

Here is the call graph for this function:

Here is the caller graph for this function: