How to train YOLO V3 keras with your own data set

preview_player
Показать описание
This is tutorial explains how to train yolov3 keras with your own data set.

keras-yolo3

labelImg

weights

this project

this project was tested with
WIN10 Pro
python 3.6.8
tensorflow-gpu 1.14.0
Keras 2.1.5
h5py 2.9.0
Рекомендации по теме
Комментарии
Автор

Андрей, здравствуйте! Спасибо большое за код!!

nhfnmqg
Автор

Hi, I am using the customized data of a cube with different colors, which identifies the shape but not the color correctly. Do I need to make any changes in the cfg file?

AjayShetty-oszl
Автор

ValueError: Cannot assign value to variable ' conv2d_58/kernel:0': Shape mismatch.The variable shape (1, 1, 1024, 255), and the assigned value shape (18, 1024, 1, 1) are incompatible.

sarpan
Автор

Thanks works fine! can you please tell me how to get the coordinates of the center of bounding box x_center and y_center ?

beec
Автор

AttributeError: module 'keras.utils.generic_utils' has no attribute
tensorflow 2.5.0
keras 2.4.0
python 3.9

Furkan-vszt