MathFrenchToast commited on
Commit
95eca90
·
1 Parent(s): 95551e2

switch to large model

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -13,7 +13,7 @@ import cv2
13
  import pandas as pd
14
  import numpy as np
15
  import torch
16
- from rfdetr import RFDETRBase
17
  from PIL import Image
18
  from typing import Union, Optional, List, Dict, Tuple
19
  import os
@@ -84,7 +84,7 @@ class RFDETRSoccerNet:
84
  print(f"📦 Loading model from {self.model_path}...")
85
 
86
  # Initialize base model
87
- self.model = RFDETRBase()
88
 
89
  # Reinitialize detection head for 4 classes (critical for compatibility)
90
  print(f"🔧 Reinitializing detection head for {self.num_classes} classes...")
 
13
  import pandas as pd
14
  import numpy as np
15
  import torch
16
+ from rfdetr import RFDETRLarge
17
  from PIL import Image
18
  from typing import Union, Optional, List, Dict, Tuple
19
  import os
 
84
  print(f"📦 Loading model from {self.model_path}...")
85
 
86
  # Initialize base model
87
+ self.model = RFDETRLarge()
88
 
89
  # Reinitialize detection head for 4 classes (critical for compatibility)
90
  print(f"🔧 Reinitializing detection head for {self.num_classes} classes...")