Why the training makes no efforts? about yolo_tensorflow

[ericdoerheit]

Have you used a pre-trained network or where you training from scratch? In the original paper it is described, that they train the first CNN layers for solving a classification task, e. g. based on the ImageNet dataset and then add some more layers and fine-tune for object detection. [Section 2.2, https://arxiv.org/abs/1506.02640] I am not sure whether this is the reason for the problem, but maybe you could try to train with pre-trained weights?

[shadowdyj]

I have the problem when I train the model without pre-training. The total loss goes from 42 to 10, and cannot get less any more. However I get loss about 6 when I fine-turn the model using the Yolo_small.ckpt .
Is it impossible to train the model without any pre-trained weight?

[zy486at189cn]

edit config.py,just as this:

#WEIGHTS_FILE = None
WEIGHTS_FILE = os.path.join(DATA_PATH, 'weights', 'YOLO_small.ckpt')

when turn on the pre-training weights YOLO_small.ckpt，the trained model can make effect!

my test.py which use the trained model ,just like this:

class Detector(object):

def __init__(self, net, weight_file):
    self.net = net
    self.weights_file = weight_file

    self.classes = cfg.CLASSES
    self.num_class = len(self.classes)
    self.image_size = cfg.IMAGE_SIZE
    self.cell_size = cfg.CELL_SIZE
    self.boxes_per_cell = cfg.BOXES_PER_CELL
    self.threshold = cfg.THRESHOLD
    self.iou_threshold = cfg.IOU_THRESHOLD
    self.boundary1 = self.cell_size * self.cell_size * self.num_class
    self.boundary2 = self.boundary1 + self.cell_size * self.cell_size * self.boxes_per_cell

    self.sess = tf.Session()
    self.sess.run(tf.global_variables_initializer())
         
    PATH = **'/media/stockerc/f/wz/project/yolo/data/pascal_voc/output/2018_02_23_11_13'
    print 'Restoring weights from: ' + PATH
    self.saver = tf.train.Saver()
    ckpt = tf.train.get_checkpoint_state(PATH)
    self.saver.restore(self.sess, ckpt.model_checkpoint_path)

[joashchn]

@ailias
i have the seem question ,i changed the input data code(only train 'car' and 'cat'),but after the trainning,nothing can be detected
i did not use any pre-trained weights, and here i have another question,if the class i want to train is not in the pre-trained weights,can i use the pre-train weights?

[jongsukchoi]

@joashchn
hi joashchn
i have same problem with you.
did you solve the this problem?
my goal is the just one object detection in image.
perhaps, your goal is the two object detection in image.

[liuhyCV]

@joashchn @ailias did you solve the problem? I have train it and test nothing in cat.jpg or person.jpg. Its training problem?

[feitiandemiaomi]

@liuhyCV Did you meet the question of testing noting in person.jpg ? Just like the original jpg ,did not have any output and circle ， did you solve the problem?

[linydf]

I also meet the same question. I train the download model-YOLO-small ,but finally,I find the result is pretty worse,i set the iteration time is thirty thousands.

[ss199302]

Has anyone solved the problem of training?

[Liu-Yicheng]

I have the same problems with you. I thought the weight that I trained without pre-training was not good that cause the problem（detect nothing）.Then I changed the parameter‘THRESHOLD ’（in the config file） to 0.01，it did work！Although the result was not right！So my conclusion is if our loss can arrive about 6，the weight can be good enough to detect !

[ss199302]

@ailias Have you solved the problem of training?

[linydf]

now my loss arrive 3. but I am failed to result this condition

[ss199302]

@linydf What data set do you use?thanks！

[xiaoshuguo750]

@joashchn @jongsukchoi
Hi!
How can I changed the code if I want to input data (only train 'car' and 'cat')?
Thank you!
Shu Guo

[leadcain84]

How do you get the 'YOLO_small.ckpt you gave'?
is it your training results? or others.

[zy486at189cn]

I download it

…

-- 发自189邮箱 App 点击下载，安全、高效管理邮箱 http://t.mail.189.cn/rcyyzhw leadcain84 <[email protected]>写到： How do you get the 'YOLO_small.ckpt you gave'? is it your training results? or others. —You are receiving this because you commented.Reply to this email directly, view it on GitHub, or mute the thread.

[MindyZLM]

@joashchn i have same problem with you.
did you solve the this problem?

[XiangqianMa]

My model can get the loss of 3, but it can not detect any right objects. Before i trained this model to fine-tune my own data, i have tried to fine-tune this model on pascal-voc, it can detect the right objects, but it's confidence was very low, such as 0.6.

[M000M]

I have the same problem. I doubt that the model is right, but there's some errors in the loss function.

[guker]

I have the same problem, do you solve it?

[guker]

Have you used a pre-trained network or where you training from scratch? In the original paper it is described, that they train the first CNN layers for solving a classification task, e. g. based on the ImageNet dataset and then add some more layers and fine-tune for object detection. [Section 2.2, https://arxiv.org/abs/1506.02640] I am not sure whether this is the reason for the problem, but maybe you could try to train with pre-trained weights?

yes， you must pretrain，not train from scrath，otherwise，loss is about 10.0 and does not decrease

[LinzzMichael]

Have you used a pre-trained network or where you training from scratch? In the original paper it is described, that they train the first CNN layers for solving a classification task, e. g. based on the ImageNet dataset and then add some more layers and fine-tune for object detection. [Section 2.2, https://arxiv.org/abs/1506.02640] I am not sure whether this is the reason for the problem, but maybe you could try to train with pre-trained weights?

yes， you must pretrain，not train from scrath，otherwise，loss is about 10.0 and does not decrease

@guker I have some problem,if my the num of detect classes is 5 not 20,how can I pre train,can I pre train with the YOLO_small.ckpt,or I need to pre train with only 5 classes?