nvjpeg/cpp/test_pydecoder.cpp

#include <iostream>
#include <pybind11/embed.h>
#include "python_nvjpeg_decoder.h"


int main() {
  py::scoped_interpreter guard{};  
  NVJpegDecoder::PythonDecoder decoder;
  if (!decoder.Init()) {
    std::cout << "Init Failed" << std::endl;
    return -1;
  }
  std::cout << 1 << std::endl;;
  py::list data;
  std::cout << 2 << std::endl;;  
  for (int i = 0; i < 10; i++) {
    std::cout << i << ' ';
    data.append(i);
  }
  decoder.BatchDecode(data);
  return 0;
}