mirrors
/
yolov7
mirror of https://github.com/WongKinYiu/yolov7.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

New Train val set split Swiss-snow-2018 to training(test19g_sy_rd_ch) val:Test46c_ml_rd_il

pull/2109/head
hanoch 2024-11-19 13:35:43 +02:00
parent 2de1a2593c
commit 6c577e95c9
3 changed files with 39955 additions and 24796 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57320
tir_od/training_set.txt
View File

File diff suppressed because it is too large Load Diff

7422
tir_od/validation_set.txt
View File

File diff suppressed because it is too large Load Diff

9
train.py
View File

@ -359,6 +359,11 @@ def train(hyp, opt, device, tb_writer=None):
nb = len(dataloader) # number of batches
assert mlc < nc, 'Label class %g exceeds nc=%g in %s. Possible class labels are 0-%g' % (mlc, nc, opt.data, nc - 1)
with open(save_dir / 'trainig_set.txt', 'w') as f:
for file in dataset.img_files:
f.write(f"{file}\n")
# Process 0
if rank in [-1, 0]:
testloader , test_dataset = create_dataloader(test_path, imgsz_test, batch_size * 2, gs, opt, # testloader
@ -370,6 +375,10 @@ def train(hyp, opt, device, tb_writer=None):
mlc = np.concatenate(test_dataset.labels, 0)[:, 0].max() # max label class
assert mlc < nc, 'Label class %g exceeds nc=%g in %s. Possible class labels are 0-%g' % ( mlc, nc, opt.data, nc - 1)
with open(save_dir / 'test_set.txt', 'w') as f:
for file in test_dataset.img_files:
f.write(f"{file}\n")
if not opt.resume:
labels = np.concatenate(dataset.labels, 0)
c = torch.tensor(labels[:, 0]) # classes