| Parker Schuh | f2a3493 | 2019-02-16 20:39:19 -0800 | [diff] [blame] | 1 | #ifndef _AOS_VISION_IMAGE_IMAGE_DATASET_H_ |
| 2 | #define _AOS_VISION_IMAGE_IMAGE_DATASET_H_ |
| 3 | |
| 4 | #include <string> |
| 5 | #include <vector> |
| 6 | |
| Stephan Pleines | d99b1ee | 2024-02-02 20:56:44 -0800 | [diff] [blame^] | 7 | namespace aos::vision { |
| Parker Schuh | f2a3493 | 2019-02-16 20:39:19 -0800 | [diff] [blame] | 8 | |
| 9 | struct DatasetFrame { |
| 10 | // TODO: These should be V4L formats ideally. |
| 11 | bool is_jpeg = true; |
| 12 | std::string data; |
| 13 | }; |
| 14 | |
| 15 | std::vector<DatasetFrame> LoadDataset(const std::string &jpeg_list_filename); |
| 16 | |
| Parker Schuh | 9e1d169 | 2019-02-24 14:34:04 -0800 | [diff] [blame] | 17 | DatasetFrame LoadFile(const std::string &jpeg_filename); |
| 18 | |
| Stephan Pleines | d99b1ee | 2024-02-02 20:56:44 -0800 | [diff] [blame^] | 19 | } // namespace aos::vision |
| Parker Schuh | f2a3493 | 2019-02-16 20:39:19 -0800 | [diff] [blame] | 20 | |
| 21 | #endif // _AOS_VISION_IMAGE_IMAGE_DATASET_H_ |