-
Notifications
You must be signed in to change notification settings - Fork 3
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
6b67799
commit f955250
Showing
3 changed files
with
69 additions
and
45 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,98 +1,122 @@ | ||
# pLitter - Plastic Litter identification and monitoring | ||
|
||
pLitter-street is a standardized, deep learning friendly dataset and pre-trained model that can be used for detecting plastic litter at streets, road sides, and other outdoor areas. Additionally, all supplementary code related to this repository is also published here. *Example video showing plastic litter detection from our model (click on image to see the YouTube video) is shown below.* | ||
An initiative of GIC-AIT to address the plastic pollution through leveraging the Deep Learning and geospatial technologies. pLitter tool and its modules enbale us to map or monitor the plastic pollution in various parts of the outdoor environment. pLitter follows 4M (MAN, MACHINE, MAPPING, AND MONITORING) and uses different tools/methods for data (image/videos) colelction mathcing to the location of plastic pollution, and Deep Learning based object detectors for litter objects identification in the collected data. | ||
|
||
<p align="center"> | ||
<a href="https://www.youtube.com/watch?v=REv0XEcWXVE" target="_blank"> | ||
<img src="https://img.youtube.com/vi/REv0XEcWXVE/0.jpg" alt="DL_Detection" width="50%"/> | ||
</a> | ||
</p> | ||
* * * * * | ||
|
||
pLitter-street helps to make predictions to detect theplastic litter in the images/videos and these detections are being used to map plastic litter distribution (as snapshot of plastic pollution) in the cities. *Example heat-map of a beach at Rayong(Thailand) is showing plastic litter distribution in a city is shown below.* | ||
## # Mapping | ||
|
||
<p align="center"> | ||
<img src="./demo/figures/example_heatmap.PNG" alt="HeatMap" width="50%"/> | ||
</p> | ||
![VM camera](./demo/figures/VM_tuktuk.jpeg) | ||
|
||
Surveying the roadsides and streets at city scale reveals the littering conditions such as identification of hotspots. The feasible and cost-effective way is using vehicle mounted cameras for data collection. AI algorithms can detect the macro plastic litter in the imagery and it can be associated with the corresponding GPS location. We successfully used this method to map the city scale level in two cities Chiang Rai and Ubon Ratchathani in Thailand, Hanwella and Mawanella in Sri Lanka, and Can Tho in Vietnam. We call this approach as pLitter-Street | ||
|
||
pLitter-Street helps to make predictions to detect theplastic litter in the images/videos and these detections are being used to map plastic litter distribution (as snapshot of plastic pollution) in the cities. *Example heat-map of a beach at Rayong(Thailand) is showing plastic litter distribution in a city is shown below.* Visit https://plitter.org/street or https://gicait.github.io/pLitter/street/ plitter street mapping here. | ||
|
||
[![](https://img.youtube.com/vi/REv0XEcWXVE/0.jpg)](https://www.youtube.com/watch?v=REv0XEcWXVE) | ![](./demo/figures/example_heatmap.PNG) | ||
:-------------------------:|:-------------------------: | ||
Demo | Heatmap | ||
|
||
Drones are useful for data collection for plastic pollution mapping in remote areas where the road network is not available and manual data collection not feasible. Addressing the integration of plastic monitoring in the riverbank and water bodies, aerial images are deployed to deliver the higher resolution of plastic waste accumulated and floated in the river and map the extended area of the river. | ||
|
||
* * * * * | ||
|
||
## # Monitoring | ||
|
||
![](./demo/figures/cctv_unit.png) | [![](https://img.youtube.com/vi/A6wZhdmWK1Y/0.jpg)](https://www.youtube.com/watch?v=A6wZhdmWK1Y) | ||
:-------------------------:|:-------------------------: | ||
CCTV Unit | Timelapse video | ||
|
||
Rivers are one of the major pathways of plastic debris into the ocean. These plastics in the rivers may dispose through direct dumping into the river flow, catalysed by the rain and exacerbated by the flooding events. For the estimation and flux monitoring of floating plastic in the river, it requires stationary cameras for continuous image capturing at location with recording evidence of floating plastics. We installed multiple cameras on the Mekong River near the golden triangle in Chiang Rai of Thailand and our cameras are in continuous function from November 2021 to till date. The advantage of our cameras is AI enabled to automatically detect the floating plastics and low cost. | ||
|
||
## Motivation | ||
* * * * * | ||
|
||
## # Motivation | ||
|
||
The Geoinformatics Center (GIC) of Asian Institute of Technology (AIT) partnered with the United Nations Environment Program (UNEP) for years, to form a network of volunteer cleanup teams to perform beach/river cleanup activities. Together UNEP and GIC worked for the CounterMEAUSRE project (https://countermeasure.asia/) in 2018, funded by the Government of Japan. The CounterMEASURE project works to identify sources and pathways of plastic pollution in river systems in Asia, particularly the Mekong and the Ganges. During the project implementation, project partners including local governments and universities collected thousands of images using GIC Mobile Application (Plastic Accumulation Hotspot Survey: https://arcg.is/1bDqbW) and it was not used to identify plastic litter. Therefore, our ultimate target is to build image recognition tools with modern machine learning techniques to quickly identify plastic litter in different scenes such as roadside, riverside, and beachside. We do believed that the outcome of this work can be used to inform policy decisions and actions to beat plastic pollution and ensure rivers are free of plastic waste. | ||
|
||
## Our Approach | ||
* * * * * | ||
|
||
## # Our Approach | ||
|
||
In the conventional sense, typical machine learning pipeline consisted of collecting data, annotating data, training, validation and prediction. This is a very successful pipeline for many problems such as face detection, character recognition, etc. But the problem with plastic litter is, simply it is trash, it can be seen in various types, shapes, forms, backgrounds etc. So, variations are too complex to capture in a single model with a single annotated dataset. So we believe, the solution for this is Active Learning. Our ultimate goal would be to go beyond domain specific plastic litter detections, and provide detection capabilities in a wide range of cities (universal plastic litter detector). Schematic diagram of active learning framework that we are using for plastic litter identification problem is shows in the below Figure. | ||
|
||
<p align="center"> | ||
<img src="./demo/figures/active_learning.PNG" alt="HeatMap" width="70%"/> | ||
</p> | ||
|
||
* * * * * | ||
|
||
## # Datasets | ||
|
||
## Datasets | ||
We have prepared datsets separatelty for street level plastic litter (pLitterStreet) and floating plastic litters (pLitterFloat). These datasets contains RGB images, plastic litter annotations (JSON format same as COCO). | ||
|
||
The datasets used in this repository can be downloaded from gdrive folder [pLitter-street](https://drive.google.com/drive/folders/165ZP5b9yU6Za8nfhdfGpoSyGFJTUpfiI?usp=sharing). Dataset contains RGB images, plastic litter annotations (JSON format same as COCO). | ||
pLitterStreet dataset contains images from following locations, | ||
|
||
pLitter dataset contains images from following locations, | ||
* Mekong river basin | ||
* Pathumthani, Ubon Ratchathani, and Chiang Rai (Thailand) | ||
* Can Tho (Vietnam) | ||
* Hanwella & Mawanella (Sri Lanka) | ||
* Mekong river basin (contains piles i.e, clusters of litter on the roadsides) | ||
* Trying add more for new locations | ||
|
||
pLitterFloat contains data from, | ||
|
||
* Mekong river, Chaing Rai, Thailand | ||
* Ruak RIVER, Chiang Rai, Thialand | ||
* Chao Praya, Bangkok, Thailand | ||
* Mekong river, Cantho, Vietnam (coming soon) | ||
|
||
*Note: We have used a open-source Annotator tool built from [COCO annotator](https://github.com/jsbroks/coco-annotator) for generation of bounding box annotations with the help of group of annotators. If you are planning to generate your own training data, it is an option* | ||
|
||
## Pre-trained models | ||
* * * * * | ||
|
||
Models that are trained separately for each of the datasets, as well as model that is trained on combined dataset can be downloaded from following links, | ||
## # Pre-trained models | ||
|
||
<!--- | ||
| Talaad Thai | Rangsit | Ubon Ratchathani I | Chiang Rai I | Ubon Ratchathani II | Chiang Rai II | <ins>Combined Dataset</ins> | | ||
| --- | --- | --- | --- | --- | --- | --- | | ||
| [Docker](#) | [Docker](#) | Docker (ongoing) | Docker (ongoing) | Docker (ongoing) | Docker (ongoing) | [Docker](#) | | ||
| [EDGE Model](#) | [EDGE Model](#) | EDGE Model (ongoing) | EDGE Model (ongoing) | EDGE Model (ongoing) | EDGE Model (ongoing) | [EDGE Model](#) | | ||
| mAP = xx | mAP = xx | mAP = (ongoing) | mAP = (ongoing) | mAP = (ongoing) | mAP = (ongoing) | mAP = (ongoing) | | ||
--> | ||
Pre-trained models will be updated very soon. | ||
|
||
| Model | mAP | | ||
| --- | --- | | ||
| Google AutoML\* [Cloud] | 0.77 | | ||
| Faster RCNN (R101-FPN) | 0.67 | | ||
| Dataset | AP | Link | | ||
| --- | --- | --- | | ||
| Street | 0.77 | - | | ||
| CCTV | 0.43 | - | | ||
|
||
*Note: \*These models are trained using [Google AutoML Vision tools](https://cloud.google.com/automl) in [Google Cloud Platform](https://cloud.google.com/).* | ||
* * * * * | ||
|
||
## Usage | ||
## # Usage | ||
|
||
Refer to [pLitter demo](/demo/pLitterStreet_demo.ipynb) | ||
Refer to [pLitter Street demo](/demo/pLitterStreet_demo.ipynb) for street level detection and mapping | ||
Refer to [pLitter CCTV demo](/demo/pLitterStreet_demo.ipynb) for detecting the floating plastics. | ||
|
||
Also visit [pLitter pages](https://plitter.org) to look at the all initiatives of pLitter. | ||
|
||
## ToDo | ||
|
||
To deal with 360 images. | ||
|
||
Train a light weight model for deploying with Rpi/jetson to fix it on garbage truck for real time prediction. | ||
* * * * * | ||
|
||
## Citation | ||
## # Citation | ||
|
||
Use the below bibtex to cite us. | ||
|
||
```BibTeX | ||
@misc{pLitterStreet_2021, | ||
title={pLitter-street, Plastic Litter detection along the streets using deep learning}, | ||
author={Sriram Reddy, Chatura Lavanga, Lakmal Deshapriya, Dan Tran, Kavinda Gunasekara, and Sujit}, | ||
author={Sriram Reddy, Lakmal Deshapriya, Chatura Lavanga, Dan Tran, Kavinda Gunasekara, Frank Yrle, Angsana Chaksan, and Sujit}, | ||
year={2021}, | ||
publisher={Github}, | ||
howpublished={\url{https://github.com/gicait/pLitter/}}, | ||
} | ||
@misc{pLitterFloat_2022, | ||
title={pLitter-float, floating plastic litter detection in the rivers}, | ||
author={Sriram Reddy, Chatura Lavanga, Kavinda Gunasekara, and Angsana Chaksan}, | ||
year={2022}, | ||
publisher={Github}, | ||
howpublished={\url{https://github.com/gicait/pLitter/}}, | ||
} | ||
``` | ||
* * * * * | ||
|
||
## Developed by | ||
|
||
[Geoinformatics Center](www.geoinfo.ait.ac.th) of [Asian Institute of Technology](www.ait.ac.th) and [Google](https://about.google/). | ||
|
||
<!-- __Our Team__ | ||
* To be added | ||
* ... --> | ||
[Geoinformatics Center](www.geoinfo.ait.ac.th) of [Asian Institute of Technology](www.ait.ac.th). | ||
|
||
## Funding | ||
## Supported by | ||
|
||
[CounterMEASURE](https://countermeasure.asia/) project of [UN Environment Programme](https://www.unep.org/). |
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.