luxdelux7 commited on
Commit
8f28078
·
verified ·
1 Parent(s): 8d76280

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -4
README.md CHANGED
@@ -97,11 +97,14 @@ More reliable detection across mixed realistic/anime domains with acceptable spe
97
  **Purpose:** Precise face mask generation
98
 
99
  **Training Approach:**
100
- - Dataset prepared using the Forbidden Vision yolo model at 512px resolution
101
  - Iterative hardmine training in multiple phases:
102
- -- started with 700 samples, trained
103
- -- run trained model on untrained set of images -> pick failures -> correct -> include in total set and retrain
104
- -- repeat process until no obvious failures exist -> final set 4k+ images
 
 
 
105
 
106
  **Features:**
107
  - Detects and includes facial features other models ignore, like protruding anime eybrows, realistic eyelashes sticking out of the face etc.
 
97
  **Purpose:** Precise face mask generation
98
 
99
  **Training Approach:**
100
+ - Dataset prepared using the Forbidden Vision YOLO model at 512px resolution
101
  - Iterative hardmine training in multiple phases:
102
+ - Train on the initial 700 samples
103
+ - Evaluate on remaining images to find failure cases
104
+ - Correct failed masks and add them to the dataset
105
+ - Retrain with the expanded dataset
106
+ - Repeat until failure cases drop to near-zero
107
+ (final dataset: 4k+ images)
108
 
109
  **Features:**
110
  - Detects and includes facial features other models ignore, like protruding anime eybrows, realistic eyelashes sticking out of the face etc.