bert/research-project
Archived
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Improved visualisation of ReLU on images.

Browse Source
This commit is contained in:
Bert Peters
2018-03-13 15:06:48 +01:00
parent 74ecd42abe
commit c447a496fd
1 changed files with 20 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
src/visualisations.cpp
View File

@@ -154,27 +154,28 @@ static Animation *getReLUAnimation(const fmri::LayerData &prevState,
const vector<float> &curPositions) {
CHECK_EQ(curState.numEntries(), prevState.numEntries()) << "Layers should be of same size!";
const auto prevData = prevState.data(), curData = curState.data();
const auto sourceNormalize = getNodeNormalizer(prevState);
const auto sinkNormalize = getNodeNormalizer(curState);
std::vector<float> changes(prevState.numEntries());
caffe::caffe_sub(prevState.numEntries(), curState.data(), prevState.data(), changes.data());
EntryList results;
for (auto i : Range(curState.numEntries())) {
results.emplace_back(curData[i] - prevData[i], make_pair(i / sourceNormalize, i / sinkNormalize));
}
results = deduplicate(results);
const auto maxValue = max_element(results.begin(), results.end())->first;
return new ActivityAnimation(results, prevPositions.data(), curPositions.data(), -10, [=](float i) -> ActivityAnimation::Color {
if (maxValue == 0) {
return {1, 1, 1};
} else {
return {1 - i / maxValue, 1 - i / maxValue, 1};
if (curState.shape().size() == 2) {
EntryList results;
for (auto i : Range(curState.numEntries())) {
results.emplace_back(changes[i], make_pair(i, i));
}
});
const auto maxValue = max_element(results.begin(), results.end())->first;
return new ActivityAnimation(results, prevPositions.data(), curPositions.data(), -10,
[=](float i) -> ActivityAnimation::Color {
if (maxValue == 0) {
return {1, 1, 1};
} else {
return {1 - i / maxValue, 1 - i / maxValue, 1};
}
});
} else {
return new ImageInteractionAnimation(changes.data(), prevState.shape(), prevPositions, curPositions, -10);
}
}
static Animation *getNormalizingAnimation(const fmri::LayerData &prevState, const LayerData &curState,