diff --git a/torchvision/models/detection/ssdlite.py b/torchvision/models/detection/ssdlite.py index d35dfbf78d3..47ecf59f1e2 100644 --- a/torchvision/models/detection/ssdlite.py +++ b/torchvision/models/detection/ssdlite.py @@ -252,7 +252,7 @@ def ssdlite320_mobilenet_v3_large( "detections_per_img": 300, "topk_candidates": 300, # Rescale the input in a way compatible to the backbone: - # The following mean/std rescale the data from [0, 1] to [-1, -1] + # The following mean/std rescale the data from [0, 1] to [-1, 1] "image_mean": [0.5, 0.5, 0.5], "image_std": [0.5, 0.5, 0.5], }