custom.weights and custom-416

[ghost]

Hello, please help So python save_model.py --weights ./data/custom.weights --output ./checkpoints/custom-416 --input_size 416 --model yolov4

I get error File "C:\Users\va\Documents\Arduino\yolov4-custom-functions-master\core\utils.py", line 143, in load_weights conv_weights = conv_weights.reshape(conv_shape).transpose([2, 3, 1, 0]) ValueError: cannot reshape array of size 4554552 into shape (1024,512,3,3) Where custom.names file?

Please help

[ghost]

Ooook :) custom.names contain one row "car" its work fine

[shubham-shahh]

close the issue if it is solved @iabesadze

[GiorgosBetsos]

I'm having same issue

python save_model.py --weights ./data/custom.weights --output ./checkpoints/custom-416 --input_size 416 --model yolov4

results in

  File "\third_party_repos\yolov4-custom-functions-master\core\utils.py", line 143, in load_weights
    conv_weights = conv_weights.reshape(conv_shape).transpose([2, 3, 1, 0])
ValueError: cannot reshape array of size 4554552 into shape (1024,512,3,3)

[GiorgosBetsos]

Ooook :) custom.names contain one row "car" its work fine

@iabesadze Can you please explain how this solves the issue?

[navidnadery]

@GiorgosBetsos Create and fill a text file named custom.names with string car Then, write it's path on line 14 of 'core/config.py' file.

[kctiintern21]

I am doing for the License plate detector same error tried the above solutions but did not work for the script: python save_model.py --weights ./data/custom.weights --output ./checkpoints/custom-416 --input_size 416 --model yolov4 Output:- File "/content/yolov4-custom-functions/core/utils.py", line 143, in load_weights conv_weights = conv_weights.reshape(conv_shape).transpose([2, 3, 1, 0]) ValueError: cannot reshape array of size 84123 into shape (512,1024,1,1)