use fix for tf.maximum for now in leaky relu

justadudewhohacks · justadudewhohacks · commit 6c3c55e1e870 · 2018-08-13T12:50:54.000+02:00
diff --git a/src/NetInput.ts b/src/NetInput.ts
@@ -86,6 +86,18 @@ export class NetInput {
     return this._inputSize
   }
 
+  public get relativePaddings(): Point[] {
+    return Array(this.inputs.length).fill(0).map(
+      (_, batchIdx) => this.getRelativePaddings(batchIdx)
+    )
+  }
+
+  public get reshapedInputDimensions(): Dimensions[] {
+    return Array(this.inputs.length).fill(0).map(
+      (_, batchIdx) => this.getReshapedInputDimensions(batchIdx)
+    )
+  }
+
   public getInputDimensions(batchIdx: number): number[] {
     return this._inputDimensions[batchIdx]
   }
diff --git a/src/tinyYolov2/leaky.ts b/src/tinyYolov2/leaky.ts
@@ -2,6 +2,8 @@ import * as tf from '@tensorflow/tfjs-core';
 
 export function leaky(x: tf.Tensor4D): tf.Tensor4D {
   return tf.tidy(() => {
-    return tf.maximum(x, tf.mul(x, tf.scalar(0.10000000149011612)))
+    const min = tf.mul(x, tf.scalar(0.10000000149011612))
+    return tf.add(tf.relu(tf.sub(x, min)), min)
+    //return tf.maximum(x, min)
   })
 }

Original file line number	Diff line number	Diff line change
`@@ -2,6 +2,8 @@ import * as tf from '@tensorflow/tfjs-core';`
`2`	`2`
`3`	`3`	`export function leaky(x: tf.Tensor4D): tf.Tensor4D {`
`4`	`4`	`return tf.tidy(() => {`
`5`		`- return tf.maximum(x, tf.mul(x, tf.scalar(0.10000000149011612)))`
	`5`	`+ const min = tf.mul(x, tf.scalar(0.10000000149011612))`
	`6`	`+ return tf.add(tf.relu(tf.sub(x, min)), min)`
	`7`	`+ //return tf.maximum(x, min)`
`6`	`8`	`})`
`7`	`9`	`}`