Peleck commited on
Commit
44e2803
·
1 Parent(s): fa8453f
.gitignore CHANGED
@@ -1,4 +0,0 @@
1
-
2
- *.pth
3
- *.onnx
4
- *.pyc
 
 
 
 
 
assets/pretrained_models/GFPGANv1.4.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6548e54cbcf248af385248f0c1193b359c37a0f98b836282b09cf48af4fd2b73
3
+ size 340256690
assets/pretrained_models/GPEN-BFR-512.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0960f836488735444d508b588e44fb5dfd19c68fde9163ad7878aa24d1d5115e
3
+ size 284250449
assets/pretrained_models/codeformer.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a1a03d62686b92fa217408cad64c6ca37aaf2788ac340693a926e6361c58e36c
3
+ size 376798532
assets/pretrained_models/det_10g.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5838f7fe053675b1c7a08b633df49e7af5495cee0493c7dcf6697200b85b5b91
3
+ size 16923827
assets/pretrained_models/faceparser.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:345f7ced4e64f7fe68a913ad717e23d971b37cb68b41f6b9a0b7124bac0740ee
3
+ size 53201152
assets/pretrained_models/inswapper_128.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e4a3f08c753cb72d04e10aa0f7dbe3deebbf39567d4ead6dce08e98aa49e16af
3
+ size 554253681
assets/pretrained_models/open-nsfw.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:864bb37bf8863564b87eb330ab8c785a79a773f4e7c43cb96db52ed8611305fa
3
+ size 23590724
assets/pretrained_models/restoreformer.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca44c1c63642078c6f4daf5645421e6daa4a640440597af836e2d332982cf3d4
3
+ size 292022041
assets/pretrained_models/w600k_r50.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c06341c33c2ca1f86781dab0e829f88ad5b64be9fba56e56bc9ebdefc619e43
3
+ size 174383860
requirements.txt CHANGED
@@ -4,3 +4,6 @@ opencv-python>=4.7.0.72
4
  opencv-python-headless>=4.7.0.72
5
  onnx==1.14.0
6
  onnxruntime==1.15.0
 
 
 
 
4
  opencv-python-headless>=4.7.0.72
5
  onnx==1.14.0
6
  onnxruntime==1.15.0
7
+ torch==2.0.0+cpu
8
+ torchvision==0.15.1+cpu
9
+ torchaudio==2.0.1