The YOLOv1 model produces the same results for each image
The model (Yolov1) is written in PyTorch. 30 epochs were trained, AdamOptimazer was used for training, learning rate = 1e-6, batch_size = 64. DeepScoresV2 dense dataset: 1362 photos in the training sample, I take only 3 classes from the dataset. But model returns the same result for all photos. What could possibly be the problem?
The task is to identify the musical notation on the sheet of music, as well as to search for special signs (accolade and keys).
Example (‘Brace’ appears in empty cells, this is an error of the function that outputs the image, everything is fine in the dataset itself):
example
The YOLOv1 model produces the same results for each image
The model (Yolov1) is written in PyTorch. 30 epochs were trained, AdamOptimazer was used for training, learning rate = 1e-6, batch_size = 64. DeepScoresV2 dense dataset: 1362 photos in the training sample, I take only 3 classes from the dataset. But model returns the same result for all photos. What could possibly be the problem?
The task is to identify the musical notation on the sheet of music, as well as to search for special signs (accolade and keys).
Example (‘Brace’ appears in empty cells, this is an error of the function that outputs the image, everything is fine in the dataset itself):
example