# MobileNet-SSD-TensorRT

**Repository Path**: gyx5668/MobileNet-SSD-TensorRT

## Basic Information

- **Project Name**: MobileNet-SSD-TensorRT
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2021-03-16
- **Last Updated**: 2021-03-16

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# MobileNet-SSD-TensorRT
**To accelerate mobileNet-ssd with tensorRT**

**TensorRT-Mobilenet-SSD can run 50fps on jetson tx2**

---

**Requierments:**

1.tensorRT4

2.cudnn7

3.opencv

---

**Run:**

```shell
cmake .
make
./build/bin/mobileNet
```

---

**Reference:**

https://github.com/saikumarGadde/tensorrt-ssd-easy

https://github.com/chuanqi305/MobileNet-SSD

I replaced depthwise with group_conv,because group_conv  has been optimized in cudnn7

I retrianed mobileNet-SSD,my number of classfication is 5

---

**TODO:**

- [x] To save serialized model 
- [x] To solve the bug of getting different result with same input
- [ ] The bottleneck of time cost lies in the decoding of pictures. "imread" cost too much ,to resolve it.
- [x] To modify the architecture, decrease the time cost



*If want to decrease the time cost of "imread",you could rebuild OpenCV[https://github.com/jetsonhacks/buildOpenCVTX2]*

*Added producer-consumer*



**The bug has been fixed**

![image](testPic/test1.png)