Code

Simplify inference with ONNX models#2060

Code

Merged

AdeelH merged 7 commits intoazavea:masterazavea/raster-vision:masterfrom AdeelH:onnx-predAdeelH/raster-vision:onnx-predCopy head branch name to clipboardFeb 14, 2024

+201-162

Commits

Commits on Feb 14, 2024

• don't require model in Learner constructor if given .onnx path

  AdeelH

  committed
  86d8d63

  Copy full SHA for 86d8d63

• don't hard-code input name for ONNX model

  AdeelH

  committed
  c114fe5

  Copy full SHA for c114fe5

• don't restrict DataConfig type for LearnerConfig subclasses

  Show description for 8ba367f
  AdeelH

  committed
  8ba367f

  Copy full SHA for 8ba367f

• allow .onnx file to be remote

  AdeelH

  committed
  4a10375

  Copy full SHA for 4a10375

• allow LearnerConfig.solver to be None

  AdeelH

  committed
  2877eb5

  Copy full SHA for 2877eb5

• do not require output_dir if not in training mode

  AdeelH

  committed
  e7f8936

  Copy full SHA for e7f8936

• refactor .predict_scene() to use standalone predict_scene_*() functions

  Show description for 822cdae
  AdeelH

  committed
  822cdae

  Copy full SHA for 822cdae