Copypasted from defoggerFeaturizer on the defogger branch.
More...
#include <defoggerfeatures.h>
|
| DefoggerFeaturizer (size_t resX, size_t resY, int32_t strideX, int32_t strideY, bool fullVision=false) |
|
torch::Tensor | featurize (tc::Frame *frame, int mapX, int mapY, int playerId, at::Device device) |
|
void | featurize_unit (torch::Tensor &feats, tc::Unit &u, int, int) |
|
void | inc_feature (torch::Tensor &feature, int32_t c, int32_t x, int32_t y) const |
|
Copypasted from defoggerFeaturizer on the defogger branch.
cherrypi::DefoggerFeaturizer::DefoggerFeaturizer |
( |
size_t |
resX, |
|
|
size_t |
resY, |
|
|
int32_t |
strideX, |
|
|
int32_t |
strideY, |
|
|
bool |
fullVision = false |
|
) |
| |
|
inline |
tc::Frame cherrypi::DefoggerFeaturizer::combine |
( |
const std::deque< tc::Frame > & |
frames, |
|
|
int |
playerId |
|
) |
| |
|
static |
torch::Tensor cherrypi::DefoggerFeaturizer::featurize |
( |
tc::Frame * |
frame, |
|
|
int |
mapX, |
|
|
int |
mapY, |
|
|
int |
playerId, |
|
|
at::Device |
device |
|
) |
| |
void cherrypi::DefoggerFeaturizer::featurize_unit |
( |
torch::Tensor & |
feats, |
|
|
tc::Unit & |
u, |
|
|
int |
perspective, |
|
|
int |
playerId |
|
) |
| |
void cherrypi::DefoggerFeaturizer::inc_feature |
( |
torch::Tensor & |
feature, |
|
|
int32_t |
c, |
|
|
int32_t |
x, |
|
|
int32_t |
y |
|
) |
| const |
size_t cherrypi::DefoggerFeaturizer::feature_size |
bool cherrypi::DefoggerFeaturizer::fullVision |
std::array<int, 234> cherrypi::DefoggerFeaturizer::itypemapper |
size_t cherrypi::DefoggerFeaturizer::resX |
size_t cherrypi::DefoggerFeaturizer::resY |
int32_t cherrypi::DefoggerFeaturizer::strideX |
int32_t cherrypi::DefoggerFeaturizer::strideY |
std::array<int, 234> cherrypi::DefoggerFeaturizer::typemapper |
The documentation for this class was generated from the following files: