autoforge 1.7.1__tar.gz → 1.7.2__tar.gz

{autoforge-1.7.1 → autoforge-1.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: AutoForge
-Version: 1.7.1
+Version: 1.7.2
 Summary: AutoForge is a Python tool for generating 3D printed layered models from an input image.
 Author-email: Hendric Voss <hvoss@techfak.uni-bielefeld.de>
 License: CC BY-NC-SA 4.0

{autoforge-1.7.1 → autoforge-1.7.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "AutoForge"
-version="1.7.1"
+version="1.7.2"
 description = "AutoForge is a Python tool for generating 3D printed layered models from an input image."
 authors = [{ name = "Hendric Voss", email = "hvoss@techfak.uni-bielefeld.de" }]
 readme = "README.md"

{autoforge-1.7.1 → autoforge-1.7.2}/src/AutoForge.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: AutoForge
-Version: 1.7.1
+Version: 1.7.2
 Summary: AutoForge is a Python tool for generating 3D printed layered models from an input image.
 Author-email: Hendric Voss <hvoss@techfak.uni-bielefeld.de>
 License: CC BY-NC-SA 4.0

{autoforge-1.7.1 → autoforge-1.7.2}/src/autoforge/Modules/Optimizer.py

@@ -5,12 +5,6 @@ import matplotlib.pyplot as plt
 import numpy as np
 import torch
 from torch.utils.tensorboard import SummaryWriter
-from torchcam.methods import SmoothGradCAMpp
-from torchcam.utils import overlay_mask
-from torchvision import models
-from torchvision.models import ResNet18_Weights
-from torchvision.transforms._functional_tensor import resize, normalize
-from torchvision.transforms.functional import to_pil_image
 from tqdm import tqdm
 
 from autoforge.Helper.CAdamW import CAdamW
@@ -119,48 +113,48 @@ class FilamentOptimizer:
             lr=self.learning_rate,
         )
 
-        # Setup the focus map component
-        # Load a state-of-the-art ResNet50 model with pretrained weights.
-        cam_model = models.resnet18(weights=ResNet18_Weights.DEFAULT).to(
-            self.target.device
-        )
-        cam_model.eval()
-        # Use ScoreCAM from torchcam on the 'layer4' block.
-        cam_extractor = SmoothGradCAMpp(cam_model, target_layer="layer4")
-        # Preprocess the target image for ResNet50.
-        target_preprocessed = normalize(
-            resize(self.target.permute(2, 0, 1), (224, 224)) / 255.0,
-            [0.485, 0.456, 0.406],
-            [0.229, 0.224, 0.225],
-        )
-        focus_maps = []
-        for _ in tqdm(range(1), desc="Creating loss function attention map..."):
-            out = cam_model(target_preprocessed.unsqueeze(0))
-            # ScoreCAM returns a dict with keys corresponding to layers.
-            focus_map = cam_extractor(out.squeeze(0).argmax().item(), out)[0]
-            # Normalize the focus map to [0, 1]
-            focus_map = (focus_map - focus_map.min()) / (
-                focus_map.max() - focus_map.min() + 1e-8
-            )
-            focus_maps.append(focus_map)
-        # get maximum over all maps
-        self.focus_map = torch.stack(focus_maps)
-        self.focus_map = self.focus_map.max(dim=0)[0]
-        # normalize again
-        self.focus_map = (self.focus_map - self.focus_map.min()) / (
-            self.focus_map.max() - self.focus_map.min() + 1e-8
-        )
-
-        # resize back to target size, current dim is [1,h,w]
-        self.focus_map = (
-            torch.nn.functional.interpolate(
-                self.focus_map.unsqueeze(0),
-                (self.H, self.W),
-                mode="bilinear",
-            )
-            .squeeze(0)
-            .squeeze(0)
-        )
+        # # Setup the focus map component
+        # # Load a state-of-the-art ResNet50 model with pretrained weights.
+        # cam_model = models.resnet18(weights=ResNet18_Weights.DEFAULT).to(
+        #     self.target.device
+        # )
+        # cam_model.eval()
+        # # Use ScoreCAM from torchcam on the 'layer4' block.
+        # cam_extractor = SmoothGradCAMpp(cam_model, target_layer="layer4")
+        # # Preprocess the target image for ResNet50.
+        # target_preprocessed = normalize(
+        #     resize(self.target.permute(2, 0, 1), (224, 224)) / 255.0,
+        #     [0.485, 0.456, 0.406],
+        #     [0.229, 0.224, 0.225],
+        # )
+        # focus_maps = []
+        # for _ in tqdm(range(1), desc="Creating loss function attention map..."):
+        #     out = cam_model(target_preprocessed.unsqueeze(0))
+        #     # ScoreCAM returns a dict with keys corresponding to layers.
+        #     focus_map = cam_extractor(out.squeeze(0).argmax().item(), out)[0]
+        #     # Normalize the focus map to [0, 1]
+        #     focus_map = (focus_map - focus_map.min()) / (
+        #         focus_map.max() - focus_map.min() + 1e-8
+        #     )
+        #     focus_maps.append(focus_map)
+        # # get maximum over all maps
+        # self.focus_map = torch.stack(focus_maps)
+        # self.focus_map = self.focus_map.max(dim=0)[0]
+        # # normalize again
+        # self.focus_map = (self.focus_map - self.focus_map.min()) / (
+        #     self.focus_map.max() - self.focus_map.min() + 1e-8
+        # )
+        #
+        # # resize back to target size, current dim is [1,h,w]
+        # self.focus_map = (
+        #     torch.nn.functional.interpolate(
+        #         self.focus_map.unsqueeze(0),
+        #         (self.H, self.W),
+        #         mode="bilinear",
+        #     )
+        #     .squeeze(0)
+        #     .squeeze(0)
+        # )
 
         # Setup best discrete solution tracking
         self.best_discrete_loss = float("inf")
@@ -202,17 +196,14 @@ class FilamentOptimizer:
             )
             self.ax[1, 1].set_title("Height Map Changes")
 
