Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions docs/src/plots.md
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@

## 1D Plotting Functions
```@docs
ScatteringPlotsExt.plotOriginalSignal1D
ScatteringPlotsExt.plotZerothLayer1D
ScatteringPlotsExt.plotFirstLayer1DSingleWavelet
ScatteringPlotsExt.gifFirstLayer1D
Expand Down
4 changes: 2 additions & 2 deletions src/scatteringplots.jl
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
"""
plotZerothLayer1D(sf; saveTo=nothing, index=1)
Function that plots the zeroth layer of the scattering transform at a specified example index.
plotOriginalSignal1D(f; title="Original Signal", saveTo=nothing, index=1)
Function that plots the original signal at a specified example index.
"""
function plotOriginalSignal1D(f; title="Original Signal", saveTo=nothing, index=1)
plt = plot(f[:,1,index], title=title, legend=false, xlim=(0, length(f[:, 1, index])+1), color=:blue, margin=5Plots.mm, size=(720,480))
Expand Down
4 changes: 4 additions & 0 deletions test/fluxtests.jl
Original file line number Diff line number Diff line change
Expand Up @@ -99,6 +99,9 @@
@test res1[1:32*3, 1] ≈ reshape(res[0][:, :, 1], (32 * 3,))
end

#=
As of this moment these tests are broken. This is merely because GPU support has not been added to
MonogenicFilterFlux.jl yet. When support is added these tests will be re-enabled.
nFilters = [1, 12, 12, 12]
@testset "2D basics" begin
n_init_channels=2
Expand Down Expand Up @@ -141,6 +144,7 @@
@test size(res1) == (totalSize, 2)
@test res1[1:32^2*n_init_channels, 1] ≈ reshape(res[0][:, :, :, 1], (32^2 * n_init_channels,))
end
=#

nFilters = [1, 10, 9]
@testset "1D integer pooling" begin
Expand Down
Loading