public YoloWrapper(string configurationFilename, string weightsFilename, string namesFilename, int gpu = 0) { objectTypeResolver = new YoloObjectTypeResolver(namesFilename); Initialize(configurationFilename, weightsFilename, gpu); container = new BboxContainer(); }
public void ResolveObjectId() { var names = new string[] { "Car", "Cat", "Bike" }; var resolver = new YoloObjectTypeResolver(names); var name = resolver.Resolve(1); Assert.AreEqual(names[1], name); name = resolver.Resolve(2); Assert.AreEqual(names[2], name); name = resolver.Resolve(0); Assert.AreEqual(names[0], name); name = resolver.Resolve(4); Assert.AreEqual("unknown key", name); }
Convert(BboxContainer container, YoloObjectTypeResolver objectTypeResolver) { var yoloItems = new List <YoloItem>(); foreach (var item in container.candidates) { if (item.w > 0 || item.h > 0) { yoloItems.Add(new YoloItem { X = (int)item.x, Y = (int)item.y, Height = (int)item.h, Width = (int)item.w, Confidence = item.prob, FrameId = (int)item.frames_counter, TrackId = (int)item.track_id, Shape = (YoloItem.ShapeType)item.shape, Type = objectTypeResolver.Resolve((int)item.obj_id) }); } } return(yoloItems); }