Skip to content

Commit 54b5e0f

Browse files
committed
Get test model file
1 parent 910f854 commit 54b5e0f

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

backends/.jvm/src/test/scala/SqueezeNetTest.scala

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,22 @@
11
package org.emergentorder.onnx.backends
22

3+
import sys.process._
4+
import java.net.URL
5+
import java.io.File
36
import java.nio.file.{Files, Paths}
7+
import scala.language.postfixOps
48
import org.emergentorder.onnx.Tensors._
59
import org.emergentorder.onnx.backends._
610
import org.emergentorder.compiletime._
711
import io.kjaer.compiletime._
812

9-
1013
import org.scalatest.flatspec.AnyFlatSpec
1114
import org.scalatest.matchers.should._
1215

1316

1417
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") !!
1520

1621
"SqueezeNet ONNX-Scala model" should "predict dummy image class" in {
1722
val squeezenetBytes = Files.readAllBytes(Paths.get("squeezenet1.1.onnx"))

0 commit comments

Comments
 (0)