Merge pull request #2 from Kek-Colossi/master

Some fixes

Author: Olexasha

.env

@@ -1,12 +1,3 @@
-# Пути к файлам
-ORIGINAL_VIDEO=./data/input_video/bob_ru.mp4
-AUDIO_PATH=./data/audio
-TMP_VIDEO_PATH=./data/tmp_video
-FINAL_VIDEO=./data/output_video/final_video.mp4
-BATCH_VIDEO_PATH=./data/video_batches
-INPUT_BATCHES_DIR=./data/default_frame_batches
-OUTPUT_BATCHES_DIR=./data/upscaled_frame_batches
-
 # Файловые параметры апскейлера
 RESOLUTION=4K
 START_BATCH_TO_UPSCALE=1
@@ -16,8 +7,7 @@ FRAMES_PER_BATCH=1000
 ALLOWED_THREADS=6
 
 # Настройка апскейла
-MODEL_DIR = ./src/utils/realesrgan/models
 MODEL_NAME = realesr-animevideov3
-REALESRGAN_SCRIPT = ./src/utils/realesrgan/realesrgan-linux/realesrgan-ncnn-vulkan
+OS=linux  # linux , win or macos
 UPSCALE_FACTOR=2
 OUTPUT_IMAGE_FORMAT=jpg

.github/ISSUE_TEMPLATE/автотесты.md

@@ -0,0 +1,33 @@
+---
+name: "Автотестирование"
+about: "Расскажи об автотестах"
+title: "[autotests]:"
+projects: ["Anime Enhancement Project Development"]
+labels: ["автотесты"]
+body:
+    type: markdown
+    value: |
+      Спасибо что уделяешь время нашему проекту!
+---
+
+### Описание задачи и кейсов:
+
+
+
+### Скриншоты / Видео (если есть визуальные изменения)
+
+(Прикрепите изображения или видео, если интерфейс изменился)
+
+
+### Памятка:
+- Использовать фреймворк pytest
+- Все тесты уносить в директорию `src/autotests/tests/<раздел_логики>/test_*.py`
+- Все имена классов для тестов должны начинаться так: `class TestName*:`
+- Все имена тестов должны начинаться так: `test_*.py`
+- Все, что создаём, за собой подчищаем. УДАЛЯЕМ ВСЕ ЛИШНЕЕ
+- Тесты должны быть независимы друг от друга
+- Тесты друг друга не вызывают никогда!
+- Разносим логику для тестов в предназначенные для них директории, в которых создаём понятные имена:
+> - класс/метод помогают тесту с какой-то логико? - уносим в `helpers`
+> - класс/метод проверяют какую-то весомую часть кода или строк попросту много? - уносим в `checkers`
+> - класс/метод используется перед или после теста? - уносим в `fixtures`

.github/ISSUE_TEMPLATE/баг-репорт.md

@@ -1,16 +1,20 @@
 ---
-name: Баг репорт
-about: Расскажи о баге в 2х словах
-title: Баг репорт
-labels: ''
-assignees: ''
-
+name: Багфикс
+about: "Расскажи о баге"
+title: "[bug]:"
+projects: ["Anime Enhancement Project Development"]
+labels: ["баг"]
+body:
+    type: markdown
+    value: |
+      Спасибо что уделяешь время нашему проекту!
 ---
 
 ### Описание бага
 
-**Краткое описание:**
+**Подробное описание:**
 Опишите суть бага простыми словами (1–2 предложения).
+При необходимости
 
 **Шаги для воспроизведения:**
 1. Опишите каждый шаг, чтобы воспроизвести проблему.
@@ -21,3 +25,8 @@ assignees: ''
 
 **Ожидаемый результат:**
 Опишите, как система должна была вести себя вместо фактического результата.
+
+
+### Скриншоты / Видео (если есть визуальные изменения)
+
+(Прикрепите изображения или видео, если интерфейс изменился)

.github/ISSUE_TEMPLATE/новая-фича.md

@@ -1,21 +1,20 @@
 ---
