From ff829f275ac6ae837d8c80036c6fb6bbe077e729 Mon Sep 17 00:00:00 2001
From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com>
Date: Thu, 11 Jun 2020 11:47:31 +0800
Subject: [PATCH] Update ssd.py

---
 ssd.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/ssd.py b/ssd.py
index 5b51ada..34c83b6 100644
--- a/ssd.py
+++ b/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',
-- 
GitLab