PyPI - autoforge - Versions diffs - 1.1.2__tar.gz → 1.2.0__tar.gz - Mend

autoforge 1.1.2tar.gz → 1.2.0tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (31) hide show

{autoforge-1.1.2 → autoforge-1.2.0}/PKG-INFO RENAMED Viewed

@@ -1,14 +1,14 @@
 Metadata-Version: 2.2
 Name: AutoForge
-Version: 1.1.2
+Version: 1.2.0
 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
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 Requires-Dist: numpy>=1.19.0
-Requires-Dist: jax>=0.3.10
-Requires-Dist: jaxlib>=0.3.10
+Requires-Dist: torch>=2.6.0
+Requires-Dist: torchvision>=0.21.0
 Requires-Dist: opencv-python>=4.2.0.32
 Requires-Dist: matplotlib>=3.3.0
 Requires-Dist: configargparse>=1.2.3
@@ -17,8 +17,6 @@ Requires-Dist: pandas>=1.1.0
 Requires-Dist: scikit-image
 Requires-Dist: scikit-learn
 Requires-Dist: optax>=0.1.3
-Provides-Extra: gpu
-Requires-Dist: jax[cuda12]; extra == "gpu"
 # AutoForge
@@ -27,25 +25,31 @@ AutoForge is a Python tool for generating 3D printed layered models from an inpu
 **TLDR:** It uses a picture to generate a 3D layer image that you can print with a 3d printer. Similar to [Hueforge](https://shop.thehueforge.com/), but without the manual work (and without the artistic control).
+## Important Information
+We recently switched from JAX to PyTorch, to allow for a more streamlined development process. \
+This can have some unforeseen consequences, so please report any bugs you find.
 ## Example
 All examples use only the 13 BambuLab Basic filaments, currently available in Hueforge.
 <div style="display: flex; justify-content: center; gap: 20px;">
   <div style="text-align: center;">
     <h3>Input Image</h3>
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi.jpg" width="300" />
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature.jpg" width="300" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi.jpg" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature.jpg" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/cat.jpg" width="200" />
   </div>
   <div style="text-align: center;">
     <h3>Autoforge Output (75 color layers (3.0mm) + 10 gray background layers (0.4mm)) </h3>
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi_discretized.jpg" width="300" />
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature_discretized.jpg" width="300" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi_discretized.jpg" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature_discretized.jpg" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/cat_discretized.jpg" width="200" />
   </div>
   <div style="text-align: center;">
     <h3>Imported Hueforge View</h3>
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi_hueforge.png" width="300" />
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature_hueforge.png" width="300" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi_hueforge.png" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature_hueforge.png" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/cat_hueforge.png" width="200" />
   </div>
 </div>
@@ -66,33 +70,17 @@ I would love to see what you made!
 ## Installation
-1. **Clone the Repository**
-   ```bash
-   git clone https://github.com/hvoss-techfak/AutoForge.git
-   cd AutoForge
-   ```
-2. **Set Up a Virtual Environment (Optional but Recommended)**
-   ```bash
-   python -m venv venv
-   source venv/bin/activate  # On Windows: venv\Scripts\activate
-   ```
-3. **Install Dependencies**
-   ```bash
-   pip install -r requirements.txt
-   ```
+To install AutoForge, simply install the current version from PyPI:
+```bash
+   pip install autoforge
+```
-See the [requirements.txt](requirements.txt) file for a complete list of dependencies.
-The optimizer is built using JAX, which benefits from a CUDA-compatible GPU for optimal performance.
-Without a dedicated GPU the process can take significantly longer (up to 10-20x slower).
-If you have a GPU, you can install the GPU version of JAX by running:
+The optimizer is built using JAX, which benefits from a CUDA-compatible GPU for optimal performance.  \
+Without a dedicated GPU the process can take significantly longer (up to 10-20x slower).  \
+If you have a GPU, you can install the GPU version of autoforge by running:\
 ```bash
-pip install -U "jax[cuda12]"
+   pip install autoforge[gpu]
 ```
 Currently, jax gpu support is only available under linux. \
@@ -108,7 +96,7 @@ The script is run from the command line and accepts several arguments. Below is
 > If you want to limit the amount of colors the program can use, you can simply delete the colors you don't want from the CSV file.
 ```bash
-python auto_forge.py --input_image path/to/input_image.jpg --csv_file path/to/materials.csv --output_folder outputs
+autoforge --input_image path/to/input_image.jpg --csv_file path/to/materials.csv --output_folder outputs
 ```
 ### Command Line Arguments
@@ -117,18 +105,26 @@ python auto_forge.py --input_image path/to/input_image.jpg --csv_file path/to/ma
 - `--input_image`: **(Required)** Path to the input image.
 - `--csv_file`: **(Required)** Path to the CSV file containing material data. The CSV should include columns for the brand, name, color (hex code), and TD values.
 - `--output_folder`: **(Required)** Folder where output files will be saved.
-- `--iterations`: Number of optimization iterations (default: 20000).
-- `--learning_rate`: Learning rate for the optimizer (default: 5e-3).
+- `--iterations`: Number of optimization iterations (default: 5000).
+- `--best_loss_iterations`: Percentage of optimization iterations after which we start to record the best loss (default: 4000).
+  **Note:** We currently do this as we increase a penalty loss for color and color swaps as the time goes on. Without this the penalties would have no impact.
+- `--learning_rate`: Learning rate for the optimizer (default: 1e-2).
 - `--layer_height`: Layer thickness in millimeters (default: 0.04).
 - `--max_layers`: Maximum number of layers (default: 75).
   **Note:** This is about 3mm + the background height
 - `--background_height`: Height of the background in millimeters (default: 0.4).
   **Note:** The background height must be divisible by the layer height.
-- `--background_color`: Background color in hexadecimal format (default: `#8e9089` aka Bambulab Grey).
-- `--max_size`: Maximum dimension (width or height) for the resized target image (default: 512).
+- `--background_color`: Background color in hexadecimal format (default: `#000000` aka Black).
+  **Note:** The solver currently assumes that you have a solid color in the background, which means a color with a TD value of 4 or less (if you have a background height of 0.4)
+- `--output_size`: Maximum dimension for target image (default: 1024).
+- `--solver_size`: Maximum dimension for solver (fast) image (default: 128).
+  **Note:** We solve on a smaller size as this is many times faster, but also a bit less accurate. Increase if you need more accuracy.
 - `--decay`: Final tau value for the Gumbel-Softmax formulation (default: 0.01).
-- `--loss`: Loss function to use. Choices are `mse`, `perceptual`, or `perceptual_l1` (default: `mse`).
 - `--visualize`: Flag to enable live visualization of the composite image during optimization.
+- `--perform_pruning`: Perform pruning after optimization (default: True).
+- `--pruning_max_colors`: Max number of colors allowed after pruning (default: 10).
+- `--pruning_max_swaps`: Max number of swaps allowed after pruning (default: 20).
+- `--random_seed`: Random seed for reproducibility (default: 0 (disabled) ).
 ## Outputs
@@ -154,6 +150,9 @@ Although I would love to be it fully color compatible with hueforge, I don't thi
 ## License
 AutoForge © 2025 by Hendric Voss is licensed under [CC BY-NC-SA 4.0](https://creativecommons.org/licenses/by-nc-sa/4.0/).
+The software is provided as-is and comes with no warranty or guarantee of support.
+The
 ## Acknowledgements
@@ -174,5 +173,6 @@ AutoForge makes use of several open source libraries:
 Example Images:
 <a href="https://www.vecteezy.com/free-photos/anime-girl">Anime Girl Stock photos by Vecteezy</a>
 <a href="https://www.vecteezy.com/free-photos/nature">Nature Stock photos by Vecteezy</a>
+<a href="https://www.vecteezy.com/free-photos/ai-generated">Ai Generated Stock photos by Vecteezy</a>
 Happy printing!

{autoforge-1.1.2 → autoforge-1.2.0}/README.md RENAMED Viewed

@@ -5,25 +5,31 @@ AutoForge is a Python tool for generating 3D printed layered models from an inpu
 **TLDR:** It uses a picture to generate a 3D layer image that you can print with a 3d printer. Similar to [Hueforge](https://shop.thehueforge.com/), but without the manual work (and without the artistic control).
+## Important Information
+We recently switched from JAX to PyTorch, to allow for a more streamlined development process. \
+This can have some unforeseen consequences, so please report any bugs you find.
 ## Example
 All examples use only the 13 BambuLab Basic filaments, currently available in Hueforge.
 <div style="display: flex; justify-content: center; gap: 20px;">
   <div style="text-align: center;">
     <h3>Input Image</h3>
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi.jpg" width="300" />
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature.jpg" width="300" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi.jpg" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature.jpg" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/cat.jpg" width="200" />
   </div>
   <div style="text-align: center;">
     <h3>Autoforge Output (75 color layers (3.0mm) + 10 gray background layers (0.4mm)) </h3>
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi_discretized.jpg" width="300" />
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature_discretized.jpg" width="300" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi_discretized.jpg" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature_discretized.jpg" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/cat_discretized.jpg" width="200" />
   </div>
   <div style="text-align: center;">
     <h3>Imported Hueforge View</h3>
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi_hueforge.png" width="300" />
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature_hueforge.png" width="300" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi_hueforge.png" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature_hueforge.png" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/cat_hueforge.png" width="200" />
   </div>
 </div>
@@ -44,33 +50,17 @@ I would love to see what you made!
 ## Installation
-1. **Clone the Repository**
-   ```bash
-   git clone https://github.com/hvoss-techfak/AutoForge.git
-   cd AutoForge
-   ```
-2. **Set Up a Virtual Environment (Optional but Recommended)**
-   ```bash
-   python -m venv venv
-   source venv/bin/activate  # On Windows: venv\Scripts\activate
-   ```
-3. **Install Dependencies**
-   ```bash
-   pip install -r requirements.txt
-   ```
+To install AutoForge, simply install the current version from PyPI:
+```bash
+   pip install autoforge
+```
-See the [requirements.txt](requirements.txt) file for a complete list of dependencies.
-The optimizer is built using JAX, which benefits from a CUDA-compatible GPU for optimal performance.
-Without a dedicated GPU the process can take significantly longer (up to 10-20x slower).
-If you have a GPU, you can install the GPU version of JAX by running:
+The optimizer is built using JAX, which benefits from a CUDA-compatible GPU for optimal performance.  \
+Without a dedicated GPU the process can take significantly longer (up to 10-20x slower).  \
+If you have a GPU, you can install the GPU version of autoforge by running:\
 ```bash
-pip install -U "jax[cuda12]"
+   pip install autoforge[gpu]
 ```
 Currently, jax gpu support is only available under linux. \
@@ -86,7 +76,7 @@ The script is run from the command line and accepts several arguments. Below is
 > If you want to limit the amount of colors the program can use, you can simply delete the colors you don't want from the CSV file.
 ```bash
-python auto_forge.py --input_image path/to/input_image.jpg --csv_file path/to/materials.csv --output_folder outputs
+autoforge --input_image path/to/input_image.jpg --csv_file path/to/materials.csv --output_folder outputs
 ```
 ### Command Line Arguments
@@ -95,18 +85,26 @@ python auto_forge.py --input_image path/to/input_image.jpg --csv_file path/to/ma
 - `--input_image`: **(Required)** Path to the input image.
 - `--csv_file`: **(Required)** Path to the CSV file containing material data. The CSV should include columns for the brand, name, color (hex code), and TD values.
 - `--output_folder`: **(Required)** Folder where output files will be saved.
-- `--iterations`: Number of optimization iterations (default: 20000).
-- `--learning_rate`: Learning rate for the optimizer (default: 5e-3).
+- `--iterations`: Number of optimization iterations (default: 5000).
+- `--best_loss_iterations`: Percentage of optimization iterations after which we start to record the best loss (default: 4000).
+  **Note:** We currently do this as we increase a penalty loss for color and color swaps as the time goes on. Without this the penalties would have no impact.
+- `--learning_rate`: Learning rate for the optimizer (default: 1e-2).
 - `--layer_height`: Layer thickness in millimeters (default: 0.04).
 - `--max_layers`: Maximum number of layers (default: 75).
   **Note:** This is about 3mm + the background height
 - `--background_height`: Height of the background in millimeters (default: 0.4).
   **Note:** The background height must be divisible by the layer height.
-- `--background_color`: Background color in hexadecimal format (default: `#8e9089` aka Bambulab Grey).
-- `--max_size`: Maximum dimension (width or height) for the resized target image (default: 512).
+- `--background_color`: Background color in hexadecimal format (default: `#000000` aka Black).
+  **Note:** The solver currently assumes that you have a solid color in the background, which means a color with a TD value of 4 or less (if you have a background height of 0.4)
+- `--output_size`: Maximum dimension for target image (default: 1024).
+- `--solver_size`: Maximum dimension for solver (fast) image (default: 128).
+  **Note:** We solve on a smaller size as this is many times faster, but also a bit less accurate. Increase if you need more accuracy.
 - `--decay`: Final tau value for the Gumbel-Softmax formulation (default: 0.01).
-- `--loss`: Loss function to use. Choices are `mse`, `perceptual`, or `perceptual_l1` (default: `mse`).
 - `--visualize`: Flag to enable live visualization of the composite image during optimization.
+- `--perform_pruning`: Perform pruning after optimization (default: True).
+- `--pruning_max_colors`: Max number of colors allowed after pruning (default: 10).
+- `--pruning_max_swaps`: Max number of swaps allowed after pruning (default: 20).
+- `--random_seed`: Random seed for reproducibility (default: 0 (disabled) ).
 ## Outputs
@@ -132,6 +130,9 @@ Although I would love to be it fully color compatible with hueforge, I don't thi
 ## License
 AutoForge © 2025 by Hendric Voss is licensed under [CC BY-NC-SA 4.0](https://creativecommons.org/licenses/by-nc-sa/4.0/).
+The software is provided as-is and comes with no warranty or guarantee of support.
+The
 ## Acknowledgements
@@ -152,5 +153,6 @@ AutoForge makes use of several open source libraries:
 Example Images:
 <a href="https://www.vecteezy.com/free-photos/anime-girl">Anime Girl Stock photos by Vecteezy</a>
 <a href="https://www.vecteezy.com/free-photos/nature">Nature Stock photos by Vecteezy</a>
+<a href="https://www.vecteezy.com/free-photos/ai-generated">Ai Generated Stock photos by Vecteezy</a>
 Happy printing!

{autoforge-1.1.2 → autoforge-1.2.0}/pyproject.toml RENAMED Viewed

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "AutoForge"
-version="1.1.2"
+version="1.2.0"
 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"
@@ -12,8 +12,8 @@ requires-python = ">=3.9"
 license = { text = "CC BY-NC-SA 4.0" }
 dependencies = [
   "numpy>=1.19.0",
-  "jax>=0.3.10",
-  "jaxlib>=0.3.10",
+  "torch>=2.6.0",
+  "torchvision>=0.21.0",
   "opencv-python>=4.2.0.32",
   "matplotlib>=3.3.0",
   "configargparse>=1.2.3",
@@ -27,10 +27,5 @@ dependencies = [
 [tool.setuptools]
 package-dir = {"" = "src"}
-[project.optional-dependencies]
-gpu = [
-  "jax[cuda12]",
-]
 [project.scripts]
 autoforge = "autoforge.auto_forge:main"

{autoforge-1.1.2 → autoforge-1.2.0}/src/AutoForge.egg-info/PKG-INFO RENAMED Viewed

@@ -1,14 +1,14 @@
 Metadata-Version: 2.2
 Name: AutoForge
-Version: 1.1.2
+Version: 1.2.0
 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
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 Requires-Dist: numpy>=1.19.0
-Requires-Dist: jax>=0.3.10
-Requires-Dist: jaxlib>=0.3.10
+Requires-Dist: torch>=2.6.0
+Requires-Dist: torchvision>=0.21.0
 Requires-Dist: opencv-python>=4.2.0.32
 Requires-Dist: matplotlib>=3.3.0
 Requires-Dist: configargparse>=1.2.3
@@ -17,8 +17,6 @@ Requires-Dist: pandas>=1.1.0
 Requires-Dist: scikit-image
 Requires-Dist: scikit-learn
 Requires-Dist: optax>=0.1.3
-Provides-Extra: gpu
-Requires-Dist: jax[cuda12]; extra == "gpu"
 # AutoForge
@@ -27,25 +25,31 @@ AutoForge is a Python tool for generating 3D printed layered models from an inpu
 **TLDR:** It uses a picture to generate a 3D layer image that you can print with a 3d printer. Similar to [Hueforge](https://shop.thehueforge.com/), but without the manual work (and without the artistic control).
+## Important Information
+We recently switched from JAX to PyTorch, to allow for a more streamlined development process. \
+This can have some unforeseen consequences, so please report any bugs you find.
 ## Example
 All examples use only the 13 BambuLab Basic filaments, currently available in Hueforge.
 <div style="display: flex; justify-content: center; gap: 20px;">
   <div style="text-align: center;">
     <h3>Input Image</h3>
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi.jpg" width="300" />
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature.jpg" width="300" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi.jpg" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature.jpg" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/cat.jpg" width="200" />
   </div>
   <div style="text-align: center;">
     <h3>Autoforge Output (75 color layers (3.0mm) + 10 gray background layers (0.4mm)) </h3>
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi_discretized.jpg" width="300" />
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature_discretized.jpg" width="300" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi_discretized.jpg" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature_discretized.jpg" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/cat_discretized.jpg" width="200" />
   </div>
   <div style="text-align: center;">
     <h3>Imported Hueforge View</h3>
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi_hueforge.png" width="300" />
-    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature_hueforge.png" width="300" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/lofi_hueforge.png" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/nature_hueforge.png" width="200" />
+    <img src="https://github.com/hvoss-techfak/AutoForge/blob/main/images/cat_hueforge.png" width="200" />
   </div>
 </div>
@@ -66,33 +70,17 @@ I would love to see what you made!
 ## Installation
-1. **Clone the Repository**
-   ```bash
-   git clone https://github.com/hvoss-techfak/AutoForge.git
-   cd AutoForge
-   ```
-2. **Set Up a Virtual Environment (Optional but Recommended)**
-   ```bash
-   python -m venv venv
-   source venv/bin/activate  # On Windows: venv\Scripts\activate
-   ```
-3. **Install Dependencies**
-   ```bash
-   pip install -r requirements.txt
-   ```
+To install AutoForge, simply install the current version from PyPI:
+```bash
+   pip install autoforge
+```
-See the [requirements.txt](requirements.txt) file for a complete list of dependencies.
-The optimizer is built using JAX, which benefits from a CUDA-compatible GPU for optimal performance.
-Without a dedicated GPU the process can take significantly longer (up to 10-20x slower).
-If you have a GPU, you can install the GPU version of JAX by running:
+The optimizer is built using JAX, which benefits from a CUDA-compatible GPU for optimal performance.  \
+Without a dedicated GPU the process can take significantly longer (up to 10-20x slower).  \
+If you have a GPU, you can install the GPU version of autoforge by running:\
 ```bash
-pip install -U "jax[cuda12]"
+   pip install autoforge[gpu]
 ```
 Currently, jax gpu support is only available under linux. \
@@ -108,7 +96,7 @@ The script is run from the command line and accepts several arguments. Below is
 > If you want to limit the amount of colors the program can use, you can simply delete the colors you don't want from the CSV file.
 ```bash
-python auto_forge.py --input_image path/to/input_image.jpg --csv_file path/to/materials.csv --output_folder outputs
+autoforge --input_image path/to/input_image.jpg --csv_file path/to/materials.csv --output_folder outputs
 ```
 ### Command Line Arguments
@@ -117,18 +105,26 @@ python auto_forge.py --input_image path/to/input_image.jpg --csv_file path/to/ma
 - `--input_image`: **(Required)** Path to the input image.
 - `--csv_file`: **(Required)** Path to the CSV file containing material data. The CSV should include columns for the brand, name, color (hex code), and TD values.
 - `--output_folder`: **(Required)** Folder where output files will be saved.
-- `--iterations`: Number of optimization iterations (default: 20000).
-- `--learning_rate`: Learning rate for the optimizer (default: 5e-3).
+- `--iterations`: Number of optimization iterations (default: 5000).
+- `--best_loss_iterations`: Percentage of optimization iterations after which we start to record the best loss (default: 4000).
+  **Note:** We currently do this as we increase a penalty loss for color and color swaps as the time goes on. Without this the penalties would have no impact.
+- `--learning_rate`: Learning rate for the optimizer (default: 1e-2).
 - `--layer_height`: Layer thickness in millimeters (default: 0.04).
 - `--max_layers`: Maximum number of layers (default: 75).
   **Note:** This is about 3mm + the background height
 - `--background_height`: Height of the background in millimeters (default: 0.4).
   **Note:** The background height must be divisible by the layer height.
-- `--background_color`: Background color in hexadecimal format (default: `#8e9089` aka Bambulab Grey).
-- `--max_size`: Maximum dimension (width or height) for the resized target image (default: 512).
+- `--background_color`: Background color in hexadecimal format (default: `#000000` aka Black).
+  **Note:** The solver currently assumes that you have a solid color in the background, which means a color with a TD value of 4 or less (if you have a background height of 0.4)
+- `--output_size`: Maximum dimension for target image (default: 1024).
+- `--solver_size`: Maximum dimension for solver (fast) image (default: 128).
+  **Note:** We solve on a smaller size as this is many times faster, but also a bit less accurate. Increase if you need more accuracy.
 - `--decay`: Final tau value for the Gumbel-Softmax formulation (default: 0.01).
-- `--loss`: Loss function to use. Choices are `mse`, `perceptual`, or `perceptual_l1` (default: `mse`).
 - `--visualize`: Flag to enable live visualization of the composite image during optimization.
+- `--perform_pruning`: Perform pruning after optimization (default: True).
+- `--pruning_max_colors`: Max number of colors allowed after pruning (default: 10).
+- `--pruning_max_swaps`: Max number of swaps allowed after pruning (default: 20).
+- `--random_seed`: Random seed for reproducibility (default: 0 (disabled) ).
 ## Outputs
@@ -154,6 +150,9 @@ Although I would love to be it fully color compatible with hueforge, I don't thi
 ## License
 AutoForge © 2025 by Hendric Voss is licensed under [CC BY-NC-SA 4.0](https://creativecommons.org/licenses/by-nc-sa/4.0/).
+The software is provided as-is and comes with no warranty or guarantee of support.
+The
 ## Acknowledgements
@@ -174,5 +173,6 @@ AutoForge makes use of several open source libraries:
 Example Images:
 <a href="https://www.vecteezy.com/free-photos/anime-girl">Anime Girl Stock photos by Vecteezy</a>
 <a href="https://www.vecteezy.com/free-photos/nature">Nature Stock photos by Vecteezy</a>
+<a href="https://www.vecteezy.com/free-photos/ai-generated">Ai Generated Stock photos by Vecteezy</a>
 Happy printing!

autoforge-1.2.0/src/AutoForge.egg-info/SOURCES.txt ADDED Viewed

@@ -0,0 +1,26 @@
+README.md
+pyproject.toml
+src/AutoForge.egg-info/PKG-INFO
+src/AutoForge.egg-info/SOURCES.txt
+src/AutoForge.egg-info/dependency_links.txt
+src/AutoForge.egg-info/entry_points.txt
+src/AutoForge.egg-info/requires.txt
+src/AutoForge.egg-info/top_level.txt
+src/autoforge/__init__.py
+src/autoforge/__main__.py
+src/autoforge/auto_forge.py
+src/autoforge/auto_forge_gradio.py
+src/autoforge/color_fit_bayesian.py
+src/autoforge/filamentcolors_library.py
+src/autoforge/old_jax_files.py
+src/autoforge/Helper/FilamentHelper.py
+src/autoforge/Helper/ImageHelper.py
+src/autoforge/Helper/OptimizerHelper.py
+src/autoforge/Helper/OutputHelper.py
+src/autoforge/Helper/PruningHelper.py
+src/autoforge/Helper/__init__.py
+src/autoforge/Loss/LossFunctions.py
+src/autoforge/Loss/PerceptionLoss.py
+src/autoforge/Loss/__init__.py
+src/autoforge/Modules/Optimizer.py
+src/autoforge/Modules/__init__.py

{autoforge-1.1.2 → autoforge-1.2.0}/src/AutoForge.egg-info/requires.txt RENAMED Viewed

@@ -1,6 +1,6 @@
 numpy>=1.19.0
-jax>=0.3.10
-jaxlib>=0.3.10
+torch>=2.6.0
+torchvision>=0.21.0
 opencv-python>=4.2.0.32
 matplotlib>=3.3.0
 configargparse>=1.2.3
@@ -9,6 +9,3 @@ pandas>=1.1.0
 scikit-image
 scikit-learn
 optax>=0.1.3
-[gpu]
-jax[cuda12]

autoforge 1.1.2__tar.gz → 1.2.0__tar.gz

autoforge 1.1.2tar.gz → 1.2.0tar.gz