|
@@ -48,7 +48,7 @@ class VOCDetection(Dataset):
|
|
|
transforms=None,
|
|
transforms=None,
|
|
|
num_workers='auto',
|
|
num_workers='auto',
|
|
|
buffer_size=100,
|
|
buffer_size=100,
|
|
|
- parallel_method='process',
|
|
|
|
|
|
|
+ parallel_method='thread',
|
|
|
shuffle=False):
|
|
shuffle=False):
|
|
|
from pycocotools.coco import COCO
|
|
from pycocotools.coco import COCO
|
|
|
super(VOCDetection, self).__init__(
|
|
super(VOCDetection, self).__init__(
|