File size: 2,134 Bytes
1f63901
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
387b6e9
1f63901
31fa226
 
1f63901
 
c1956a2
 
 
31fa226
 
1f63901
 
 
03bdb09
1f63901
 
 
 
387b6e9
 
1f63901
 
 
 
387b6e9
 
1dc0436
387b6e9
efb8e8f
 
387b6e9
6b9ae8a
 
1dc0436
 
 
 
6b9ae8a
1f63901
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
---
license: mit
pipeline_tag: image-to-image
library_name: pytorch
tags:
  - computer-vision
  - image-to-image
  - image-restoration
  - image-enhancement
  - super-resolution
  - comfyui
  - pytorch
  - swinir
  - transformer
  - anime-upscale
---

# SnJake Anime Upscale

A **experimental** lightweight upscaler (x2) for anime/illustration images. Designed for clean, sharp results with minimal artifacts. V2 is slightly sharper and removes edge noise artifacts.

![Baikal Swin Anime](https://cdn-uploads.huggingface.co/production/uploads/658814fd586088fd274d8cc1/R3AU887ILEP5o6a_4NIPj.png)

## Examples

![Example_1](https://cdn-uploads.huggingface.co/production/uploads/658814fd586088fd274d8cc1/xga0aU-5vGWIjbBhgNb4q.png)
![Example_2](https://cdn-uploads.huggingface.co/production/uploads/658814fd586088fd274d8cc1/3kHf5838HNKXUDGwyA2CJ.png)
![Example_3](https://cdn-uploads.huggingface.co/production/uploads/658814fd586088fd274d8cc1/nU8Oh0mLAa-HH6viy26oJ.png)

# How to use in ComfyUI

The model is designed to work with the **SnJake Anime Upscale** ComfyUI node.

1. Install the node from [GITHUB REPO](https://github.com/SnJake/SnJake_Baikal_Swin_Anime).
2. Download the weights from this repository.
3. Place the file(s) into `ComfyUI/models/anime_upscale/`.
4. Select the weights in the node dropdown and run the workflow.

# Training Details
V1:
- Dataset: 40,000 images from Danbooru2024: https://huggingface.co/datasets/deepghs/danbooru2024
- Validation: 600 images
- Epochs: 70

V2: 
- Slightly sharper output, no edge noise artifacts.
- Epochs: 20
- Dataset: 49,606 images from Danbooru2024: https://huggingface.co/datasets/deepghs/danbooru2024
- Perceptual backbone: convnextv2_tiny.fcmae_ft_in22k_in1k, fine‑tuned on anime to improve feature sensitivity.
- Loss schedule: gradual ramp‑in of perceptual/auxiliary losses for stable training.

V2.1:
- Removed Nearest from resample_methods
- Epochs: 30

V2.2:
- Epochs: 40 (For now)

Training code is included in `training_code/` for reference.

## Disclaimer

This project was made purely for curiosity and personal interest. The code was written by GPT-5.2 Codex.