VCC/HailoImageProcessor
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

removed loop shoving data till it crashed

Browse source
This commit is contained in:
Nicholas Young 2024-10-17 18:13:55 -07:00
parent 6f95699d19
commit 6f4239adb9
1 changed files with 7 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
src/main.zig
View file

@ -7,7 +7,7 @@ const hlo = @cImport({
const Allocator = std.mem.Allocator; const Allocator = std.mem.Allocator;
const assert = std.debug.assert; const assert = std.debug.assert;
const hef_file = "yolov7.hef"; const hef_file = "yolov3.hef";
const max_edge_layers = 32; const max_edge_layers = 32;
@ -93,22 +93,13 @@ pub fn main() !void {
assert(status == hlo.HAILO_SUCCESS); assert(status == hlo.HAILO_SUCCESS);
const frame_count = input_frame_size / @sizeOf(u8); const frame_count = input_frame_size / @sizeOf(u8);
std.debug.print("Trying to create {d} u8s\n", .{ frame_count });
var input_data: [:0]u8 = try alloc.allocSentinel(u8, frame_count, 0); var input_data: [:0]u8 = try alloc.allocSentinel(u8, frame_count, 0);
var prng = std.Random.DefaultPrng.init(blk: { for (0..(frame_count + 1)) |i| {
var seed: u64 = undefined; input_data[i] = 5;
try std.posix.getrandom(std.mem.asBytes(&seed));
break :blk seed;
});
const rand = prng.random();
for (0..frame_count) |i| {
input_data[i] = rand.int(u8);
} }
std.debug.print("Random numbers filled!\n", .{});
std.debug.print("Getting vstream info\n", .{}); std.debug.print("Getting vstream info\n", .{});
var stream_info: hlo.hailo_vstream_info_t = undefined; var stream_info: hlo.hailo_vstream_info_t = undefined;
@ -116,12 +107,10 @@ pub fn main() !void {
assert(status == hlo.HAILO_SUCCESS); assert(status == hlo.HAILO_SUCCESS);
std.debug.print("Writing vstream data\n", .{}); std.debug.print("Writing vstream data\n", .{});
// Here we simulate taking our data and chunking it into frame_size, then uploading each frame until we have uploaded //for (0..100) |_| {
// all of our data status = hlo.hailo_vstream_write_raw_buffer(input_vstreams[0], input_data.ptr, input_frame_size);
for (0..frame_count) |_| {
status = hlo.hailo_vstream_write_raw_buffer(input_vstreams[0], @ptrCast(input_data.ptr), input_frame_size);
assert(status == hlo.HAILO_SUCCESS); assert(status == hlo.HAILO_SUCCESS);
} //}
std.debug.print("Input written\n", .{}); std.debug.print("Input written\n", .{});
@ -131,6 +120,7 @@ pub fn main() !void {
std.debug.print("input flushed!\n", .{}); std.debug.print("input flushed!\n", .{});
_ = hlo.hailo_release_output_vstreams(&output_vstreams, output_vstream_size); _ = hlo.hailo_release_output_vstreams(&output_vstreams, output_vstream_size);
_ = hlo.hailo_release_input_vstreams(&input_vstreams, input_vstream_size); _ = hlo.hailo_release_input_vstreams(&input_vstreams, input_vstream_size);
_ = hlo.hailo_release_hef(hef); _ = hlo.hailo_release_hef(hef);