Skip to content

Commit d69eaba

Browse files
committed
- Added custom models to allow usability of undefined models
1 parent 617f62e commit d69eaba

File tree

2 files changed

+33
-0
lines changed

2 files changed

+33
-0
lines changed
Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
namespace MOT.CORE.ReID.Models
2+
{
3+
public class CustomReidModel : IReidModel
4+
{
5+
public int Width { get; set; }
6+
public int Height { get; set; }
7+
public int BatchSize { get; set; }
8+
public int Channels { get; set; }
9+
public int OutputVectorSize { get; set; }
10+
public string[] Outputs { get; set; }
11+
public string Input { get; set; }
12+
}
13+
}
Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
namespace MOT.CORE.YOLO.Models
2+
{
3+
public class CustomYoloModel : IYoloModel
4+
{
5+
public int Width { get; set; }
6+
public int Height { get; set; }
7+
public int Depth { get; set; }
8+
public int Dimensions { get; set; }
9+
public int[] Strides { get; set; }
10+
public int[][][] Anchors { get; set; }
11+
public int[] Shapes { get; set; }
12+
public float Confidence { get; set; }
13+
public float MulConfidence { get; set; }
14+
public float Overlap { get; set; }
15+
public int Channels { get; set; }
16+
public int BatchSize { get; set; }
17+
public string[] Outputs { get; set; }
18+
public string Input { get; set; }
19+
}
20+
}

0 commit comments

Comments
 (0)