Merge pull request #62 from alganzory/feature/nsfw-only-mode

allow nsfw only mode (no face detection)

src/modules/detector.js

@@ -319,6 +319,11 @@ const containsGenderFace = (detections, detectMale, detectFemale) => {
 	}
 
 	const faces = detections.face;
+	if (detectMale && detectFemale)
+		return faces.some(
+			(face) =>
+				face.age > 20 
+		);
 	if (detectMale || detectFemale)
 		return faces.some(
 			(face) =>
@@ -330,7 +335,7 @@ const containsGenderFace = (detections, detectMale, detectFemale) => {
 					detectFemale
 				)
 		);
-	else return true; // If no gender specified, return true cause there's a face
+	else return false;
 };
 // export the human variable and the HUMAN_CONFIG object
 export {

src/modules/settings.js

@@ -56,12 +56,7 @@ class Settings {
 
 	shouldDetect() {
 		if (!this._settings.status) return false;
-		if (!this.shouldDetectImages() && !this.shouldDetectVideos())
-			return false;
-
-		// at least 1 gender should be selected for detection to start,
-		// this could change in the future (if we wanna allow nsfw only detection)
-		return this.shouldDetectGender();
+		return (this.shouldDetectImages() || this.shouldDetectVideos());
 	}
 
 	isBlurryStartMode() {

src/offscreen.js

@@ -93,6 +93,7 @@ const runDetection = async (img, isVideo = false) => {
 		human.tf.dispose(tensor);
 		return "nsfw";
 	}
+	if (!settings.shouldDetectGender()) return false; // no need to run gender detection if it's not enabled
 	const predictions = await humanModelClassify(tensor);
 	// console.log("offscreen human result", predictions);
 	human.tf.dispose(tensor);