Method Reference

dali/test/dali_test_decoder.h:32–59

class GenericDecoderTest : public DALISingleOpTest<ImgType> {
 public:
  vector<std::shared_ptr<TensorList<CPUBackend>>> Reference(
      const vector<TensorList<CPUBackend> *> &inputs, Workspace *ws) override {
    // single input - encoded images
    // single output - decoded images

    TensorList<CPUBackend> out(inputs[0]->num_samples());
    std::vector<Tensor<CPUBackend>> tmp_out(inputs[0]->num_samples());

    const TensorList<CPUBackend> &encoded_data = *inputs[0];

    const int c = this->GetNumColorComp();
    for (int i = 0; i < encoded_data.num_samples(); ++i) {
      auto *data = encoded_data.tensor<unsigned char>(i);
      auto data_size = volume(encoded_data.tensor_shape(i));

      this->DecodeImage(data, data_size, c, this->ImageType(), &tmp_out[i]);
    }

    out.SetupLike(tmp_out[0]);
    for (int i = 0; i < encoded_data.num_samples(); ++i) {
      out.SetSample(i, tmp_out[i]);
    }

    vector<std::shared_ptr<TensorList<CPUBackend>>> outputs;
    outputs.push_back(std::make_shared<TensorList<CPUBackend>>());
    outputs[0]->Copy(out);
    return outputs;
  }

 protected:
  virtual OpSpec DecodingOp() const { return OpSpec(); }