You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
This pure-pytorch implement is up to 2 times faster than the official Tensorflow version without any trick.
28
-
29
-
Recorded on 2020-04-26,
30
-
31
-
official git version: <https://github.com/google/automl/commit/006668f2af1744de0357ca3d400527feaa73c122>
32
-
33
-
| coefficient | FPS(this repo, tested on RTX2080Ti) | FPS(official, tested on T4) | Ratio |
34
-
| :------: | :------: | :------: | :-----: |
35
-
| D0 | 36.20 | 42.1 | 0.86X |
36
-
| D1 | 29.69 | 27.7 | 1.07X |
37
-
| D2 | 26.50 | 19.7 | 1.35X |
38
-
| D3 | 22.73 | 11.8 | 1.93X |
39
-
| D4 | 14.75 | 7.1 | 2.08X |
40
-
| D5 | 7.11 | 3.6 | 1.98X |
41
-
| D6 | 5.30 | 2.6 | 2.03X |
42
-
| D7 | 3.73 | - | - |
43
-
44
-
Test method (this repo):
45
-
46
-
Run this test on 2080Ti, Ubuntu 19.10 x64.
47
-
48
-
1. Prepare a image tensor with the same content, size (1,3,512,512)-pytorch.
49
-
2. Initiate everything by inferring once.
50
-
3. Run 10 times with batchsize 1 and calculate the average time, including post-processing and visualization, to make the test more practical.
51
-
52
-
___
53
-
54
25
## Update Log
55
26
56
27
[2020-07-23] supports efficientdet-d7x, mAP 53.9, using efficientnet-b7 as its backbone and an extra deeper pyramid level of BiFPN. For the sake of simplicity, let's call it efficientdet-d8.
@@ -235,8 +206,9 @@ Check out this [tutorial](tutorial/train_shape.ipynb) if you are new to this. Yo
0 commit comments