Nyako/nsfw-vid-detect
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

0.0.2a

Browse Source 
Add example
This commit is contained in:
Hell13Cat
2022-06-30 22:31:06 +03:00
parent ae846f2e3e
commit c9a7b13c5a
9 changed files with 75 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.gitignore
+4 -1
View File
@@ -4,7 +4,10 @@ data/vids/*
data/screen/* data/screen/*
data/json/* data/json/*
data/*.* data/*.*
!data/vids/example.mp4
!data/example.mp4.html
!data/json/example.mp4.json
!data/screen/example.mp4
# Byte-compiled / optimized / DLL files # Byte-compiled / optimized / DLL files
__pycache__/ __pycache__/
*.py[cod] *.py[cod]
data/example.mp4.html
+31
View File
@@ -0,0 +1,31 @@
<html>
<head>
<title>example.mp4</title>
</head>
<h2>example.mp4</h2><br>
<h3>1080x1920</h3><br>
<h3>16.2 s.</h3><br>
<video height="50%" src="vids/example.mp4" controls></video><br>
<hr>
<div>
<p>screen/example.mp4/1.jpg</p>
<img src="screen/example.mp4/1.jpg" alt="screen/example.mp4/1.jpg" height="50%"><br>
<p>drawings : 0.16<br>hentai : 0.03<br>neutral : 99.03<br>porn : 0.63<br>sexy : 0.15<br></p>
</div>
<hr><div>
<p>screen/example.mp4/6.jpg</p>
<img src="screen/example.mp4/6.jpg" alt="screen/example.mp4/6.jpg" height="50%"><br>
<p>drawings : 0.28<br>hentai : 0.03<br>neutral : 99.3<br>porn : 0.3<br>sexy : 0.09<br></p>
</div>
<hr><div>
<p>screen/example.mp4/11.jpg</p>
<img src="screen/example.mp4/11.jpg" alt="screen/example.mp4/11.jpg" height="50%"><br>
<p>drawings : 1.16<br>hentai : 0.03<br>neutral : 98.55<br>porn : 0.15<br>sexy : 0.1<br></p>
</div>
<hr><div>
<p>screen/example.mp4/16.jpg</p>
<img src="screen/example.mp4/16.jpg" alt="screen/example.mp4/16.jpg" height="50%"><br>
<p>drawings : 0.14<br>hentai : 0.02<br>neutral : 98.66<br>porn : 0.69<br>sexy : 0.49<br></p>
</div>
<hr>
</html>
data/json/example.mp4.json
+39
View File
@@ -0,0 +1,39 @@
{
"height": 1080.0,
"width": 1920.0,
"duration": 16.2,
"frames": {
"1": {
"drawings": 0.001603447599336505,
"hentai": 0.00029177090618759394,
"neutral": 0.9903027415275574,
"porn": 0.006316039711236954,
"sexy": 0.0014860795345157385,
"file": "screen/example.mp4/1.jpg"
},
"6": {
"drawings": 0.0028055706061422825,
"hentai": 0.0002788607671391219,
"neutral": 0.9929991960525513,
"porn": 0.0030449822079390287,
"sexy": 0.0008713431307114661,
"file": "screen/example.mp4/6.jpg"
},
"11": {
"drawings": 0.011603194288909435,
"hentai": 0.00034240508102811873,
"neutral": 0.9855231046676636,
"porn": 0.0014979398110881448,
"sexy": 0.001033419743180275,
"file": "screen/example.mp4/11.jpg"
},
"16": {
"drawings": 0.0014052210608497262,
"hentai": 0.00019437579612713307,
"neutral": 0.9866281151771545,
"porn": 0.006891936529427767,
"sexy": 0.004880368243902922,
"file": "screen/example.mp4/16.jpg"
}
}
}
data/screen/example.mp4/1.jpg
BIN
View File
Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 361 KiB

data/screen/example.mp4/11.jpg
BIN
View File
Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 356 KiB

data/screen/example.mp4/16.jpg
BIN
View File
Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 334 KiB

data/screen/example.mp4/6.jpg
BIN
View File
Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 370 KiB

data/vids/example.mp4
BIN
View File
Binary file not shown.
main.py
+1 -1
View File
@@ -14,7 +14,7 @@ def save_frame(screen_folder, secs, vid):
return save_name return save_name
model = predict.load_model('./data/model/nsfw_mobilenet2.224x224.h5') model = predict.load_model('./data/model/nsfw_mobilenet2.224x224.h5')
file_path = "./data/vids/5.mp4" file_path = "./data/vids/example.mp4"
vid = cv2.VideoCapture( file_path ) vid = cv2.VideoCapture( file_path )
height = vid.get(cv2.CAP_PROP_FRAME_HEIGHT) height = vid.get(cv2.CAP_PROP_FRAME_HEIGHT)
width = vid.get(cv2.CAP_PROP_FRAME_WIDTH) width = vid.get(cv2.CAP_PROP_FRAME_WIDTH)