File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed
backends/.jvm/src/test/scala Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change 1
1
package org .emergentorder .onnx .backends
2
2
3
+ import sys .process ._
4
+ import java .net .URL
5
+ import java .io .File
3
6
import java .nio .file .{Files , Paths }
7
+ import scala .language .postfixOps
4
8
import org .emergentorder .onnx .Tensors ._
5
9
import org .emergentorder .onnx .backends ._
6
10
import org .emergentorder .compiletime ._
7
11
import io .kjaer .compiletime ._
8
12
9
-
10
13
import org .scalatest .flatspec .AnyFlatSpec
11
14
import org .scalatest .matchers .should ._
12
15
13
16
14
17
class ONNXScalaSpec extends AnyFlatSpec with Matchers {
18
+
19
+ new URL (" https://s3.amazonaws.com/onnx-model-zoo/squeezenet/squeezenet1.1/squeezenet1.1.onnx" ) #> new File (" squeezenet1.1.onnx" ) !!
15
20
16
21
" SqueezeNet ONNX-Scala model" should " predict dummy image class" in {
17
22
val squeezenetBytes = Files .readAllBytes(Paths .get(" squeezenet1.1.onnx" ))
You can’t perform that action at this time.
0 commit comments