experiencor / keras-yolo2

Easy training on custom dataset. Various backends (MobileNet and SqueezeNet) supported. A YOLO demo to detect raccoon run entirely in brower is accessible at https://git.io/vF7vI (not on Windows).

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to train my Custom Dataset with this notebook

316usman opened this issue · comments

Hi, I have been trying for a long time to train my own data set with this notebook but I have faced many issues and solved some of them
Firsy My Data Set is as follows
images folder ----> with training images of size 640x512
labels folder ---> text files for each image above with following data 0 311.489379882812 204.399459838867 36.0547180175781 24.1059265136719 i-e class label, x , y, w, h, (these values are not normalized.

I have modified preprocessinmg.py to output the dataset as required by the notebook but still I get the error

c:\Users\Usman\anaconda3\envs\tf-gpu\lib\site-packages\numpy\core\fromnumeric.py:43 _wrapit
    result = getattr(asarray(obj), method)(*args, **kwds)

ValueError: cannot reshape array of size 10 into shape (1,1,1,1,2)

and i cannt seen to find where my error is

Ihave boon stuck at this for like 4 doays now please help