diff options
| author | Anthony Barbier <anthony.barbier@arm.com> | 2018-07-23 14:41:59 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
| commit | 40606df5cc5e687e4d72af842af8d65fc4228cbc (patch) | |
| tree | 13900c3ec6d9fa95327d2d7056fdbc23c0571f18 /utils/GraphUtils.h | |
| parent | 7d66a8e3f603f2cd363f04a750847e3f9eabdfd4 (diff) | |
| download | ComputeLibrary-40606df5cc5e687e4d72af842af8d65fc4228cbc.tar.gz | |
COMPMID-1414: Print the image being validated during graph validation
Also enable the asserts in release mode as they should stop any example from running
Change-Id: I308daa53c2096758f088872191b2fcba2131255c
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/140988
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'utils/GraphUtils.h')
| -rw-r--r-- | utils/GraphUtils.h | 23 |
1 files changed, 13 insertions, 10 deletions
diff --git a/utils/GraphUtils.h b/utils/GraphUtils.h index 6d537865ba..6cec2dae4d 100644 --- a/utils/GraphUtils.h +++ b/utils/GraphUtils.h @@ -181,21 +181,23 @@ class ValidationInputAccessor final : public graph::ITensorAccessor public: /** Constructor * - * @param[in] image_list File containing all the images to validate - * @param[in] images_path Path to images. - * @param[in] bgr (Optional) Fill the first plane with blue channel (default = false - RGB format) - * @param[in] preprocessor (Optional) Image pre-processing object (default = nullptr) - * @param[in] start (Optional) Start range - * @param[in] end (Optional) End range + * @param[in] image_list File containing all the images to validate + * @param[in] images_path Path to images. + * @param[in] bgr (Optional) Fill the first plane with blue channel (default = false - RGB format) + * @param[in] preprocessor (Optional) Image pre-processing object (default = nullptr) + * @param[in] start (Optional) Start range + * @param[in] end (Optional) End range + * @param[out] output_stream (Optional) Output stream * * @note Range is defined as [start, end] */ ValidationInputAccessor(const std::string &image_list, std::string images_path, - std::unique_ptr<IPreprocessor> preprocessor = nullptr, - bool bgr = true, - unsigned int start = 0, - unsigned int end = 0); + std::unique_ptr<IPreprocessor> preprocessor = nullptr, + bool bgr = true, + unsigned int start = 0, + unsigned int end = 0, + std::ostream &output_stream = std::cout); // Inherited methods overriden: bool access_tensor(ITensor &tensor) override; @@ -206,6 +208,7 @@ private: std::unique_ptr<IPreprocessor> _preprocessor; bool _bgr; size_t _offset; + std::ostream &_output_stream; }; /** Output Accessor used for network validation */ |