Update ssd.py

ssd.py

@@ -11,6 +11,10 @@ from PIL import Image,ImageFont, ImageDraw
 from torch.autograd import Variable
 
 MEANS = (104, 117, 123)
+#--------------------------------------------#
+#   使用自己训练好的模型预测需要修改2个参数
+#   model_path和classes_path都需要修改！
+#--------------------------------------------#
 class SSD(object):
     _defaults = {
         "model_path": 'model_data/ssd_weights.pth',