Commit 390f538d authored by Ross Girshick's avatar Ross Girshick

DataLayer -> RoIDataLayer

parent 9054e662
......@@ -12,7 +12,7 @@ import numpy as np
import yaml
from multiprocessing import Process, queues
class DataLayer(caffe.Layer):
class RoIDataLayer(caffe.Layer):
"""Fast R-CNN data layer."""
def _shuffle_roidb_inds(self):
......@@ -36,7 +36,7 @@ class DataLayer(caffe.Layer):
db_inds = self._get_next_minibatch_inds()
minibatch_db = [self._roidb[i] for i in db_inds]
if cfg.TRAIN.USE_PREFETCH:
self._prefetch_process = Process(target=DataLayer._prefetch,
self._prefetch_process = Process(target=RoIDataLayer._prefetch,
args=(minibatch_db,
self._num_classes,
self._prefetch_queue))
......
......@@ -9,7 +9,7 @@ layer {
top: 'bbox_loss_weights'
python_param {
module: 'roi_data_layer.layer'
layer: 'DataLayer'
layer: 'RoIDataLayer'
param_str: "'num_classes': 21"
}
}
......
......@@ -9,7 +9,7 @@ layer {
top: 'bbox_loss_weights'
python_param {
module: 'roi_data_layer.layer'
layer: 'DataLayer'
layer: 'RoIDataLayer'
param_str: "'num_classes': 21"
}
}
......
......@@ -9,7 +9,7 @@ layer {
top: 'bbox_loss_weights'
python_param {
module: 'roi_data_layer.layer'
layer: 'DataLayer'
layer: 'RoIDataLayer'
param_str: "'num_classes': 21"
}
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment