-
Notifications
You must be signed in to change notification settings - Fork 0
/
変更点.txt
62 lines (52 loc) · 1.83 KB
/
変更点.txt
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
変更点
###"C:\Users\taich\anaconda3\envs\test\Lib\site-packages\torch\utils\data\sampler.py"の
generator = torch.Generator() -----> generator = torch.Generator(device='cuda')
###yolact/utils/augmentations.pyの
def __init__(self):
self.sample_options = (
# using entire original input image
None,
# sample a patch s.t. MIN jaccard w/ obj in .1,.3,.4,.7,.9
(0.1, None),
(0.3, None),
(0.7, None),
(0.9, None),
# randomly sample a patch
(None, None),
)
||
||
||
||
||
V
def __init__(self):
self.sample_options = np.array(
[
# using entire original input image
None,
# sample a patch s.t. MIN jaccard w/ obj in .1,.3,.4,.7,.9
(0.1, None),
(0.3, None),
(0.7, None),
(0.9, None),
# randomly sample a patch
(None, None),
],
dtype=object
)
###yolact/train.pyの
data_loader = data.DataLoader(dataset, args.batch_size,
num_workers=args.num_workers,
shuffle=True, collate_fn=detection_collate,
pin_memory=True)
||
||
||
||
||
V
data_loader = data.DataLoader(dataset, args.batch_size,
num_workers=args.num_workers,
shuffle=False, collate_fn=detection_collate,
pin_memory=True)