Use another Activation Function in YOLOv7

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

GNU General Public License v3.0

13.24k stars 4.18k forks source link

Use another Activation Function in YOLOv7 #1236

Open eposav4 opened 1 year ago

eposav4 commented 1 year ago

Hello and Good day. How do you change the activation function in Yolov7? For example, the default activation function of yolov7 (except tiny) is SiLU and I want to change to ReLU

I found a solution but it's for yolov5.

yulin010101 commented 1 year ago

You can change it manually in models/common.py.

eposav4 commented 1 year ago

@yulin010101 thanks for the info .. last question how the best.pt selected like on what epoch did the best.pt saved ?

In yolov5 according to this comment link it said that the highest map@.5:.95

TusharChauhaan commented 1 year ago

@eposav4 I think it is based upon the fitness score which is the weighted sum of map@50 and map@50:95 which you can find in the utils/metrics.py script.