CODEWITHSUNDEEP
neural-networkartificial-intelligencecaffeobject-detectionnvidia-digits
Bondolin
Bondolin

Reputation: 3121

NVidia DIGITS DetectNet alternatives

I am looking for detection/localization CNNs to run within the NVidia DIGITS training platform. So far it seems they only support their homebrew DetectNet for this purpose. Looking around it seems that other SOTA networks such as faster-RCNN, SSD, and YOLO might compete with DetectNet in terms of performance and accuracy, but it does not look like they currently have any support in DIGITS. (Faster-RCNN has a fairly popular implementation, but it is run out of a version of Caffe not supported by DIGITS.)

If anyone has had any success obtaining and using SOTA detection networks with NVidia DIGITS, would you mind supplying links/documentation regarding?

Upvotes: 2

Views: 1115

Answers (1)

jefflgaol
jefflgaol

Reputation: 73

Currently, NVIDIA provides NVIDIA Transfer Learning Toolkit which is useful for training CNNs. NVIDIA provides SSD, Faster RCNN, DetectNet, etc. You can find more here: https://developer.nvidia.com/transfer-learning-toolkit

Upvotes: 1

Related Questions