Skip to content

Commit

Permalink
Fix pipeline
Browse files Browse the repository at this point in the history
  • Loading branch information
MalcolmMielle committed Mar 19, 2024
1 parent b2dcb67 commit d16290e
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 8 deletions.
2 changes: 1 addition & 1 deletion .flake8
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
[flake8]
max-line-length = 88
extend-ignore = E203
extend-ignore = E203, F722
7 changes: 1 addition & 6 deletions thermo_nerf/flir_thermal_images/custom_flir.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,11 +3,10 @@

import numpy as np
import pandas as pd
from FlirImageExtractor.flir_image_extractor import FlirImageExtractor
from matplotlib import cm
from PIL import Image

from FlirImageExtractor.flir_image_extractor import FlirImageExtractor


def get_min_max_temperatures(path_to_csv_files: str) -> tuple[float, float]:
"""
Expand Down Expand Up @@ -87,10 +86,6 @@ def save_normalised_thermal_images(
self.absolute_max_temperature,
)

# temperature = temperature.reshape(640, 480)
# image = Image.fromarray(temperature)
# image.save(Path(path_to_thermal_images_curated, Path(path).stem + ".tiff"))

gray_scale_image = temperature.reshape(640, 480).astype("uint8")

Image.fromarray(gray_scale_image, mode="L").save(
Expand Down
3 changes: 2 additions & 1 deletion thermo_nerf/thermal_nerf/thermal_field_head.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,8 @@ def forward(
"""
if not self.net:
raise SystemError(
"in_dim not set. Must be provided to constructor, or set_in_dim() should be called."
"in_dim not set. Must be provided to constructor, or set_in_dim()"
" should be called."
)
out_tensor = self.net(in_tensor)
if self.activation:
Expand Down

0 comments on commit d16290e

Please sign in to comment.