Skip to content

Commit 951caec

Browse files
committed
update readme, yolov3-spp migrated to trt7
1 parent 4abc7b8 commit 951caec

File tree

10 files changed

+797
-519
lines changed

10 files changed

+797
-519
lines changed

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,7 @@ Some tricky operations encountered in these models, already solved, but might ha
7878
|-|-|:-:|:-:|:-:|:-:|
7979
| YOLOv3(darknet53) | Xavier | 1 | FP16 | 320x320 | 55 |
8080
| YOLOv3-spp(darknet53) | Xeon E5-2620/GTX1080 | 1 | FP32 | 256x416 | 94 |
81+
| YOLOv3-spp(darknet53) | Xeon E5-2620/GTX1080 | 1 | FP16 | 608x608 | 38.5 |
8182
| YOLOv4(CSPDarknet53) | Xeon E5-2620/GTX1080 | 1 | FP16 | 608x608 | 35.7 |
8283
| YOLOv4(CSPDarknet53) | Xeon E5-2620/GTX1080 | 4 | FP16 | 608x608 | 40.9 |
8384
| YOLOv4(CSPDarknet53) | Xeon E5-2620/GTX1080 | 8 | FP16 | 608x608 | 41.3 |

arcface/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ The mxnet implementation is from [deepinsight/insightface.](https://github.com/d
44

55
The pretrained model is [LResNet50E-IR,ArcFace@ms1m-refine-v1.](https://github.com/deepinsight/insightface/wiki/Model-Zoo#32-lresnet50e-irarcfacems1m-refine-v1)
66

7-
The two images used in this project are joey0.ppm and joey1.ppm, download them from [Google Drive.](https://drive.google.com/drive/folders/1ctqpkRCRKyBZRCNwo9Uq4eUoMRLtFq1e)
7+
The two input images used in this project are joey0.ppm and joey1.ppm, download them from [Google Drive.](https://drive.google.com/drive/folders/1ctqpkRCRKyBZRCNwo9Uq4eUoMRLtFq1e). The input image is 112x112, and generated from `get_input()` in `insightface/deploy/face_model.py`, which is cropped and aligned face image.
88

99
<p align="center">
1010
<img src="https://user-images.githubusercontent.com/15235574/83122953-f45f8d80-a106-11ea-84b0-4f6ff91b5924.jpg">

yolov3-spp/CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,8 @@ cuda_add_library(yololayer SHARED ${PROJECT_SOURCE_DIR}/yololayer.cu)
3232
find_package(OpenCV)
3333
include_directories(OpenCV_INCLUDE_DIRS)
3434

35-
add_executable(yolov3-spp ${PROJECT_SOURCE_DIR}/plugin_factory.cpp ${PROJECT_SOURCE_DIR}/yolov3-spp.cpp)
36-
target_link_libraries(yolov3-spp nvinfer nvinfer_plugin)
35+
add_executable(yolov3-spp ${PROJECT_SOURCE_DIR}/yolov3-spp.cpp)
36+
target_link_libraries(yolov3-spp nvinfer)
3737
target_link_libraries(yolov3-spp cudart)
3838
target_link_libraries(yolov3-spp yololayer)
3939
target_link_libraries(yolov3-spp ${OpenCV_LIBS})

yolov3-spp/common.h

Lines changed: 0 additions & 356 deletions
This file was deleted.

0 commit comments

Comments
 (0)