PyTorch_YOLOv4 icon indicating copy to clipboard operation
PyTorch_YOLOv4 copied to clipboard

您好,这个支持darknet版的yolov4-p5和p6的cfg吗?

Open SpongeBab opened this issue 2 years ago • 4 comments

https://github.com/AlexeyAB/darknet/blob/master/cfg/yolov4-p6.cfg

SpongeBab avatar Oct 19 '21 02:10 SpongeBab

yes, just change the activation before yolo layers from logistic to linear and remove some unsupported fields in yolo layers.

WongKinYiu avatar Nov 02 '21 02:11 WongKinYiu

Hi @WongKinYiu , are all the provided weights in this repo are with the logistical wh in yolo layers? so the yolov4.weights is different from AlexeyAB's one? Or only those pacsp models are with the logistical wh?

atangfan avatar Dec 27 '21 13:12 atangfan

@atangfan The .weights-files are identical in https://github.com/AlexeyAB/darknet and https://github.com/WongKinYiu/PyTorch_YOLOv4 but different cfg-files:

  • for Darknet you should use activation=logistic
  • for PyTorch_YOLOv4 you should use activation=linear

AlexeyAB avatar Dec 27 '21 14:12 AlexeyAB

Where can we get the pt files? @AlexeyAB @WongKinYiu Since train.py requires only those.

parthkvv avatar Sep 30 '22 15:09 parthkvv