libtorch yolov3
1.0.0
تطبيق Libtorch لخوارزمية الكشف عن الكائنات YOLO v3، المكتوبة بلغة C++ النقية. إنه سريع وسهل التكامل مع إنتاجك، ويتم دعم كل من وحدة المعالجة المركزية (CPU) ووحدة معالجة الرسومات (GPU). استمتع ~
هذا المشروع مستوحى من إصدار pytorch، وقد قمت بإعادة كتابته باستخدام C++.
mkdir build && cd build
cmake3 -DCMAKE_PREFIX_PATH="your libtorch path" ..
# if there are multi versions of gcc, then tell cmake which one your want to use, e.g.:
cmake3 -DCMAKE_PREFIX_PATH="your libtorch path" -DCMAKE_C_COMPILER=/usr/local/bin/gcc -DCMAKE_CXX_COMPILER=/usr/local/bin/g++ ..
أول ما عليك فعله هو الحصول على ملف الأوزان للإصدار 3:
cd models
wget https://pjreddie.com/media/files/yolov3.weights
على صورة واحدة:
./yolo-app ../imgs/person.jpg
كما اختبرت، سيستغرق الأمر 25 مللي ثانية على وحدة معالجة الرسومات (1080 ti). يرجى تشغيل مهمة الاستدلال أكثر من مرة، وحساب متوسط التكلفة.