項(xiàng)目名稱：AI 動(dòng)漫風(fēng)格生成器
技術(shù)棧：Python (Flask, OpenCV, PyTorch), JavaScript (HTML/CSS), Stable Diffusion, ControlNet
項(xiàng)目描述
開發(fā)了一款基于深度學(xué)習(xí)的 AI 動(dòng)漫風(fēng)格生成器，支持用戶上傳圖片并生成動(dòng)漫風(fēng)格的線稿和完整畫風(fēng)。項(xiàng)目集成了 StableDiffusion 和 ControlNet 模型，通過 Flask 提供后端服務(wù)，前端實(shí)現(xiàn)圖片上傳、線稿生成和畫風(fēng)遷移功能。
核心功能
圖片上傳與預(yù)覽：支持用戶拖拽或選擇圖片上傳，實(shí)時(shí)預(yù)覽上傳的圖片。
線稿生成：基于 OpenCV 實(shí)現(xiàn)圖片線稿生成，提供黑白素描效果。
畫風(fēng)遷移：集成 Stable Diffusion 和 ControlNet 模型，根據(jù)用戶輸入的指令生成動(dòng)漫風(fēng)格圖片。
結(jié)果展示與下載：生成結(jié)果實(shí)時(shí)展示，并提供下載鏈接。
技術(shù)亮點(diǎn)
深度學(xué)習(xí)模型集成：
使用 PyTorch 加載和運(yùn)行 Stable Diffusion 和 ControlNet 模型，支持 GPU 加速。
動(dòng)態(tài)選擇數(shù)據(jù)類型（torch.float16 或 torch.float32），確保模型在 CPU 和 GPU 上均可運(yùn)行。
前后端交互：
使用 Flask 構(gòu)建 RESTful API，處理圖片上傳、線稿生成和畫風(fēng)遷移請求。
前端通過 Fetch API 與后端交互，實(shí)現(xiàn)異步請求和動(dòng)態(tài)內(nèi)容更新。
文件處理與存儲(chǔ)：
使用 OpenCV 處理圖片，生成線稿并保存到指定目錄。
通過 Flask 的 send_from_directory 提供文件下載功能。
用戶體驗(yàn)優(yōu)化：
實(shí)現(xiàn)拖拽上傳、實(shí)時(shí)預(yù)覽和錯(cuò)誤提示功能，提升用戶交互體驗(yàn)。
使用 CSS 美化界面，確保頁面布局簡潔美觀。
項(xiàng)目成果
成功實(shí)現(xiàn)圖片上傳、線稿生成和畫風(fēng)遷移功能，生成效果符合預(yù)期。
支持多種圖片格式（PNG、JPG、JPEG），最大上傳文件限制為 16MB。
項(xiàng)目代碼結(jié)構(gòu)清晰，模塊化設(shè)計(jì)便于維護(hù)和擴(kuò)展。