search

nvdla / hw

RTL, Cmodel, and testbench for NVDLA
Other
1.74k stars 568 forks source link

Does nvdla support object detection #171

Closed Adithyak1998 closed 6 years ago

Adithyak1998 commented 6 years ago

I recently raised an issue in nvdla sw: https://github.com/nvdla/sw/issues/67. I found out that nvdla_compiler doesn't support object detection. I just wanted to clarify if the NVDLA HW supports object detection.

prasshantg commented 6 years ago

@Adithyak1998 this depends on what layers are being used in object detection network. NVDLA HW supports almost all type of layers used in CNN except some such as ROI pooling.

Adithyak1998 commented 6 years ago

Does it support power layer.

I want these to work: https://raw.githubusercontent.com/NVIDIA/caffe/caffe-0.15/examples/kitti/detectnet_network.prototxt https://raw.githubusercontent.com/NVIDIA/caffe/caffe-0.15/examples/kitti/detectnet_network-2classes.prototxt

nvhook commented 6 years ago

Closing as resolved in nvdla/sw#67 .