assert best possible recall > 0.9 before training
This commit is contained in:
1 parent
19e68e8a7b
commit
31f3310029
3 files changed
+30
-12
No files matched your search
@@ -191,7 +191,7 @@ def train(hyp):
|
||||
model.class_weights = labels_to_class_weights(dataset.labels, nc).to(device) # attach class weights
|
||||
model.names = data_dict['names']
|
||||
|
||||
# class frequency
|
||||
# Class frequency
|
||||
labels = np.concatenate(dataset.labels, 0)
|
||||
c = torch.tensor(labels[:, 0]) # classes
|
||||
# cf = torch.bincount(c.long(), minlength=nc) + 1.
|
||||
@@ -199,6 +199,9 @@ def train(hyp):
|
||||
plot_labels(labels)
|
||||
tb_writer.add_histogram('classes', c, 0)
|
||||
|
||||
# Check anchors
|
||||
check_best_possible_recall(dataset, anchors=model.model[-1].anchor_grid, thr=hyp['anchor_t'])
|
||||
|
||||
# Exponential moving average
|
||||
ema = torch_utils.ModelEMA(model)
|
||||
|
||||
|
||||
Reference in new issue
Block a user