add stride to datasets.py

This commit is contained in:
Glenn Jocher committed 2020-06-24 13:02:27 -07:00
1 parent 3b062254a6
commit b8557f87e3
3 files changed
+7 -4

No files matched your search

+1
View File
@@ -73,6 +73,7 @@ def test(data,
batch_size,
rect=True, # rectangular inference
single_cls=opt.single_cls, # single class mode
stride=int(max(model.stride)), # model stride
pad=0.5) # padding
batch_size = min(batch_size, len(dataset))
nw = min([os.cpu_count(), batch_size if batch_size > 1 else 0, 8]) # number of workers