xxxxxxxxxx
34
let inputImage;
let stylizationModel;
function preload() {
inputImage = loadImage('Kitten_0.png'); // Replace 'input.jpg' with your input image file
}
function setup() {
createCanvas(inputImage.width, inputImage.height);
image(inputImage, 0, 0);
// Load the pre-trained style transfer model
ml5.styleTransfer('arbitrary', modelLoaded);
}
function modelLoaded(model) {
stylizationModel = model;
console.log('Style transfer model loaded.');
}
function mouseReleased() {
if (stylizationModel) {
// Perform style transfer
stylizationModel.transfer(inputImage, function (err, stylizedImage) {
if (err) {
console.error('Style transfer error:', err);
} else {
// Display the stylized image on the canvas
image(stylizedImage, 0, 0);
}
});
}
}