javier/Photovoltaic_Fault_Detector
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

100 Example

Browse Source
This commit is contained in:
Daniel Saavedra
2020-06-03 22:46:26 -04:00
parent 8b0f8a44a2
commit 3b2459404c
105 changed files with 4019 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
keras-yolo3-master/utils/__pycache__/bbox.cpython-37.pyc
View File

Binary file not shown.

5
keras-yolo3-master/utils/bbox.py
View File

@@ -58,7 +58,7 @@ def bbox_iou(box1, box2):
return float(intersect) / union
def draw_boxes(image, boxes, labels, obj_thresh, quiet=True):
def draw_boxes(image, boxes, labels, obj_thresh, quiet=True, number_color = False):
for box in boxes:
label_str = ''
label = -1
@@ -77,6 +77,9 @@ def draw_boxes(image, boxes, labels, obj_thresh, quiet=True):
[box.xmin-3, box.ymin-height-16],
[box.xmin+width+6, box.ymin-height-16],
[box.xmin+width+6, box.ymin]], dtype='int32')
# Use one only color
if number_color:
label = number_color
cv2.rectangle(img=image, pt1=(box.xmin,box.ymin), pt2=(box.xmax,box.ymax), color=get_color(label), thickness=1)
cv2.fillPoly(img=image, pts=[region], color=get_color(label))