Browse Source

rename test

5.0
Jirka 4 years ago
parent
commit
ef6f5b33a8
2 changed files with 20 additions and 20 deletions
  1. +12
    -12
      eval.py
  2. +8
    -8
      train.py

+ 12
- 12
eval.py View File

@@ -5,7 +5,7 @@ from models.experimental import *
from utils.datasets import *


def evaluate(data,
def test(data,
weights=None,
batch_size=16,
imgsz=640,
@@ -254,16 +254,16 @@ if __name__ == '__main__':
print(opt)

if opt.task in ['val', 'test']: # run normally
evaluate(opt.data,
opt.weights,
opt.batch_size,
opt.img_size,
opt.conf_thres,
opt.iou_thres,
opt.save_json,
opt.single_cls,
opt.augment,
opt.verbose)
test(opt.data,
opt.weights,
opt.batch_size,
opt.img_size,
opt.conf_thres,
opt.iou_thres,
opt.save_json,
opt.single_cls,
opt.augment,
opt.verbose)

elif opt.task == 'study': # run over a range of settings and save/plot
for weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']:
@@ -272,7 +272,7 @@ if __name__ == '__main__':
y = [] # y axis
for i in x: # img-size
print('\nRunning %s point %s...' % (f, i))
r, _, t = evaluate(opt.data, weights, opt.batch_size, i, opt.conf_thres, opt.iou_thres, opt.save_json)
r, _, t = test(opt.data, weights, opt.batch_size, i, opt.conf_thres, opt.iou_thres, opt.save_json)
y.append(r + t) # results and times
np.savetxt(f, y, fmt='%10.4g') # save
os.system('zip -r study.zip study_*.txt')

+ 8
- 8
train.py View File

@@ -291,14 +291,14 @@ def train(hyp):
ema.update_attr(model, include=['md', 'nc', 'hyp', 'gr', 'names', 'stride'])
final_epoch = epoch + 1 == epochs
if not opt.notest or final_epoch: # Calculate mAP
results, maps, times = eval.evaluate(opt.data,
batch_size=batch_size,
imgsz=imgsz_test,
save_json=final_epoch and opt.data.endswith(os.sep + 'coco.yaml'),
model=ema.ema,
single_cls=opt.single_cls,
dataloader=testloader,
save_dir=log_dir)
results, maps, times = eval.test(opt.data,
batch_size=batch_size,
imgsz=imgsz_test,
save_json=final_epoch and opt.data.endswith(os.sep + 'coco.yaml'),
model=ema.ema,
single_cls=opt.single_cls,
dataloader=testloader,
save_dir=log_dir)

# Write
with open(results_file, 'a') as f:

Loading…
Cancel
Save