| 68 | }; |
| 69 | |
| 70 | class RGBImageReader : public ImageBase |
| 71 | { |
| 72 | public: |
| 73 | RGBImageReader(const std::string& filename, const nvinfer1::Dims& dims, const std::vector<float>& mean, const std::vector<float>& std); |
| 74 | std::unique_ptr<float> process() const; |
| 75 | private: |
| 76 | std::vector<float> mMean; |
| 77 | std::vector<float> mStd; |
| 78 | }; |
| 79 | |
| 80 | class ArgmaxImageWriter : public ImageBase |
| 81 | { |