search

WongKinYiu / PyTorch_YOLOv4

PyTorch implementation of YOLOv4
1.87k stars 585 forks source link

[Question] About save darknet format weights. #204

Open CaptainEven opened 3 years ago

CaptainEven commented 3 years ago

In function save_Weights, models.py in master's branch, what's the reason for setting cutoff=-1 by default?

WongKinYiu commented 3 years ago

-1 means all.

CaptainEven commented 3 years ago

@WongKinYiu I don't think so, because end_index -1(the last item) is excluded in python slice, as a result [:-1] means the last layer is excluded, am i right?

WongKinYiu commented 3 years ago

the last layer is yolo layer, and yolo layer do not has weights.