วิธีการแก้ไขพิกัดของกล่องขอบเขตในแอปพลิเคชันการตรวจจับแบบกําหนดเอง
-
ฟังก์ชันที่ใช้ในการจัดเรียงสถาปัตยกรรมแบบจําลองเพื่อสร้างกล่องขอบเขตพร้อมพิกัดที่ถูกต้องดังนี้:
detections = result.reshape(-1, 5)
for i, detection in enumerate(detections):
xmin, ymin, xmax, ymax, confidence = detection
if confidence > 0.2:
xmin = int(max((xmin * image.shape[1]), 10))
ymin = int(max((ymin * image.shape[0]), 10))
xmax = int(min((xmax * image.shape[1]), image.shape[1] - 10))
ymax = int(min((ymax * image.shape[0]), image.shape[0] - 10)) -
ได้รับพิกัดที่ไม่ถูกต้องของกล่องขอบเขต
ตั้งค่า ระยะขอบจากขอบภาพเป็นอย่างน้อยสิบพิกเซลเมื่อวาดกล่องขอบเขตเป็นด้านล่าง:
xmin = int(max(xmin, 10))
ymin = int(max(ymin, 10))
xmax = int(min(xmax, image.shape[1] - 10))
ymax = int(min(ymax, image.shape[0] - 10))