assert best possible recall > 0.9 before training

This commit is contained in:
Glenn Jocher committed 2020-06-13 15:05:41 -07:00
1 parent 19e68e8a7b
commit 31f3310029
3 files changed
+30 -12

No files matched your search

+4 -1
View File
@@ -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)