├── .dockerignore ├── .github └── workflows │ ├── bandit.yml │ ├── ci.yml │ ├── dependency-review.yml │ ├── docker.yml │ └── python-publish.yml ├── .gitignore ├── Dockerfile ├── README.md ├── config.conf ├── filaments.csv ├── images ├── attribution.txt ├── cat.jpg ├── cat_discretized.png ├── chameleon.jpg ├── chameleon_discretized.png ├── lofi.jpg ├── lofi_discretized.png ├── nature.jpg ├── nature_discretized.png └── test_images │ ├── bird.jpg │ ├── cat.jpg │ ├── chameleon.jpg │ ├── colorful_women.jpg │ ├── corgi.jpg │ ├── family.jpg │ ├── group_of_people.jpg │ ├── landscape.jpg │ ├── lofi.jpg │ ├── mandala.jpg │ ├── nature.jpg │ ├── ocean.jpg │ ├── oil_painting.jpg │ ├── panorama.jpg │ └── tourist.jpg ├── pyproject.toml ├── requirements.txt ├── run_all_demo_images.sh ├── src └── autoforge │ ├── Helper │ ├── AmpUtils.py │ ├── CAdamW.py │ ├── FilamentHelper.py │ ├── Heightmaps │ │ ├── ChristofidesHeightMap.py │ │ ├── DepthEstimateHeightMap.py │ │ └── __init__.py │ ├── ImageHelper.py │ ├── OptimizerHelper.py │ ├── OtherHelper.py │ ├── OutputHelper.py │ ├── PruningHelper.py │ ├── __init__.py │ └── filamentcolors_library.py │ ├── Init_comparer │ ├── __init__.py │ ├── auto_forge_init_check.py │ └── plot_results.py │ ├── Loss │ ├── LossFunctions.py │ ├── PerceptionLoss.py │ └── __init__.py │ ├── Misc │ ├── __init__.py │ └── color_fit_all.py │ ├── Modules │ ├── Optimizer.py │ └── __init__.py │ ├── __init__.py │ ├── __main__.py │ ├── auto_forge.py │ └── priority_mask_editor.py └── tests ├── conftest.py ├── test_amp_autocast_probe.py ├── test_cadamw.py ├── test_christofides_height_map.py ├── test_christofides_heightmap.py ├── test_depth_adjusted_height_map.py ├── test_depth_estimate_height_map.py ├── test_end_to_end_optimizer.py ├── test_filament_and_output.py ├── test_filament_helper.py ├── test_flatforge.py ├── test_heightmap_init.py ├── test_image_helper.py ├── test_loss_functions.py ├── test_optimizer_helper.py ├── test_optimizer_module.py ├── test_optimizer_smoke.py ├── test_output_helper.py ├── test_output_helper_extended.py ├── test_perception_and_pruning.py ├── test_property_based.py └── test_pruning_helper.py /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/.dockerignore -------------------------------------------------------------------------------- /.github/workflows/bandit.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/.github/workflows/bandit.yml -------------------------------------------------------------------------------- /.github/workflows/ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/.github/workflows/ci.yml -------------------------------------------------------------------------------- /.github/workflows/dependency-review.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/.github/workflows/dependency-review.yml -------------------------------------------------------------------------------- /.github/workflows/docker.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/.github/workflows/docker.yml -------------------------------------------------------------------------------- /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/.gitignore -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/Dockerfile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/README.md -------------------------------------------------------------------------------- /config.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/config.conf -------------------------------------------------------------------------------- /filaments.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/filaments.csv -------------------------------------------------------------------------------- /images/attribution.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/attribution.txt -------------------------------------------------------------------------------- /images/cat.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/cat.jpg -------------------------------------------------------------------------------- /images/cat_discretized.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/cat_discretized.png -------------------------------------------------------------------------------- /images/chameleon.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/chameleon.jpg -------------------------------------------------------------------------------- /images/chameleon_discretized.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/chameleon_discretized.png -------------------------------------------------------------------------------- /images/lofi.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/lofi.jpg -------------------------------------------------------------------------------- /images/lofi_discretized.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/lofi_discretized.png -------------------------------------------------------------------------------- /images/nature.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/nature.jpg -------------------------------------------------------------------------------- /images/nature_discretized.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/nature_discretized.png -------------------------------------------------------------------------------- /images/test_images/bird.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/test_images/bird.jpg -------------------------------------------------------------------------------- /images/test_images/cat.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/test_images/cat.jpg -------------------------------------------------------------------------------- /images/test_images/chameleon.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/test_images/chameleon.jpg -------------------------------------------------------------------------------- /images/test_images/colorful_women.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/test_images/colorful_women.jpg -------------------------------------------------------------------------------- /images/test_images/corgi.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/test_images/corgi.jpg -------------------------------------------------------------------------------- /images/test_images/family.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/test_images/family.jpg -------------------------------------------------------------------------------- /images/test_images/group_of_people.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/test_images/group_of_people.jpg -------------------------------------------------------------------------------- /images/test_images/landscape.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/test_images/landscape.jpg -------------------------------------------------------------------------------- /images/test_images/lofi.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/test_images/lofi.jpg -------------------------------------------------------------------------------- /images/test_images/mandala.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/test_images/mandala.jpg -------------------------------------------------------------------------------- /images/test_images/nature.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/test_images/nature.jpg -------------------------------------------------------------------------------- /images/test_images/ocean.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/test_images/ocean.jpg -------------------------------------------------------------------------------- /images/test_images/oil_painting.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/test_images/oil_painting.jpg -------------------------------------------------------------------------------- /images/test_images/panorama.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/test_images/panorama.jpg -------------------------------------------------------------------------------- /images/test_images/tourist.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/images/test_images/tourist.jpg -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_all_demo_images.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/run_all_demo_images.sh -------------------------------------------------------------------------------- /src/autoforge/Helper/AmpUtils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Helper/AmpUtils.py -------------------------------------------------------------------------------- /src/autoforge/Helper/CAdamW.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Helper/CAdamW.py -------------------------------------------------------------------------------- /src/autoforge/Helper/FilamentHelper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Helper/FilamentHelper.py -------------------------------------------------------------------------------- /src/autoforge/Helper/Heightmaps/ChristofidesHeightMap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Helper/Heightmaps/ChristofidesHeightMap.py -------------------------------------------------------------------------------- /src/autoforge/Helper/Heightmaps/DepthEstimateHeightMap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Helper/Heightmaps/DepthEstimateHeightMap.py -------------------------------------------------------------------------------- /src/autoforge/Helper/Heightmaps/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/autoforge/Helper/ImageHelper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Helper/ImageHelper.py -------------------------------------------------------------------------------- /src/autoforge/Helper/OptimizerHelper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Helper/OptimizerHelper.py -------------------------------------------------------------------------------- /src/autoforge/Helper/OtherHelper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Helper/OtherHelper.py -------------------------------------------------------------------------------- /src/autoforge/Helper/OutputHelper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Helper/OutputHelper.py -------------------------------------------------------------------------------- /src/autoforge/Helper/PruningHelper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Helper/PruningHelper.py -------------------------------------------------------------------------------- /src/autoforge/Helper/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/autoforge/Helper/filamentcolors_library.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Helper/filamentcolors_library.py -------------------------------------------------------------------------------- /src/autoforge/Init_comparer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/autoforge/Init_comparer/auto_forge_init_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Init_comparer/auto_forge_init_check.py -------------------------------------------------------------------------------- /src/autoforge/Init_comparer/plot_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Init_comparer/plot_results.py -------------------------------------------------------------------------------- /src/autoforge/Loss/LossFunctions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Loss/LossFunctions.py -------------------------------------------------------------------------------- /src/autoforge/Loss/PerceptionLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Loss/PerceptionLoss.py -------------------------------------------------------------------------------- /src/autoforge/Loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/autoforge/Misc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/autoforge/Misc/color_fit_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Misc/color_fit_all.py -------------------------------------------------------------------------------- /src/autoforge/Modules/Optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/Modules/Optimizer.py -------------------------------------------------------------------------------- /src/autoforge/Modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/autoforge/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/autoforge/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/__main__.py -------------------------------------------------------------------------------- /src/autoforge/auto_forge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/auto_forge.py -------------------------------------------------------------------------------- /src/autoforge/priority_mask_editor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/src/autoforge/priority_mask_editor.py -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/test_amp_autocast_probe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_amp_autocast_probe.py -------------------------------------------------------------------------------- /tests/test_cadamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_cadamw.py -------------------------------------------------------------------------------- /tests/test_christofides_height_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_christofides_height_map.py -------------------------------------------------------------------------------- /tests/test_christofides_heightmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_christofides_heightmap.py -------------------------------------------------------------------------------- /tests/test_depth_adjusted_height_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_depth_adjusted_height_map.py -------------------------------------------------------------------------------- /tests/test_depth_estimate_height_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_depth_estimate_height_map.py -------------------------------------------------------------------------------- /tests/test_end_to_end_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_end_to_end_optimizer.py -------------------------------------------------------------------------------- /tests/test_filament_and_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_filament_and_output.py -------------------------------------------------------------------------------- /tests/test_filament_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_filament_helper.py -------------------------------------------------------------------------------- /tests/test_flatforge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_flatforge.py -------------------------------------------------------------------------------- /tests/test_heightmap_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_heightmap_init.py -------------------------------------------------------------------------------- /tests/test_image_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_image_helper.py -------------------------------------------------------------------------------- /tests/test_loss_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_loss_functions.py -------------------------------------------------------------------------------- /tests/test_optimizer_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_optimizer_helper.py -------------------------------------------------------------------------------- /tests/test_optimizer_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_optimizer_module.py -------------------------------------------------------------------------------- /tests/test_optimizer_smoke.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_optimizer_smoke.py -------------------------------------------------------------------------------- /tests/test_output_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_output_helper.py -------------------------------------------------------------------------------- /tests/test_output_helper_extended.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_output_helper_extended.py -------------------------------------------------------------------------------- /tests/test_perception_and_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_perception_and_pruning.py -------------------------------------------------------------------------------- /tests/test_property_based.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_property_based.py -------------------------------------------------------------------------------- /tests/test_pruning_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hvoss-techfak/AutoForge/HEAD/tests/test_pruning_helper.py --------------------------------------------------------------------------------