Генерация тензоров с элементами на основе положения в xtensor - PullRequest
Новая
       17

Генерация тензоров с элементами на основе положения в xtensor

0 голосов
/ 18 января 2019

Я пытаюсь построить структуру данных для представления изображения RGB в xtensor (трехмерная матрица, имеющая форму в виде (WIDTH, HEIGHT, 3). Каждый «пиксель» содержит данные, собранные функцией координат пикселей. В основном, я хочу повторить то, что этот код делает в python: 

image = [[cell_info(x, y) for x in range(WIDTH)]
                          for y in range(HEIGHT)]

где информация о ячейке возвращает список из 3 элементов, представляющих цветовые каналы.

Полагаю, правильным способом для этого должно быть использование xgenerator, но, честно говоря, я не могу понять, как использовать этот класс.

1 Ответ

0 голосов
/ 21 января 2019

Я нашел решение: Я изменил cell_info, чтобы принять параметр int channel, чтобы он возвращал целое число вместо массива. Тогда я написал это: 

class img_generator_fn {
        public:
          using value_type = int;
          img_generator_fn(const Map *map, const Position &center, shared_ptr<const Player> player,
                           const unsigned int field_radius)
              : m_map(map), m_center(center), m_player(player), m_translation(-(field_radius + 1)) {}
          ~img_generator_fn() { m_map = nullptr; }
          inline auto operator()(const unsigned int x, const unsigned int y, const unsigned int channel) const {
             return m_map->at(m_center + Position(x, y))->cell_info(m_player, channel);
          }
          template <class It> inline auto element(It, It end) const {
             return m_map->at(m_center + Position(*(end - 2) + m_translation, (*(end - 3)) + m_translation))
                 ->cell_info(m_player, *(end - 1));
          }

        private:
          const Map *m_map;
          const Position &m_center;
          shared_ptr<const Player> m_player;
          const unsigned int m_translation;
       };

       template <unsigned int field_side> auto field(const Position &center, shared_ptr<const Player> player) const {
          const array<unsigned int, 3> shape = {field_side, field_side, 3};
          auto gen = xt::detail::make_xgenerator(img_generator_fn(this, center, player, (field_side - 1) / 2), shape);
          return xt::xtensor_fixed<int, xt::xshape<field_side, field_side, 3>>(gen);
       }

Здесь Map представляет двумерную матрицу, это структура, которая вместе с player содержит информацию, которую я хочу сохранить в изображении. Функция at подбирает ячейку карты в указанной позиции (ячейка карты будет преобразована в пиксель). Функция field генерирует изображение с центром вокруг center из данной карты, используя xgenerator.

...