Ashrafb commited on
Commit
318d7ba
·
verified ·
1 Parent(s): e65ec95

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +6 -1
main.py CHANGED
@@ -17,7 +17,12 @@ app = FastAPI()
17
 
18
  # Initialize the VToonify model and MTCNN detector
19
  model = None
20
- detector = MTCNN()
 
 
 
 
 
21
 
22
  def load_model():
23
  global model
 
17
 
18
  # Initialize the VToonify model and MTCNN detector
19
  model = None
20
+ # Example of adjusting MTCNN parameters
21
+ detector = MTCNN(min_face_size=20, scale_factor=0.709)
22
+
23
+ # Log detection results
24
+ results = detector.detect_faces(frame_rgb)
25
+ logging.info(f"Detection results: {results}")()
26
 
27
  def load_model():
28
  global model