You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hey, I'm training my custom dataset of four classes and for some reason .train() throws the following error
cfg/yolo_custom.cfg parsing /home/iot/darkflow/data/ann/
Parsing for ['Apple', 'Cucumber', 'Tomato', 'Carrot']
Statistics:
Dataset size: 0
Dataset of 0 instance(s)
~/darkflow/darkflow/net/flow.py in train(self)
37 loss_op = self.framework.loss
38
---> 39 for i, (x_batch, datum) in enumerate(batches):
40 if not i: self.say(train_stats.format(
41 self.FLAGS.lr, self.FLAGS.batch,
~/darkflow/darkflow/net/yolo/data.py in shuffle(self)
100 print('Dataset of {} instance(s)'.format(size))
101 if batch > size: self.FLAGS.batch = batch = size
--> 102 batch_per_epoch = int(size / batch)
103
104 for i in range(self.FLAGS.epoch):
ZeroDivisionError: division by zero
I've rechecked my dataset and all the jpg files have a corresponding text file.
My options dictionary looks like this
options = {"model": "cfg/yolo_custom.cfg",
"load": "bin/yolo.weights",
"batch": 8,
"epoch": 100,
"gpu": 0,
"train": True,
"verbalize":True,
"annotation":"/home/iot/darkflow/data/ann/",
"dataset": "/home/iot/darkflow/data/images/"}
The text was updated successfully, but these errors were encountered:
Hey, I'm training my custom dataset of four classes and for some reason .train() throws the following error
cfg/yolo_custom.cfg parsing /home/iot/darkflow/data/ann/
Parsing for ['Apple', 'Cucumber', 'Tomato', 'Carrot']
Statistics:
Dataset size: 0
Dataset of 0 instance(s)
ZeroDivisionError Traceback (most recent call last)
in ()
----> 1 tfnet.train()
~/darkflow/darkflow/net/flow.py in train(self)
37 loss_op = self.framework.loss
38
---> 39 for i, (x_batch, datum) in enumerate(batches):
40 if not i: self.say(train_stats.format(
41 self.FLAGS.lr, self.FLAGS.batch,
~/darkflow/darkflow/net/yolo/data.py in shuffle(self)
100 print('Dataset of {} instance(s)'.format(size))
101 if batch > size: self.FLAGS.batch = batch = size
--> 102 batch_per_epoch = int(size / batch)
103
104 for i in range(self.FLAGS.epoch):
ZeroDivisionError: division by zero
I've rechecked my dataset and all the jpg files have a corresponding text file.
My options dictionary looks like this
options = {"model": "cfg/yolo_custom.cfg",
"load": "bin/yolo.weights",
"batch": 8,
"epoch": 100,
"gpu": 0,
"train": True,
"verbalize":True,
"annotation":"/home/iot/darkflow/data/ann/",
"dataset": "/home/iot/darkflow/data/images/"}
The text was updated successfully, but these errors were encountered: