balancap / SSD-Tensorflow

Single Shot MultiBox Detector in TensorFlow
4.11k stars 1.89k forks source link

what should i do to train my own data? #217

Open Zzmc opened 6 years ago

Zzmc commented 6 years ago

how to adjust those code to fit my data(book, cup, keyboard, computer, apple, mouse)

TRAIN_STATISTICS = { 'none': (0, 0), 'aeroplane': (238, 306), 'bicycle': (243, 353), 'bird': (330, 486), 'boat': (181, 290), 'bottle': (244, 505), 'bus': (186, 229), 'car': (713, 1250), 'cat': (337, 376), 'chair': (445, 798), 'cow': (141, 259), 'diningtable': (200, 215), 'dog': (421, 510), 'horse': (287, 362), 'motorbike': (245, 339), 'person': (2008, 4690), 'pottedplant': (245, 514), 'sheep': (96, 257), 'sofa': (229, 248), 'train': (261, 297), 'tvmonitor': (256, 324), 'total': (5011, 12608), } TEST_STATISTICS = { 'none': (0, 0), 'aeroplane': (1, 1), 'bicycle': (1, 1), 'bird': (1, 1), 'boat': (1, 1), 'bottle': (1, 1), 'bus': (1, 1), 'car': (1, 1), 'cat': (1, 1), 'chair': (1, 1), 'cow': (1, 1), 'diningtable': (1, 1), 'dog': (1, 1), 'horse': (1, 1), 'motorbike': (1, 1), 'person': (1, 1), 'pottedplant': (1, 1), 'sheep': (1, 1), 'sofa': (1, 1), 'train': (1, 1), 'tvmonitor': (1, 1), 'total': (20, 20), } SPLITS_TO_SIZES = { 'train': 5011, 'test': 4952, } SPLITS_TO_STATISTICS = { 'train': TRAIN_STATISTICS, 'test': TEST_STATISTICS, }

Salight commented 6 years ago

@Zzmc I have the same problem,have you solved it?