Machine-Learning
/
aiuNN
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

finetune_class #1

Merged
Fabel merged 96 commits from finetune_class into develop 2025-02-26 12:13:09 +00:00
Conversation 0 Commits 96 Files Changed 15 +3 -3
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit e28fd3d330 - Show all commits

6
src/aiunn/inference.py
View File

@ -131,6 +131,6 @@ class ImageUpscaler:
return self.postprocess_image(output_tensor)
# Example usage:
# upscaler = ImageUpscaler(model_path="/path/to/best_model")
# upscaled_image = upscaler.upscale_image("/path/to/low_res_image.jpg")
# upscaled_image.save("/path/to/upscaled_image.jpg")
upscaler = ImageUpscaler(model_path="/root/vision/aiuNN/best_model")
upscaled_image = upscaler.upscale_image("/root/vision/aiuNN/input.jpg")
upscaled_image.save("upscaled_image.jpg")