We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 81c9033 commit 03f5f8cCopy full SHA for 03f5f8c
lib/roi_data_layer/roibatchLoader.py
@@ -51,7 +51,7 @@ def __init__(self, roidb, ratio_list, ratio_index, batch_size, num_classes, trai
51
# for ratio cross 1, we make it to be 1.
52
target_ratio = 1
53
54
- self.ratio_list_batch[left_idx:(right_idx+1)] = target_ratio
+ self.ratio_list_batch[left_idx:(right_idx+1)] = torch.tensor(target_ratio.astype(np.float64)) # trainset ratio list ,each batch is same number
55
56
57
def __getitem__(self, index):
0 commit comments