Advertisement
daisukiseiya

Untitled

May 11th, 2018
93
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.92 KB | None | 0 0
  1. // Dataset
  2. In order to create an object detection algorithm that can identify litter, we needed a huge dataset that consisted of litter found on google streets.
  3.  
  4. Finding litter by ourselves was no easy task. So to automate this time-consuming process, we ended up having to create script that could gathered thousands of random street images, whether they contained litter or not.
  5.  
  6. The next step was preprocessing these images for training.
  7.  
  8. This consisted of getting the human coders to use image labeling software to individually box and label all pieces of litter found in the street images that contained them. In total, over 3000 images were coded, with over 9000 pieces of litter labeled.
  9.  
  10. // Object Detection
  11. For our first prototype, our primary focus was getting tensorflow to identify all objects in an image. This was done by using an rcnn model that was pretrained to a dataset of common objects.
  12.  
  13. Here, identified objects are boxed in green and along with its accuracy score or confidence level of the made prediction.
  14.  
  15. On top shows the number of objects identified and a ranking assigned based off a set range of numbers where
  16. 0 - 2 = rank 1
  17. 3 - 5 = rank 2
  18. 6 - 8 = rank 3
  19. 9 + = rank 4
  20.  
  21. How objects are identified using the rcnn model is a pretty complex process. The jist of how rcnn works is that
  22. First, it selectively searching an image for objects by making a bunch of region proposals where it groups together adjacent pixels by texture, color and intensity.
  23.  
  24. Next, the features of these region proposals are computed and lastly passed into a linear regression model where the regions are classified as objects and then coordinates of boxes that surrounds the objects are tightened.
  25.  
  26. For our second prototype, our focus was training our own rcnn model to specialize in identifying litter only using the litter dataset we gathered.
  27.  
  28. Ill hand it back to andrew who will discuss the training process.
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement