tiny fix in fasterrcnn

d49dc2e4 · chenyuntc · 9bce6bfe · d49dc2e4
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

model/faster_rcnn.py model/faster_rcnn.py +1 -0

未找到文件。
--- a/model/faster_rcnn.py
+++ b/model/faster_rcnn.py
@@ -92,6 +92,7 @@ class FasterRCNN(nn.Module):
        # mean and std
        self.loc_normalize_mean = loc_normalize_mean
        self.loc_normalize_std = loc_normalize_std
+        self.use_preset('evaluate')

    @property
    def n_class(self):