Skip to content

Commit

Permalink
Added a small descriptive comment regarding normalization in helper.py (
Browse files Browse the repository at this point in the history
#326)

* Added a small descriptive comment regarding normalization in helper.py

* remove unrelated commited stuff

* remove unrelated commited stuff

---------

Co-authored-by: Nicki Skafte <[email protected]>
  • Loading branch information
emiltoft97 and SkafteNicki authored Jan 24, 2025
1 parent 899a15b commit cc6d10a
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion s1_development_environment/exercise_files/helper.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,8 @@ def imshow(image, ax=None, title=None, normalize=True):
if ax is None:
fig, ax = plt.subplots()
image = image.numpy().transpose((1, 2, 0))


# Sets standard for how to normalize tensor images
if normalize:
mean = np.array([0.485, 0.456, 0.406])
std = np.array([0.229, 0.224, 0.225])
Expand Down

0 comments on commit cc6d10a

Please sign in to comment.