pretrained-yolo-v4 icon indicating copy to clipboard operation
pretrained-yolo-v4 copied to clipboard

Published 20 hours ago • verified publisher icon matlab-deep-learning

yolov4Transferlearn.m error

Open chn7960 opened this issue 4 years ago • 2 comments

About yolov4Transferlearn.m.

The following error occurred.

What is the problem?

image

chn7960 avatar Aug 24 '21 09:08 chn7960

@chn7960 Sorry for the delayed response. There are two options to avoid the out of memory issues during training.

  1. You can reduce the 'miniBatchsize' (at line no. 181 of yolov4TransferLearn.m file) and train the model. The current 'miniBatchSize' is 4, you can reduce it to a smaller size depending upon your GPU memory.

  2. You can change the 'networkInputSize' (at line no. 108 of yolov4TransferLearn.m file) to a smaller input size, say [416,416,3] and train the model. You will have to change the 'inputSize' in detectYOLOv4.m (at line no. 21 of detectYOLOv4.m file) to the input size that you specified during training, say [416,416,3] to correctly perform the inference.

Please try this and let me know if you face any additional issues.

sbairagy-MW avatar Sep 17 '21 05:09 sbairagy-MW

thank you all for the help, but when i train my own data, the following error appear, how do i solve the problem, thank you image

trandong20 avatar Dec 31 '21 07:12 trandong20