copilot/OpenSeeFace
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Training with less data is now possible too.

Browse source
This commit is contained in:
Emiliana 2020-02-09 18:09:09 +01:00
parent b5706183c2
commit b5218a1eb0
2 changed files with 34 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

8
Unity/Plugins/SVMModel.cs
View file

@ -122,12 +122,12 @@ public class SVMModel {
DestroyModel();
if (rows > 10000)
return false;
if (cols < 1 || rows < 1 || features.Length != rows * cols || labels.Length != rows)
if (cols < 1 || rows < 1 || features.Length < rows * cols || labels.Length < rows)
return false;
int max = -1;
HashSet<int> knownClasses = new HashSet<int>();
for (int i = 0; i < labels.Length; i++) {
for (int i = 0; i < rows; i++) {
labels[i] = (float)Math.Round(labels[i]);
int l = (int)labels[i];
if (l > max)
@ -154,9 +154,9 @@ public class SVMModel {
accuracy = 0f;
if (!haveModel)
return null;
if (rows < 1 || features.Length != rows * cols || labels.Length != rows)
if (rows < 1 || features.Length < rows * cols || labels.Length < rows)
return null;
for (int i = 0; i < labels.Length; i++) {
for (int i = 0; i < rows; i++) {
labels[i] = (float)Math.Round(labels[i]);
if (labels[i] < 0 || labels[i] >= maxClasses)
return null;