提交 53c50566 编写于 作者: Bubbliiiing's avatar Bubbliiiing

update yolo.py

上级 e96fc51f
...@@ -113,7 +113,7 @@ _defaults = { ...@@ -113,7 +113,7 @@ _defaults = {
# 验证集损失较低不代表mAP较高,仅代表该权值在验证集上泛化性能较好。 # 验证集损失较低不代表mAP较高,仅代表该权值在验证集上泛化性能较好。
# 如果出现shape不匹配,同时要注意训练时的model_path和classes_path参数的修改 # 如果出现shape不匹配,同时要注意训练时的model_path和classes_path参数的修改
#--------------------------------------------------------------------------# #--------------------------------------------------------------------------#
"model_path" : 'model_data/yolov5_s_v6.1.pth', "model_path" : 'model_data/yolov7_weights.pth',
"classes_path" : 'model_data/coco_classes.txt', "classes_path" : 'model_data/coco_classes.txt',
#---------------------------------------------------------------------# #---------------------------------------------------------------------#
# anchors_path代表先验框对应的txt文件,一般不修改。 # anchors_path代表先验框对应的txt文件,一般不修改。
...@@ -126,9 +126,11 @@ _defaults = { ...@@ -126,9 +126,11 @@ _defaults = {
#---------------------------------------------------------------------# #---------------------------------------------------------------------#
"input_shape" : [640, 640], "input_shape" : [640, 640],
#------------------------------------------------------# #------------------------------------------------------#
# 所使用的YoloV5的版本。s、m、l、x # 所使用到的yolov7的版本,本仓库一共提供两个:
# l : 对应yolov7
# x : 对应yolov7_x
#------------------------------------------------------# #------------------------------------------------------#
"phi" : 's', "phi" : 'l',
#---------------------------------------------------------------------# #---------------------------------------------------------------------#
# 只有得分大于置信度的预测框会被保留下来 # 只有得分大于置信度的预测框会被保留下来
#---------------------------------------------------------------------# #---------------------------------------------------------------------#
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册