-
Notifications
You must be signed in to change notification settings - Fork 0
/
eval_bt.py
79 lines (66 loc) · 5.44 KB
/
eval_bt.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
from ultralytics.models import YOLO
import numpy as np
import random
import os
import torch
def seed_torch(seed=0):
random.seed(seed)
os.environ['PYTHONHASHSEED'] = str(seed)
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.cuda.manual_seed_all(seed) # if you are using multi-GPU.
torch.backends.cudnn.benchmark = False
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.enabled = False
seed_torch(0)
# Load a model
ckpt_path = "runs/detect/endovis/weights/best.pt"
# ckpt_path = "runs/detect/endovis-spd/weights/best.pt"
# ckpt_path = "runs/detect/endovis-spd-cbma/weights/best.pt"
# ckpt_path = "runs/detect/endovis-rmt1-p1/weights/best.pt"
# ckpt_path = "runs/detect/endovis-rmt1-p2/weights/best.pt"
# ckpt_path = "runs/detect/endovis-rmt1-p5/weights/best.pt"
# ckpt_path = "runs/detect/endovis-rmt2-p1/weights/best.pt"
# ckpt_path = "runs/detect/endovis-rmt2-p2/weights/best.pt"
# ckpt_path = "runs/detect/endovis-rmt2-p5/weights/best.pt"
# ckpt_path = "runs/detect/endovis-rmt3-p1/weights/best.pt"
# ckpt_path = "runs/detect/endovis-rmt3-p2/weights/best.pt"
# ckpt_path = "runs/detect/endovis-rmt3-p5/weights/best.pt"
# ckpt_path = "runs/detect/endovis-mup/weights/best.pt"
# ckpt_path = "runs/detect/endovis-all/weights/best.pt"
# Load a model
model = YOLO(ckpt_path) # build a new model from scratch
name = ckpt_path.split("/")[-3] + "-bt"
hyp = {
'seed': 0,
'name': name,
'data': 'datasets/endovis_bt.yaml',
'epochs': 100,
'device': '1',
'workers': 8,
# 'mixup':0.5,
}
model.eval(**hyp) # train the model
# nohup yolo task=detect mode=val name=bt-endovis model=runs/detect/endovis/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-spd-cf model=runs/detect/endovis-spd-cf/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-spd-cf.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-spd-ct model=runs/detect/endovis-spd-ct/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-spd-ct.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-spd-cfnew model=runs/detect/endovis-spd-cfnew/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-spd-cfnew.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-spd-ctnew model=runs/detect/endovis-spd-ctnew/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-spd-ctnew.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-spd-cfnew-bias model=runs/detect/endovis-spd-cfnew-bias/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-spd-cfnew-bias.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-spd-ctnew-bias model=runs/detect/endovis-spd-ctnew-bias/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-spd-ctnew-bias.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-rmt1-p1 model=runs/detect/endovis-rmt1-p1/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-rmt1-p1.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-rmt1-p2 model=runs/detect/endovis-rmt1-p2/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-rmt1-p2.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-rmt1-p5 model=runs/detect/endovis-rmt1-p5/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-rmt1-p5.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-rmt2-p1 model=runs/detect/endovis-rmt2-p1/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-rmt2-p1.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-rmt2-p2 model=runs/detect/endovis-rmt2-p2/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-rmt2-p2.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-rmt2-p5 model=runs/detect/endovis-rmt2-p5/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-rmt2-p5.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-rmt3-p1 model=runs/detect/endovis-rmt3-p1/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-rmt3-p1.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-rmt3-p2 model=runs/detect/endovis-rmt3-p2/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-rmt3-p2.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-rmt3-p5 model=runs/detect/endovis-rmt3-p5/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-rmt3-p5.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-mup model=runs/detect/endovis-mup/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-mup.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-all model=runs/detect/endovis-all/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-all.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-all-spd-cfalse-sgd model=runs/detect/endovis-all-spd-cfalse-sgd/weights/best.pt data=datasets/endovis_bt.yaml device=1 > bt-endovis-all-spd-cfalse-sgd.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-v5 model=runs/detect/endovis-v5/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-v5.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-v6 model=runs/detect/endovis-v6/weights/best.pt data=datasets/endovis_bt.yaml device=3 > bt-endovis-v6.log 2>&1 &
# nohup yolo task=detect mode=val name=bt-endovis-v8n-mup model=runs/detect/endovis-v8n-mup/weights/best.pt data=datasets/endovis_bt.yaml device=0 > bt-endovis-v8n-mup.log 2>&1 &