Skip to content

Repository for a thermal imaging solution aimed at vehicle tracking in drone-captured videos, incorporating oriented bounding boxes and SHAI for superior detection capabilities.

Notifications You must be signed in to change notification settings

z00bean/ThermalDroneVehicleOBB

Repository files navigation

Analyzing Highway Work Zone Traffic Dynamics via Thermal Videos and Deep Learning

1 Miner School of Computer & Information Sciences, University of Massachusetts Lowell, 1 University Avenue, Lowell, USA.

2 College of Automobile and Traffic Engineering, Nanjing Forestry University, China.

3 Department of Civil & Environmental Engineering, University of Massachusetts Lowell, 1 University Avenue, Lowell, USA.

This work was supported by Massachusetts Department of Transportation

This research is part of the Smart Work Zone Control and Performance Evaluation Based on Trajectory Data project, funded by the Massachusetts Department of Transportation. For further analysis and research involving radar and ground thermal cameras, please visit the repository at: https://github.com/z00bean/SmartWorkZoneControl.

Code review is currently in progress. Additional details will be provided after acceptance of paper.

ThermalDrone_VehicleOBB

Repository for a thermal imaging solution aimed at vehicle tracking in drone-captured videos, incorporating oriented bounding boxes and SHAI for superior detection capabilities.

Medford Danvers
Medford, MA: I-93, near Exit 21. (https://maps.app.goo.gl/muLRTy4BFLoJyUoj7) Danvers, MA: I-93, near Exit 10. (https://maps.app.goo.gl/j4ysifxbs8VmehxP6)

Comparing deep learning-based methods with traditional computer vision tracking.

Deep learning models offer superior accuracy and robustness in detecting and tracking vehicles but require significant computational resources and training data. On the other hand, traditional computer vision methods are less resource-intensive and can be quickly implemented but might lack the precision and adaptability of deep learning techniques, especially in complex or dynamic scenes.

A notable challenge in processing drone video data arises during nighttime conditions, where low light and glare from headlights can significantly degrade video quality. Basic computer vision tracking methods have shown promise in analyzing thermal drone footage, especially due to the distinct features observable in top-down views. Techniques such as background subtraction, contour detection, and the use of more traditional tracking methods such as Kernelized Correlation Filter tracker, leverage the simplicity and high contrast of vehicles against the road surface. These methods are particularly effective in scenarios where deep learning models might be computationally intensive or require extensive training data. This approach, while beneficial in terms of computational efficiency and effectiveness for certain scenarios, comes with the caveat that parameter adjustments are necessary for each new location; at times, even drone movement could necessitate recalibrations, adding a layer of complexity to their application.

Medford, MA: Heatmap, OBB + tracking. Danvers, MA: Heatmap, OBB + Tracking.

Thermal imaging allows drones to capture images in complete darkness or through obstacles like smoke, providing a significant utility. A comparison of nighttime images taken simultaneously with regular RGB and thermal cameras from a drone illustrates this point. In the RGB images, vehicles are barely visible. However, in thermal images, objects display distinct shapes and unique thermal signatures. This difference presents a challenge, especially when objects appear small in drone-captured images, making detection and classification difficult.

RGB vs. Thermal

This research presents three key contributions:

  1. Dataset: We introduce a novel thermal drone dataset designed for nighttime use. This dataset features over 13,500 instances of vehicle annotations marked with oriented bounding boxes.
  2. Model: We propose an enhanced model architecture building on the YOLOv8 model to improve detection of small objects. This includes a specialized training strategy tailored for this specific application.
  3. Selective SAHI: We develop a method called Selective Slicing Aided Hyper Inference (SAHI) that selectively processes image slices based on their overlap with roadway segmentation results. This approach concentrates the model's attention on the most relevant areas, enhancing the detection's efficiency and effectiveness.
The study further illustrates how the proposed model and thermal imaging can be effectively utilized to monitor and analyze work zones under nocturnal or low-visibility conditions.
PAFPN-P2
Sample output- OBB detections
Visualization of roadway segmentation and work zone region estimation.

In our analysis based on drone videos at the Medford location, we closely monitored 140 vehicles as they traversed over the rumble strips. Our findings revealed that only 40 of these vehicles exhibited a speed reduction greater than 15% upon encountering the rumble strips. This indicates that the presence of rumble strips, while effective for some vehicles, does not universally lead to significant speed adjustments. Furthermore, within the context of lane changes, particularly in the second lane from the left—which also leads directly to a downstream exit ramp—only 22 vehicles (or approximately 15%) engaged in immediate lane changes within a 160-foot distance following the rumble strips. This observation underscores the complexity of interpreting vehicle behavior, as not changing lanes could also align with the intention to exit, rather than not being influenced by the rumble strips.

At the Danvers location, the analysis of trajectories for 100 vehicles revealed a consistent pattern, where a similar proportion (15%) of drivers executed immediate lane changes upon encountering rumble strips. This consistency across both locations suggests a broader trend where the presence of rumble strips influences a subset of drivers to adjust their speed or change lanes, albeit a relatively small fraction.

About

Repository for a thermal imaging solution aimed at vehicle tracking in drone-captured videos, incorporating oriented bounding boxes and SHAI for superior detection capabilities.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published