Skip to content

Conversation

ChitteshKumar
Copy link
Contributor

Description

This pull request introduces a script for performing Neural Style Transfer using a pre-trained model from TensorFlow Hub. The code implements the following key features:

  • Model Loading: It loads a pre-trained model from TensorFlow Hub (arbitrary-image-stylization-v1-256) for style transfer.
  • Image Uploading: Users can upload content and style images through interactive widgets. The images are saved in designated folders (images for content images and styles for style images) for further processing.
  • Image Processing: It includes a function to load and preprocess images, converting them to the appropriate tensor format required by the model.
  • Style Transfer Execution: The uploaded content and style images are passed to the model, generating a styled image that combines the content of the content image with the artistic style of the style image.
  • Result Display: The script displays the input content and style images, as well as the final styled output.
  • Image Export: The generated styled image is saved in a generatedImages folder for user access.

This script enhances the user experience by providing an interactive interface for image uploads and visualizations, facilitating easy experimentation with different styles and content images.

Fixes #1252 issue.

Type of change

  • [✅] Added a new machine learning frameworks, libraries or software.

Checklist:

  • [✅ ] My code follows the style guidelines of this project
  • [ ✅] I have performed a self-review of my own code
  • [ ✅] I have commented my code, particularly in hard-to-understand areas
  • [ ✅] My changes generate no new warnings

Copy link

github-actions bot commented Oct 6, 2024

Thank you for submitting your pull request! 🙌 We'll review it as soon as possible. In the meantime, please ensure that your changes align with our CONTRIBUTING.md. If there are any specific instructions or feedback regarding your PR, we'll provide them here. Thanks again for your contribution! 😊

@ChitteshKumar
Copy link
Contributor Author

Hi @sanjay-kv , I have completed the assigned task #1252 . And created a pull request, kindly check it and provide feedback.
Looking forward to getting the request merged

Thank you

@sanjay-kv sanjay-kv merged commit 8e7a848 into recodehive:main Oct 7, 2024
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

💡[Feature]: Add Neural Style Transfer Feature for Creative AI and Beginner Learning

2 participants