Skip to content
Snippets Groups Projects
Commit 66444aba authored by Tristan Faine's avatar Tristan Faine Committed by Solene Tarride
Browse files

fix linting issue

parent f5e6b784
No related branches found
No related tags found
1 merge request!76Add predicted objects to predict command
This commit is part of merge request !76. Comments created here will be created in the context of that merge request.
Hide whitespace changes
Inline Side-by-side
dan/predict/attention.py
+ 2
− 3
View file @ 66444aba
......@@ -3,7 +3,6 @@ import re
import cv2
import numpy as np
import math
from PIL import Image
from dan import logger
......@@ -74,8 +73,8 @@ def plot_attention(
:param scale: Scaling factor for the output gif image
:param outname: Name of the gif image
"""
height, width, _ = image.shape
height, width, _ = image.shape
attention_map = []
# Convert to PIL Image and create mask
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment