update mAP

Luo-Chang · Dec 22, 2017 · f7d84f0 · f7d84f0
1 parent 6de4623
commit f7d84f0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/README.MD b/README.MD
@@ -17,10 +17,10 @@ VGG16 train on trainval and test on test, Note, the training show great randomne
 |              Implementation              | mAP         |
 | :--------------------------------------: | ----------- |
 | [origin paper](https://arxiv.org/abs/1506.01497) | 0.699       |
-| using caffe pretrained model (enable with`--caffe-pretrain`) | 0.700-0.708 |
-|    using torchvision pretrained model    | 0.690-0.701 |
+| using caffe pretrained model (enable with`--caffe-pretrain`) | 0.702-0.712 |
+|    using torchvision pretrained model    | 0.693-0.701 |
 | model converted from [chainercv](https://github.com/chainer/chainercv/tree/master/examples/faster_rcnn) (reported 0.706) | 0.7053      |
-| current the best i've ever seen (ruotian's) | 0.711       |
+
 
 - Speed
 

diff --git a/trainer.py b/trainer.py
@@ -213,7 +213,7 @@ def load(self, path, load_optimizer=True, parse_opt=False,):
         if 'model' in state_dict:
             self.faster_rcnn.load_state_dict(state_dict['model'])
         else:  # legacy way, for backward compatibility
-            self.load_state_dict(state_dict)
+            self.faster_rcnn.load_state_dict(state_dict)
             return self
         if parse_opt:
             opt._parse(state_dict['config'])