search

WongKinYiu / ScaledYOLOv4

Scaled-YOLOv4: Scaling Cross Stage Partial Network
GNU General Public License v3.0
2.02k stars 575 forks source link

Pre-training weight #361

Open Fateeeeee opened 2 years ago

Fateeeeee commented 2 years ago

Hello, where are the pre-training weights of YOLOv4-CSP ending in pt? I only saw the ones ending in .weights.

WongKinYiu commented 2 years ago

.weights ant .pt are convertable, just load .weights and then save it to .pt.

Fateeeeee commented 2 years ago

.weights ant .pt are convertable, just load .weights and then save it to .pt.

Um, when I converted it, the following error occurred during load. " _pickle.UnpicklingError: invalid load key, '\x00'. "

WongKinYiu commented 2 years ago

it seems you are using code in different branch. do you mean you need weights of yolov4-csp for .yaml style config file?

Fateeeeee commented 2 years ago

Right! This is what I need.

ismaelfer commented 2 years ago

Is it recomended to train using pre-traned weights in a custom project? If so, what weights are recomended? COCO?