add new trained model link and upload docker image

zjxcc · Nov 27, 2019 · 8da4b11 · 8da4b11
1 parent b061efb
commit 8da4b11
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/MODEL_ZOO.md b/MODEL_ZOO.md
@@ -12,7 +12,7 @@
 | [RetinaNet](https://github.com/DetectionTeamUCAS/RetinaNet_Tensorflow_Rotation) | ResNet50_v1 600->800 | DOTA1.0 trainval | DOTA1.0 test | 65.11 | [model](https://drive.google.com/file/d/1vnJd4FEvy61yvcYqhZJgfnxJ3fByULOR/view?usp=sharing) | H | smooth L1 + **atan(theta)** | 90 | 1x | No | 1X GeForce RTX 2080 Ti | 1 | cfgs_res50_dota_v16.py |     
 | [RetinaNet](https://github.com/DetectionTeamUCAS/RetinaNet_Tensorflow_Rotation) | ResNet50_v1 600->800 | DOTA1.0 trainval | DOTA1.0 test | 64.10 | [model](https://drive.google.com/file/d/1SgiDME_gHzKrFxoZSjS9E-_QbGiBr9lV/view?usp=sharing) | H | smooth L1 | **180** | 1x | No | 1X GeForce RTX 2080 Ti | 1 | cfgs_res50_dota_v15.py |     
 |  |  |  |  |  |  |  |  |  |  |  |  |  |
-| [RetinaNet](https://github.com/DetectionTeamUCAS/RetinaNet_Tensorflow_Rotation) (baseline) | ResNet50_v1 600->800 | DOTA1.0 trainval | DOTA1.0 test | 61.94 | - | R | smooth L1 | 90 | 1x | No | 1X GeForce RTX 2080 Ti | 1 | cfgs_res50_dota_v1.py |
+| [RetinaNet](https://github.com/DetectionTeamUCAS/RetinaNet_Tensorflow_Rotation) (baseline) | ResNet50_v1 600->800 | DOTA1.0 trainval | DOTA1.0 test | 62.76 | [model](https://drive.google.com/file/d/1n0O6qLJjdDewb_9FDgsGkISevL7SLD8_/view?usp=sharing) | R | smooth L1 | 90 | 1x | No | 1X GeForce RTX 2080 Ti | 1 | cfgs_res50_dota_v1.py |
 | [RetinaNet](https://github.com/DetectionTeamUCAS/RetinaNet_Tensorflow_Rotation) (baseline) | ResNet50_v1 600->800 | DOTA1.0 trainval | DOTA1.0 test | 62.25 | - | R | smooth L1 | 90 | **2x** | No | **8X** GeForce RTX 2080 Ti | 1 | cfgs_res50_dota_v10.py |
 | [RetinaNet](https://github.com/DetectionTeamUCAS/RetinaNet_Tensorflow_Rotation) | ResNet50_v1 600->800 | DOTA1.0 trainval | DOTA1.0 test | 68.65 | [model](https://drive.google.com/file/d/17RLZK0CwIgqtCAnifa0huWCa3EAaTw_l/view?usp=sharing) | R | [**iou-smooth L1**](https://arxiv.org/abs/1811.07126) | 90 | 1x | No | 1X GeForce RTX 2080 Ti | 1 | cfgs_res50_dota_v5.py |    
 |  |  |  |  |  |  |  |  |  |  |  |  |  |  |
@@ -22,8 +22,8 @@
 | [R<sup>3</sup>Det](https://arxiv.org/abs/1908.05612) | **ResNet152_v1** 600->800 | DOTA1.0 trainval | DOTA1.0 test | 72.81 | -  | H + R | smooth L1 | 90 | **4x** | Yes | 8X GeForce RTX 2080 Ti | 1 | - |
 | [R<sup>3</sup>Det*](https://arxiv.org/abs/1908.05612) | **ResNet152_v1** 600->800 | DOTA1.0 trainval | DOTA1.0 test | 73.74 | -  | H + R | smooth L1 | 90 | **4x** | Yes | 8X GeForce RTX 2080 Ti | 1 | - |
 |  |  |  |  |  |  |  |  |  |  |  |  |  |  |
-| **R<sup>3</sup>Det++** | ResNet50_v1 600->800 | DOTA1.0 trainval | DOTA1.0 test | 68.54 | -  | H + R | smooth L1 | 90 | 2x | No | 2X GeForce RTX 2080 Ti | 1 | cfgs_res50_dota_r3det_plusplus_v2.py |
-| R<sup>3</sup>Det++ | **ResNet152_v1** 600->800 | DOTA1.0 trainval | DOTA1.0 test | 74.41 | -  | H + R | smooth L1 | 90 | 4x | Yes | 8X GeForce RTX 2080 Ti | 1 | - |
+| **R<sup>3</sup>Det++** | ResNet50_v1 600->800 | DOTA1.0 trainval | DOTA1.0 test | 69.07 | [model](https://drive.google.com/file/d/1IucVSnPomj7UAH_aBApDv38ZKeufnNeb/view?usp=sharing) | H + R | smooth L1 | 90 | 2x | No | 4X GeForce RTX 2080 Ti | 1 | cfgs_res50_dota_r3det_plusplus_v2.py |
+| R<sup>3</sup>Det++ | **ResNet152_v1** 600->800 | DOTA1.0 trainval | DOTA1.0 test | 74.41 | - | H + R | smooth L1 | 90 | 4x | Yes | 8X GeForce RTX 2080 Ti | 1 | - |
 | R<sup>3</sup>Det++ | ResNet152_v1 **MS** | DOTA1.0 trainval | DOTA1.0 test | 76.56 | [model](https://drive.google.com/file/d/1DTEwh1Uyj14PgCjGFZW4jOdWdMw7GJQf/view?usp=sharing)  | H + R + more | smooth L1 | 90 | 6x | Yes | 4X GeForce RTX 2080 Ti | 1 | cfgs_res152_dota_r3det_plusplus_v1.py |
 
 [R<sup>3</sup>Det*](https://arxiv.org/abs/1908.05612): R<sup>3</sup>Det with two refinement stages      

diff --git a/README.md b/README.md
@@ -37,11 +37,12 @@ More results and trained models are available in the [MODEL_ZOO.md](MODEL_ZOO.md
 ![1](demo1.png)
 
 ## My Development Environment
+**docker images: docker pull yangxue2docker/yx-tf-det:tensorflow1.13.1-cuda10-gpu-py3**      
 1、python3.5 (anaconda recommend)               
-2、cuda >= 9.0                     
+2、cuda >= 10.0                     
 3、[opencv(cv2)](https://pypi.org/project/opencv-python/)       
 4、[tfplot 0.2.0](https://github.com/wookayin/tensorflow-plot) (optional)            
-5、tensorflow 1.12                            
+5、tensorflow 1.13                              
 
 ## Download Model
 ### Pretrain weights