Skip to content

Commit 07ad263

Browse files
committed
added Hardware info to README.md and minor clean up in main.cpp
1 parent c8cb3fd commit 07ad263

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

README.md

+8
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,12 @@ layer of the FaceNet model. Link to the repo:
99
Moreover, this project uses an adapted version of [PKUZHOU's implementation](https://github.com/PKUZHOU/MTCNN_FaceDetection_TensorRT)
1010
of the mtCNN for face detection. More info below.
1111

12+
## Hardware
13+
* NVIDIA Jetson Nano
14+
* Raspberry Pi v2 camera
15+
16+
If you want to use a USB camera instead of Raspi Camera set the boolean _isCSICam_ to false in [main.cpp](./src/main.cpp).
17+
1218

1319
## Dependencies
1420
cuda 10.0 + cudnn 7.5 <br> TensorRT 5.1.x <br> OpenCV 3.x <br>
@@ -115,6 +121,8 @@ Performance on **NVIDIA Jetson AGX Xavier**:
115121
Please respect all licenses of OpenCV and the data the machine learning models (mtCNN and Google FaceNet)
116122
were trained on.
117123

124+
125+
118126
## Info
119127
Niclas Wesemann <br>
120128

src/step02_main.cpp src/main.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ int main()
2626
// USER DEFINED VALUES
2727
const string uffFile="../facenetModels/facenet.uff";
2828
const string engineFile="../facenetModels/facenet.engine";
29-
DataType dtype = DataType::kHALF; // ToDo calibrator for INT8
29+
DataType dtype = DataType::kHALF;
3030
//DataType dtype = DataType::kFLOAT;
3131
bool serializeEngine = true;
3232
int batchSize = 1;

0 commit comments

Comments
 (0)