Make maxFaceDistance configurable
This commit is contained in:
parent
15fcfbde44
commit
0ed8df8c17
|
@ -15,6 +15,7 @@ export default function MLDebug() {
|
||||||
const [minClusterSize, setMinClusterSize] = useState<number>(4);
|
const [minClusterSize, setMinClusterSize] = useState<number>(4);
|
||||||
const [minFaceSize, setMinFaceSize] = useState<number>(24);
|
const [minFaceSize, setMinFaceSize] = useState<number>(24);
|
||||||
const [batchSize, setBatchSize] = useState<number>(50);
|
const [batchSize, setBatchSize] = useState<number>(50);
|
||||||
|
const [maxFaceDistance, setMaxFaceDistance] = useState<number>(0.55);
|
||||||
const [mlResult, setMlResult] = useState<MLSyncResult>({
|
const [mlResult, setMlResult] = useState<MLSyncResult>({
|
||||||
allFaces: [],
|
allFaces: [],
|
||||||
clustersWithNoise: {
|
clustersWithNoise: {
|
||||||
|
@ -64,7 +65,8 @@ export default function MLDebug() {
|
||||||
clusterFaceDistance,
|
clusterFaceDistance,
|
||||||
minClusterSize,
|
minClusterSize,
|
||||||
minFaceSize,
|
minFaceSize,
|
||||||
batchSize
|
batchSize,
|
||||||
|
maxFaceDistance
|
||||||
);
|
);
|
||||||
setMlResult(result);
|
setMlResult(result);
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
|
@ -108,6 +110,13 @@ export default function MLDebug() {
|
||||||
<button onClick={() => setBatchSize(200)}>200</button>
|
<button onClick={() => setBatchSize(200)}>200</button>
|
||||||
<button onClick={() => setBatchSize(500)}>500</button>
|
<button onClick={() => setBatchSize(500)}>500</button>
|
||||||
|
|
||||||
|
<p></p>
|
||||||
|
<div>MaxFaceDistance: {maxFaceDistance}</div>
|
||||||
|
<button onClick={() => setMaxFaceDistance(0.45)}>0.45</button>
|
||||||
|
<button onClick={() => setMaxFaceDistance(0.5)}>0.5</button>
|
||||||
|
<button onClick={() => setMaxFaceDistance(0.55)}>0.55</button>
|
||||||
|
<button onClick={() => setMaxFaceDistance(0.6)}>0.6</button>
|
||||||
|
|
||||||
<p></p>
|
<p></p>
|
||||||
<button onClick={onSync}>Run ML Sync</button>
|
<button onClick={onSync}>Run ML Sync</button>
|
||||||
|
|
||||||
|
|
|
@ -32,7 +32,7 @@ class MachineLearningService {
|
||||||
private clusteringService: ClusteringService;
|
private clusteringService: ClusteringService;
|
||||||
|
|
||||||
private clusterFaceDistance = 0.4;
|
private clusterFaceDistance = 0.4;
|
||||||
private maxFaceDistance = 0.6;
|
private maxFaceDistance = 0.55;
|
||||||
private minClusterSize = 4;
|
private minClusterSize = 4;
|
||||||
private minFaceSize = 24;
|
private minFaceSize = 24;
|
||||||
private batchSize = 50;
|
private batchSize = 50;
|
||||||
|
@ -63,12 +63,14 @@ class MachineLearningService {
|
||||||
clusterFaceDistance: number,
|
clusterFaceDistance: number,
|
||||||
minClusterSize: number,
|
minClusterSize: number,
|
||||||
minFaceSize: number,
|
minFaceSize: number,
|
||||||
batchSize: number
|
batchSize: number,
|
||||||
|
maxFaceDistance: number
|
||||||
) {
|
) {
|
||||||
this.clusterFaceDistance = clusterFaceDistance;
|
this.clusterFaceDistance = clusterFaceDistance;
|
||||||
this.minClusterSize = minClusterSize;
|
this.minClusterSize = minClusterSize;
|
||||||
this.minFaceSize = minFaceSize;
|
this.minFaceSize = minFaceSize;
|
||||||
this.batchSize = batchSize;
|
this.batchSize = batchSize;
|
||||||
|
this.maxFaceDistance = maxFaceDistance;
|
||||||
|
|
||||||
// setWasmPath('/js/tfjs/');
|
// setWasmPath('/js/tfjs/');
|
||||||
await tf.ready();
|
await tf.ready();
|
||||||
|
|
|
@ -7,7 +7,8 @@ export class MachineLearningWorker {
|
||||||
clusterFaceDistance,
|
clusterFaceDistance,
|
||||||
minClusterSize,
|
minClusterSize,
|
||||||
minFaceSize,
|
minFaceSize,
|
||||||
batchSize
|
batchSize,
|
||||||
|
maxFaceDistance
|
||||||
) {
|
) {
|
||||||
if (!(typeof navigator !== 'undefined')) {
|
if (!(typeof navigator !== 'undefined')) {
|
||||||
console.log(
|
console.log(
|
||||||
|
@ -22,7 +23,8 @@ export class MachineLearningWorker {
|
||||||
clusterFaceDistance,
|
clusterFaceDistance,
|
||||||
minClusterSize,
|
minClusterSize,
|
||||||
minFaceSize,
|
minFaceSize,
|
||||||
batchSize
|
batchSize,
|
||||||
|
maxFaceDistance
|
||||||
);
|
);
|
||||||
const results = await mlService.sync(token);
|
const results = await mlService.sync(token);
|
||||||
console.log('Ran machine learning sync from worker', results);
|
console.log('Ran machine learning sync from worker', results);
|
||||||
|
|
Loading…
Reference in a new issue