An open API service indexing awesome lists of open source software.

https://github.com/lahi-ru/image_style_transfer

This project implements neural style transfer using PyTorch. The model combines the content of one image with the style of another to generate a stylized output.
https://github.com/lahi-ru/image_style_transfer

ai-art artificial-intelligence computer-vision deep-learning generative-model image-processing neutral-style-transfer python pytorch

Last synced: 8 months ago
JSON representation

This project implements neural style transfer using PyTorch. The model combines the content of one image with the style of another to generate a stylized output.

Awesome Lists containing this project

README

          

# Image Style Transfer Project

This project implements neural style transfer using PyTorch. The model combines the content of one image with the style of another to generate a stylized output.

## Features
- Uses a pre-trained VGG-19 model for feature extraction.
- Customizable content and style images.
- Outputs high-quality stylized images.

## Requirements
- Python 3.8 or higher
- PyTorch
- Torchvision
- Pillow

## Installation
1. Clone the repository:
```bash
git clone https://github.com/username/image-style-transfer.git
cd image-style-transfer

2. Install the required dependencies:
```bash
pip install -r requirements.txt
```

## Usage

1. Add your `content.jpg` and `style.jpg` images to the root directory.
2. Run the main script:

```bash
python main.py
```

3. The stylized image will be saved in the `output/` directory.

## Example

### Content Image:
*(Add image preview here)*

### Style Image:
*(Add image preview here)*

### Stylized Output:
*(Add image preview here)*

## Contributing

Contributions are welcome! Please feel free to submit a pull request or report issues.

## License

This project is licensed under the MIT License. See the LICENSE file for details.