Commit f83e9f8 1 parent e04da2a commit f83e9f8 Copy full SHA for f83e9f8
File tree 1 file changed +4
-3
lines changed
1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -358,7 +358,8 @@ def predict(
358
358
Can be specified as 'jpg' or 'png'
359
359
verbose: int
360
360
0: no print
361
- 1: print slice/prediction durations, number of slices, model loading/file exporting durations
361
+ 1: print slice/prediction durations, number of slices
362
+ 2: print model loading/file exporting durations
362
363
"""
363
364
# assert prediction type
364
365
assert (
@@ -441,7 +442,7 @@ def predict(
441
442
postprocess_match_metric = postprocess_match_metric ,
442
443
postprocess_match_threshold = postprocess_match_threshold ,
443
444
postprocess_class_agnostic = postprocess_class_agnostic ,
444
- verbose = verbose ,
445
+ verbose = 1 if verbose else 0 ,
445
446
)
446
447
object_prediction_list = prediction_result .object_prediction_list
447
448
durations_in_seconds ["slice" ] += prediction_result .durations_in_seconds ["slice" ]
@@ -544,7 +545,7 @@ def predict(
544
545
save_json (coco_json , save_path )
545
546
546
547
# print prediction duration
547
- if verbose == 1 :
548
+ if verbose == 2 :
548
549
print (
549
550
"Model loaded in" ,
550
551
durations_in_seconds ["model_load" ],
You can’t perform that action at this time.
0 commit comments