squeezeDet icon indicating copy to clipboard operation
squeezeDet copied to clipboard

How to train SqueezeDet by own dataset with one class?

Open jwwangchn opened this issue 6 years ago • 6 comments

I want to train SqueezeDet use our dataset, this dataset only has one class, but when I use this dataset to train SqueezeDet, it met some error, so I want to know how to train SqueezeDet by own dataset with one class?

jwwangchn avatar Aug 17 '18 03:08 jwwangchn

I am also facing same issue. Can anyone please let me know how to train SqueezeDet for own dataset with one class?

Shashikant255 avatar Aug 24 '18 06:08 Shashikant255

@Shashikant255 I use two same classes to train SqueezeDet, it works.

jwwangchn avatar Aug 27 '18 02:08 jwwangchn

@jwwangchn Thanks for your help. I have one more doubt regarding label (bounding box information about object i.e x, y, w, h) file. If any image doesn't have any class or object then its corresponding label file should be empty?

Shashikant255 avatar Aug 27 '18 06:08 Shashikant255

@Shashikant255 As far as I know, you should delete these label files. File reader does not care about images without labels files.

jwwangchn avatar Aug 28 '18 02:08 jwwangchn

You need to change the configuration file, period. Adapt the configuration files such that only one class is specified and make sure no label file is empty and depends on your size of images image resolution (HXW) you have will have to change parameters of anchor boxes.

acmaheri avatar Apr 24 '19 09:04 acmaheri

"CLASSES": 1, "CLASS_NAMES": ["my_object"] is this correct way top modify squeeze.config for single object detection? or background needs to be added in CLASS_NAMES and CLASSES = 2?

I am getting no detections of actual object. All detections are in background. Does anybody have a good explanation for this wrong behaviour?

Caliber2 avatar Apr 29 '19 06:04 Caliber2