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);
}