-name: Новая фича
-about: Расскажи о фиче в 2х словах
-title: Issue на новую фичу
-labels: ''
-assignees: ''
-
+name: "Разработка"
+about: "Расскажи о фиче"
+title: "[dev]:"
+projects: ["Anime Enhancement Project Development"]
+labels: ["новая фича"]
+body:
+    type: markdown
+    value: |
+      Спасибо что уделяешь время нашему проекту!
 ---
 
-### Что сделано
-
-Краткое описание изменений и зачем они нужны.
+### Описание:
 
-### Задача
 
-Ссылка на задачу (issue, тикет и т.д.).
 
-### Чеклист
+### Памятка:
 
 - [ ] Код работает корректно
 - [ ] Добавлены или обновлены тесты (если нужно)

.github/pull_request_template.md

@@ -0,0 +1,24 @@
+### Описание изменений
+Кратко опишите, какие изменения были внесены, зачем они были нужны и как они решают поставленную задачу.
+
+### Ссылка на задачу
+Связанная задача: [#номер_задачи](ссылка_на_задачу) из проекта [Anime Enhancement Project Development]([https://github.com/ссылка_на_канбан_доску](https://github.com/orgs/Kek-Colossi/projects/1)).
+
+### Контекст для ревью
+- Основные изменения: перечислите ключевые изменения и укажите, на что обратить внимание при проверке.
+- Ожидаемое поведение: опишите, какое поведение кода ожидается после внесения этих изменений.
+
+### Задачи для завершения
+- [ ] Убедитесь, что в issue заполнены приоритет, сложность и тип задачи.
+- [ ] Назначьте ревьюера: @имя_ревьюера.
+- [ ] Проведите локальные тесты для проверки работоспособности изменений если они.
+
+### Тестирование
+Опишите, как тестировались изменения, и приложите результаты, если это необходимо. Укажите команды, которые использовались для запуска тестов.
+
+### Дополнительные замечания
+Добавьте любые дополнительные комментарии или инструкции для ревьюера. Например, если есть нюансы в коде или временные решения, которые нужно учесть.
+
+---
+
+**Примечание**: Пожалуйста, убедитесь, что данное изменение соответствует общему стилю кода и стандартам проекта. 

main.py

@@ -27,6 +27,7 @@ def main():
     start_batch = START_BATCH_TO_UPSCALE
     end_batch = 0
     tmp_videos_counter = 1
+    # Экземпляр tmp видео ~32 минуты, обновляется после ~32 минут
     tmp_builder = VideoHandler(
         fps=fps, tmp_video_name=f"tmp_video_{tmp_videos_counter}"
     )

src/config/settings.py

@@ -6,17 +6,13 @@
 ROOT_DIR = os.getcwd()
 
 # Пути к файлам
-ORIGINAL_VIDEO = os.getenv("ORIGINAL_VIDEO", f"{ROOT_DIR}/data/input_video/bob_ru.mp4")
-AUDIO_PATH = os.getenv("ORIGINAL_AUDIO", f"{ROOT_DIR}/data/audio")
-TMP_VIDEO_PATH = os.getenv("TMP_FINAL_VIDEO", f"{ROOT_DIR}/data/tmp_video")
-FINAL_VIDEO = os.getenv("FINAL_VIDEO", f"{ROOT_DIR}/data/output_video/final_video.mp4")
-BATCH_VIDEO_PATH = os.getenv("BATCH_VIDEO_PATH", f"{ROOT_DIR}/data/video_batches")
-INPUT_BATCHES_DIR = os.getenv(
-    "INPUT_BATCHES_DIR", f"{ROOT_DIR}/data/default_frame_batches"
-)
-OUTPUT_BATCHES_DIR = os.getenv(
-    "OUTPUT_BATCHES_DIR", f"{ROOT_DIR}/data/upscaled_frame_batches"
-)
+ORIGINAL_VIDEO = os.path.join(ROOT_DIR, "data", "input_video", "bob_ru.mp4")
+AUDIO_PATH = os.path.join(ROOT_DIR, "data", "audio")
+TMP_VIDEO_PATH = os.path.join(ROOT_DIR, "data", "tmp_video")
+FINAL_VIDEO = os.path.join(ROOT_DIR, "data", "output_video", "final_video.mp4")
+BATCH_VIDEO_PATH = os.path.join(ROOT_DIR, "data", "video_batches")
+INPUT_BATCHES_DIR = os.path.join(ROOT_DIR, "data", "default_frame_batches")
+OUTPUT_BATCHES_DIR = os.path.join(ROOT_DIR, "data", "upscaled_frame_batches")
 
 # Файловые параметры апскейлера
 RESOLUTION = os.getenv("RESOLUTION", "4K")
@@ -27,11 +23,11 @@
 ALLOWED_THREADS = int(os.getenv("ALLOWED_THREADS", 6))
 
 # Настройка апскейла
-MODEL_DIR = os.getenv("MODEL_DIR", f"{ROOT_DIR}/src/utils/realesrgan/models")
+MODEL_DIR = os.path.join(ROOT_DIR, "src", "utils", "realesrgan", "models")
 MODEL_NAME = os.getenv("MODEL_NAME", "realesr-animevideov3")
-REALESRGAN_SCRIPT = os.getenv(
-    "REALSESRGAN_DIR",
-    f"{ROOT_DIR}/src/utils/realesrgan/realesrgan-linux/realesrgan-ncnn-vulkan",
+_OS = os.getenv("_OS", "linux")
+REALESRGAN_SCRIPT = os.path.join(
+    ROOT_DIR, "src", "utils", "realesrgan", f"realesrgan-{_OS}", f"realesrgan-ncnn-vulkan{".exe" if _OS == 'win' else ""}"
 )
 UPSCALE_FACTOR = int(os.getenv("UPSCALE_FACTOR", 2))
 OUTPUT_IMAGE_FORMAT = os.getenv("OUTPUT_IMAGE_FORMAT", "jpg")

src/utils/realesrgan/realesrgan-win/README_windows.md

@@ -0,0 +1,42 @@
+# RealESRGAN
+
+We have provided the following models:
+
+1. realesr-animevideov3 (default)
+2. realesrgan-x4plus
+3. realesrgan-x4plus-anime
+
+Command:
+
+1. ./realesrgan-ncnn-vulkan.exe -i input.jpg -o output.png
+2. ./realesrgan-ncnn-vulkan.exe -i input.jpg -o output.png -n realesr-animevideov3
+3. ./realesrgan-ncnn-vulkan.exe -i input_folder -o outputfolder -n realesr-animevideov3 -s 2 -f jpg
+4. ./realesrgan-ncnn-vulkan.exe -i input_folder -o outputfolder -n realesr-animevideov3 -s 4 -f jpg
+
+
+Commands for enhancing anime videos:
+
+1. Use ffmpeg to extract frames from a video (Remember to create the folder `tmp_frames` ahead)
+
+    ffmpeg -i onepiece_demo.mp4 -qscale:v 1 -qmin 1 -qmax 1 -vsync 0 tmp_frames/frame%08d.jpg
+
+2. Inference with Real-ESRGAN executable file (Remember to create the folder `out_frames` ahead)
+
+    ./realesrgan-ncnn-vulkan.exe -i tmp_frames -o out_frames -n realesr-animevideov3 -s 2 -f jpg
+
+3. Merge the enhanced frames back into a video
+
+    ffmpeg -i out_frames/frame%08d.jpg -i onepiece_demo.mp4 -map 0:v:0 -map 1:a:0 -c:a copy -c:v libx264 -r 23.98 -pix_fmt yuv420p output_w_audio.mp4
+
+------------------------
+
+GitHub: https://github.com/xinntao/Real-ESRGAN/
+Paper: https://arxiv.org/abs/2107.10833
+
+------------------------
+
+This executable file is **portable** and includes all the binaries and models required. No CUDA or PyTorch environment is needed.
+
+Note that it may introduce block inconsistency (and also generate slightly different results from the PyTorch implementation), because this executable file first crops the input image into several tiles, and then processes them separately, finally stitches together.
+
+This executable file is based on the wonderful [Tencent/ncnn](https://github.com/Tencent/ncnn) and [realsr-ncnn-vulkan](https://github.com/nihui/realsr-ncnn-vulkan) by [nihui](https://github.com/nihui).