-            target_vis = self.target.cpu().detach().numpy()
-            focus_vis = self.focus_map.cpu().detach().numpy()
-
-            overlay = overlay_mask(
-                to_pil_image(target_vis),
-                to_pil_image(focus_vis, mode="F"),
-                alpha=0.5,
-            )
-
-            self.focus_map_ax = self.ax[1, 2].imshow(overlay)
-            self.ax[1, 2].set_title("Image Focus Map (currently unused)")
+            # target_vis = self.target.cpu().detach().numpy()
+            # focus_vis = self.focus_map.cpu().detach().numpy()
+            #
+            # overlay = overlay_mask(
+            #     to_pil_image(target_vis),
+            #     to_pil_image(focus_vis, mode="F"),
+            #     alpha=0.5,
+            # )
 
             # Compute and store the initial height map for later difference computation.
             with torch.no_grad():
@@ -286,7 +277,7 @@ class FilamentOptimizer:
             material_TDs=self.material_TDs,
             background=self.background,
             add_penalty_loss=10.0,
-            focus_map=self.focus_map,
+            focus_map=None,
             focus_strength=0.0,
         )
 
@@ -609,9 +600,6 @@ class FilamentOptimizer:
                     rng_seed=seed,
                 )
 
-            # 3) Compute the "discrete" loss function
-            #    If you have a separate function for that, just call it.
-            #    Otherwise, reuse the standard compute_loss with add_penalty_loss=False, etc.
             current_disc_loss = compute_loss(
                 comp=comp_disc,
                 target=self.target,

{autoforge-1.7.1 → autoforge-1.7.2}/src/autoforge/auto_forge.py

@@ -390,7 +390,7 @@ def main():
         for i in tbar:
             loss_val = optimizer.step(record_best=i % 8 == 0)
 
-            optimizer.visualize(interval=25)
+            optimizer.visualize(interval=100)
             optimizer.log_to_tensorboard(interval=100)
 
             if (i + 1) % 100 == 0: