Update vtoonify_model.py
Browse files- vtoonify_model.py +21 -1
vtoonify_model.py
CHANGED
|
@@ -36,7 +36,27 @@ class Model():
|
|
| 36 |
self.device = device
|
| 37 |
self.style_types = {
|
| 38 |
'cartoon1': ['vtoonify_d_cartoon/vtoonify_s026_d0.5.pt', 26],
|
| 39 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
}
|
| 41 |
|
| 42 |
self.face_detector = self._create_insightface_detector()
|
|
|
|
| 36 |
self.device = device
|
| 37 |
self.style_types = {
|
| 38 |
'cartoon1': ['vtoonify_d_cartoon/vtoonify_s026_d0.5.pt', 26],
|
| 39 |
+
'cartoon1-d': ['vtoonify_d_cartoon/vtoonify_s_d.pt', 26],
|
| 40 |
+
'cartoon2-d': ['vtoonify_d_cartoon/vtoonify_s_d.pt', 64],
|
| 41 |
+
'cartoon3-d': ['vtoonify_d_cartoon/vtoonify_s_d.pt', 153],
|
| 42 |
+
'cartoon4': ['vtoonify_d_cartoon/vtoonify_s299_d0.5.pt', 299],
|
| 43 |
+
'cartoon4-d': ['vtoonify_d_cartoon/vtoonify_s_d.pt', 299],
|
| 44 |
+
'cartoon5-d': ['vtoonify_d_cartoon/vtoonify_s_d.pt', 8],
|
| 45 |
+
'comic1-d': ['vtoonify_d_comic/vtoonify_s_d.pt', 28],
|
| 46 |
+
'comic2-d': ['vtoonify_d_comic/vtoonify_s_d.pt', 18],
|
| 47 |
+
'arcane1': ['vtoonify_d_arcane/vtoonify_s000_d0.5.pt', 0],
|
| 48 |
+
'arcane1-d': ['vtoonify_d_arcane/vtoonify_s_d.pt', 0],
|
| 49 |
+
'arcane2': ['vtoonify_d_arcane/vtoonify_s077_d0.5.pt', 77],
|
| 50 |
+
'arcane2-d': ['vtoonify_d_arcane/vtoonify_s_d.pt', 77],
|
| 51 |
+
'caricature1': ['vtoonify_d_caricature/vtoonify_s039_d0.5.pt', 39],
|
| 52 |
+
'caricature2': ['vtoonify_d_caricature/vtoonify_s068_d0.5.pt', 68],
|
| 53 |
+
'pixar': ['vtoonify_d_pixar/vtoonify_s052_d0.5.pt', 52],
|
| 54 |
+
'pixar-d': ['vtoonify_d_pixar/vtoonify_s_d.pt', 52],
|
| 55 |
+
'illustration1-d': ['vtoonify_d_illustration/vtoonify_s054_d_c.pt', 54],
|
| 56 |
+
'illustration2-d': ['vtoonify_d_illustration/vtoonify_s004_d_c.pt', 4],
|
| 57 |
+
'illustration3-d': ['vtoonify_d_illustration/vtoonify_s009_d_c.pt', 9],
|
| 58 |
+
'illustration4-d': ['vtoonify_d_illustration/vtoonify_s043_d_c.pt', 43],
|
| 59 |
+
'illustration5-d': ['vtoonify_d_illustration/vtoonify_s086_d_c.pt', 86],
|
| 60 |
}
|
| 61 |
|
| 62 |
self.face_detector = self._create_insightface_detector()
|