หน้าแรก>การเขียนโปรแกรมที่เกี่ยวข้อง>ซี/ซี++
ดาวน์โหลด

libtorch-yolov3

การใช้ Libtorch ของอัลกอริธึมการตรวจจับวัตถุ YOLO v3 ซึ่งเขียนด้วย C++ ล้วนๆ รวดเร็ว ง่ายต่อการรวมเข้ากับการผลิตของคุณ และรองรับทั้ง CPU และ GPU สนุก ~

โปรเจ็กต์นี้ได้รับแรงบันดาลใจจากเวอร์ชัน pytorch ฉันเขียนใหม่ด้วย C++

ความต้องการ

  1. LibTorch v1.0.0
  2. สุดา
  3. OpenCV (เพิ่งใช้ในตัวอย่าง)

เพื่อรวบรวม

  1. ซีเมค3
  2. gcc5.4+ 
 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++ ..

กำลังเรียกใช้เครื่องตรวจจับ

สิ่งแรกที่คุณต้องทำคือรับไฟล์น้ำหนักสำหรับ v3: 

 cd models
wget https://pjreddie.com/media/files/yolov3.weights

ในภาพเดียว: 

 ./yolo-app ../imgs/person.jpg

ตามที่ฉันทดสอบ GPU จะใช้เวลา 25 มิลลิวินาที ( 1080 ti ) โปรดรันงานการอนุมานมากกว่าหนึ่งครั้ง และคำนวณต้นทุนเฉลี่ย

ขยาย
ข้อมูลเพิ่มเติม
แอปที่เกี่ยวข้อง
แนะนำสำหรับคุณ
ข้อมูลที่เกี่ยวข้อง ทั้งหมด