kristoferssolo/Traffic-Light-Detector
1
0
Fork 0
mirror of https://github.com/kristoferssolo/Traffic-Light-Detector.git synced 2025-10-21 20:00:36 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added type hintings

Browse Source
This commit is contained in:
Kristofers Solo 2022-12-10 17:00:58 +02:00
parent 5268913a43
commit f78272c874
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/detector/train_traffic_light_color.py
View File

@ -151,8 +151,8 @@ def train_traffic_light_color() -> None:
labels.extend([3] * len(img_3_not_traffic_light))
# Create NumPy array
labels_np = np.ndarray(shape=(len(labels), 4))
images_np = np.ndarray(shape=(len(labels), shape[0], shape[1], 3))
labels_np: np.ndarray[int, np.dtype[np.generic]] = np.ndarray(shape=(len(labels), 4))
images_np: np.ndarray[int, np.dtype[np.generic]] = np.ndarray(shape=(len(labels), shape[0], shape[1], 3))
# Create a list of all the images in the traffic lights data set
img_all = []
@ -177,7 +177,7 @@ def train_traffic_light_color() -> None:
logger.info(f"Labels: {len(labels)}")
# Perform one-hot encoding
for idx in range(len(labels_np)):
for idx, _ in enumerate(labels_np):
# We have four integer labels, representing the different colors of the
# traffic lights.
labels_np[idx] = np.array(to_categorical(labels[idx], 4))