Welcome to the CSCI4140 Group Project Page by Group 15. This project is to build an online user-specific image editor for which user can edit their own images with some specific objects or try to transfer their images into some ideal style depending on their own choices. This document provides an overview of this project, including the setting up steps and functions of the website.
In the editor, you can upload your own content or style images. Of course, if you have no idea about what to choose, you can enter some keywords and search through Google. Our website will show you the first 5 images according to your keywords, and you can choose one as your selection.
Compared to the traditional image editor, which can only deal with the whole image, our editor can support users in searching for some objects' names and directly editing the target region. You can input some keywords to search whether the target object exists, and then our system will show a list of target region images to allow the user to choose. You can either use basic CamanJS or try style transfer and then combine the edited images with the original ones. The keywords of our systems are based on the COCO2017 datasets1.
In the style transfer part, users can upload their own style images (please use .jpg
format), try some provided style images like Starry Night, or select some images from Google. Yeah, users can also make changes to the whole image or just change some target regions.
- Please download the pre-trained style transfer model's parameters here, and save the
model
folder into/mttm/model
. - Please download the pre-trained object detection model from here2, and save it to
/mttm/Pix2SeqV2-Pytorch-master/weights
. - Run
pip install -r requirements.txt
to download basic Python libraries. You may need to manually download some additional requirements that may not included. - Move to
/mttm/
and runnpm install
to download NodeJS requirements. - To run the front-end server locally, you can go to
/html/
and runhttp-server
(you may need to install it throughnpm install http-server
), then you can visit the website athttp://127.0.0.1:8080
or other local URL. You can also usehttp-server --cors
to allow CORS. - To run the back-end server locally, you can go to
/mttm/
and runnode app.js
to create a port athttp:/localhost:3000
. - To run the Google Image Search function, you need to follow the steps at here to get your own Google Image Search API key and Google project cx. Then, fill in the missing part in the
/mttm/code/search_image.py
.
- You may find the dataset here. You can also find all the supported keywords for our editor here
- The pre-trained pix2Seq model is originally implemented here.
Thank you for visiting this project. If you have any further questions, please feel free to contact me.