Skip to content

Commit

Permalink
fix: Change import in detect.py
Browse files Browse the repository at this point in the history
  • Loading branch information
yakhyo committed Nov 2, 2024
1 parent 8c990e9 commit fed71d3
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions detect.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@

from config import cfg
from models.faceboxes import FaceBoxes
from utils.prior_box import PriorBox
from layers import PriorBox
from utils.box_utils import decode, nms


Expand Down Expand Up @@ -109,7 +109,7 @@ def parse_args():

parser.add_argument(
'--weights',
default='./weights/faceboxes.pth',
default='./weights/last.pth',
type=str,
help='Path to the trained model state dict file.'
)
Expand All @@ -122,13 +122,13 @@ def parse_args():
parser.add_argument('--image-path', type=str, help='Path to the image for inference.')
parser.add_argument(
'--conf-threshold',
default=0.05,
default=0.02,
type=float,
help='Minimum confidence threshold for considering detections.'
)
parser.add_argument(
'--pre-nms-top-k',
default=300,
default=5000,
type=int,
help='Number of top bounding boxes to consider for NMS.'
)
Expand All @@ -139,7 +139,7 @@ def parse_args():
type=int,
help='Number of top bounding boxes to keep after NMS.'
)
parser.add_argument('--vis-threshold', default=0.15, type=float, help='Visualization threshold for bounding boxes')
parser.add_argument('--vis-threshold', default=0.6, type=float, help='Visualization threshold for bounding boxes')

args = parser.parse_args()
return args
Expand Down

0 comments on commit fed71d3

Please sign in to comment.