Skip to content

Commit

Permalink
corrected dims of the images (HWCN -> WHCN)
Browse files Browse the repository at this point in the history
  • Loading branch information
aditkumar72 committed Jun 19, 2021
1 parent db8342a commit a2bb58f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tutorials/60-minute-blitz/60-minute-blitz.jl
Original file line number Diff line number Diff line change
Expand Up @@ -304,7 +304,7 @@ image.(X[rand(1:end, 10)])

# Defining a `getarray` function would help in converting the matrices to `Float` type.

getarray(X) = float.(permutedims(channelview(X), (2, 3, 1)))
getarray(X) = float.(permutedims(channelview(X), (3, 2, 1)))
imgs = [getarray(X[i].img) for i in 1:50000]

# The first 49k images (in batches of 1000) will be our training set, and the rest is
Expand Down

0 comments on commit a2bb58f

Please sign in to comment.