Implement basic visualisation for pooling layers.
This commit is contained in:
26
src/PoolingLayerAnimation.hpp
Normal file
26
src/PoolingLayerAnimation.hpp
Normal file
@@ -0,0 +1,26 @@
|
||||
#pragma once
|
||||
|
||||
#include "Animation.hpp"
|
||||
#include "LayerData.hpp"
|
||||
#include "Texture.hpp"
|
||||
|
||||
namespace fmri
|
||||
{
|
||||
class PoolingLayerAnimation : public Animation
|
||||
{
|
||||
public:
|
||||
PoolingLayerAnimation(const LayerData& prevData, const LayerData& curData, const std::vector<float>& prevPositions, const std::vector<float>& curPositions, float xDist);
|
||||
|
||||
void draw(float timeStep) override;
|
||||
|
||||
private:
|
||||
Texture original;
|
||||
Texture downSampled;
|
||||
std::vector<float> startingPositions;
|
||||
std::vector<float> deltas;
|
||||
std::vector<float> textureCoordinates;
|
||||
|
||||
static Texture loadTextureForData(const LayerData& data);
|
||||
static std::vector<float> computePositions(const std::vector<float> &nodePositions, float scaling = 1);
|
||||
};
|
||||
}
|
||||
Reference in New Issue
Block a user