search

rykov8 / ssd_keras

Port of Single Shot MultiBox Detector to Keras
MIT License
1.1k stars 553 forks source link

question on training with other dataset #27

Open Walid-Ahmed opened 7 years ago

Walid-Ahmed commented 7 years ago

@lvaleriu

Hi your code worked fine when I applied it to images, video and cam. I am trying to classify vehicles and pedestrians, I checked the file gt_pascal.pkl and read one of its values by

import pickle
f = open("gt_pascal.pkl")
data = pickle.load(f)    #<type 'dict'>
print(data.get('frame05183.png'))

I have several question on training and I hope you can help me

1- Where is frame05183.png stored? 2-To do my training with 3 classes, I believe I will have to edit "gt_pascal.pkl", but where should I store my images? 3-What objects are you actually trying to train to detect in SSD_training.ipynb?

Thanks a lot

wuyanlun commented 7 years ago

hi, @Walid-Ahmed have you solved this problem? I don't understand the "gt_pascal.pkl" either. Whether the train picture in SSD_training.ipynb is not avariable? Thanks a lot!

lantuzi commented 7 years ago

gt_pascal.pkl is the training data, each row is one object. Its structure is same as and created by PASCAL_VOC/get_data_from_XML.py as I found. Thanks to the great work of rykov8