Update README.md

xiao-hua-sheng · Jul 7, 2022 · 074d26a · 074d26a
1 parent 40a29d7
commit 074d26a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/README.md b/README.md
@@ -8,5 +8,6 @@
 2、教师模型的加载在trainer.py 131行左右实现
 
 3、distill损失的实现见yolo_head.py 255行distill_loss，实现很简单，具体参数根据自己训练情况进行调整。
+  c_obj_scale = t_obj_scale.unsqueeze(-1).repeat(1,1,1,1,num_class)  num_class:检测类别，自行修改。 
 
 4、项目中增加了focal_loss和eiou，如果需要和原版进行对比请自行修改相关文件。