H2O Catalog is a collection of 100 different use-cases (with datasets, models, outputs) created using H2O Hydrogen Torch platform. The models cover a range of use cases and industries, making them useful for anyone looking to apply deep learning to their data. To view more about Hydrogen Torch checkout the following links:
H2O Hydrogen Torch is democratizing AI by simplifying and streamlining the process of making state-of-the-art deep learning models for all data scientists, from the novice to the expert. Deep learning is able to learn and make intelligent decisions on its own, even with limited and unstructured data available. H2O Hydrogen Torch unlocks value from this unstructured data to help teams understand it at scale. For the enterprise, H2O Hydrogen achieves AI transformation by changing the way they deliver value to both customers and internal teams.
View Website: https://h2oai.github.io/ht-catalog/
The models in this project have a wide range of problem types corresponding to different types of unstructured data (image/video, text, audio)
Object DetectionImage Metric LearningImage SegmentationImage ClassificationText Sequence to SequenceAudio ClassificationImage RegressionText ClassificationImage Semantic SegmentationText RegressionText Token ClassificationText Metric LearningText Span PredictionSpeech Recognition3D Image Semantic SegmentationAudio Regression
The use-cases described in this project are relevant to different industries, for example:
BankingManufacturingAI4GoodMediaHealthcareGovernmentAgricultureHealthcareSecurityMarketingResearchSports
Following is the collection of 100 use-cases by different industries, different problem types, different data-types, built in H2O Hydrogen Torch. Click on any use-case to get more details.
| # | Use-Case | Industry | Prediction Target | Problem Type | Data Type |
|---|---|---|---|---|---|
| 1. | Document Signature Detection | Banking | Detect signatures in the digital documents | Object Detection | Image |
| 2. | Document Signature Verification | Banking | Verify, match, and compare the digital signatures | Image Metric Learning | Image |
| 3. | Smoke Segmentation | Manufacturing | Segment the presence and location of smoke within images | Image Segmentation | Image |
| 4. | Smoke Classification | Manufacturing | Classify whether a given image contain smoke(s) or not | Image Classification | Image |
| 5. | Underwater Starfish Detection | AI4Good | Detect the underwater starfish using object detection techniques | Object Detection | Image |
| 6. | Food Type Classification | Media | Categorize food types based on their visual characteristics | Image Classification | Image |
| 7. | Surface Crack Classification | Manufacturing | Classify whether a given image contain crack(s) or not | Image Classification | Image |
| 8. | Brain Tumor Classification | Healthcare | Predict the type of tumor based on medical imaging data | Image Classification | Image |
| 9. | Road Quality Classification | Government | Classify the roads based on their quality | Image Classification | Image |
| 10. | Building Segmentation | Government | Segment buildings in the satellite images | Object Detection | Image |
| 11. | Hurriance Damage Classification | AI4Good | Classify if a given region(area) contains flooding damage or not | Image Classification | Image |
| 12. | Flooding Area Segmentation | AI4Good | Segment the water regions in flooding area | Image Segmentation | Image |
| 13. | Product Defect Classification | Manufacturing | Classify the defects in the machinery | Image Classification | Image |
| 14. | Flooding Area Segmentation (multi-label) | AI4Good | Segment the flooding area in UAS (Unarmed Aerial System) imageries | Image Segmentation | Image |
| 15. | Road Segmentation | Government | Segment the roads, potholes, and footpaths in a given image | Image Segmentation | Image |
| 16. | Bird Species Classification | AI4Good | Classify the bird species in a given image | Image Classification | Image |
| 17. | Pest Species Classification | Agriculture | Classify the pest species in a given image | Image Classification | Image |
| 18. | Concrete Fiber segmentation | Manufacturing | Identify and segment fibers in concrete images | Image Segmentation | Image |
| 19. | Document Summarization | Media | Generate a concise summary of a longer text | Text Sequence to Sequence | Text |
| 20. | Whale Audio Detection | AI4Good | Identify whether an audio signal is related to a whale or not | Audio Classification | Audio |
| 21. | Pill Quality Classification | Healthcare | Classify whether a pill is dirty or not | Image Classification | Image |
| 22. | Pill Detection | Healthcare | Detect the pills in a given image | Object Detection | Image |
| 23. | Steel Defect Classification | Manufacturing | Identify and classify defects in steel surfaces | Image Classification | Image |
| 24. | Textile Defect Classification | Manufacturing | identify whether the textile is damage or not | Image Classification | Image |
| 25. | PCB Defect Detection | Manufacturing | Detect and classify the defects in Printed Circuit Boards | Object Detection | Image |
| 26. | Tea Leaves Disease Classification | Agriculture | Identify and classify the disease in the leaves | Image Classification | Image |
| 27. | Brain MRI Segmentation | Healthcare | Segment FLAIR abnormality regions in brain MR images | Image Segmentation | Image |
| 28. | Casting Defect Classification | Manufacturing | Detect and classify the casting defects | Image Classification | Image |
| 29. | Public Crowd Counting | Government | Estimate the number of people in a given area | Image Regression | Image |
| 30. | Document Stamp Detection | Banking | Detect stamps in the digital documents using object detection techniques | Image Segmentation | Image |
| 31. | ASL Alphabet Classification | AI4Good | Classify the American Sign Language (ASL) alphabet letters from images | Image Classification | Image |
| 32. | Crop Classification | Agriculture | Identify different types of crops in an agricultural field | Image Classification | Image |
| 33. | Urban Road Detection | Government | Detect and segment the roads in urban areas | Image Segmentation | Image |
| 34. | Weather Classification | AI4Good | Classify the weather type from a given image | Image Classification | Image |
| 35. | Wind Turbines Detection | Manufacturing | Detect the wind turbines using object detection techniques | Object Detection | Image |
| 36. | Skin Cancer Segmentation | Healthcare | Detect and segment skin lesions in medical images | Image Segmentation | Image |
| 37. | Bird Voice Classification | AI4Good | Identify the bird by using their audio | Audio Classification | Audio |
| 38. | Urban Sound Classification | Government | Classify categories of audio signals in the urban environments | Audio Classification | Audio |
| 39. | Printer Defect Detection | Manufacturing | Detect malicious defects in 3D printer operations | Image Classification | Image |
| 40. | AI-Generated Image Detection | Banking | Classify whether an image is generated by an AI or not | Image Classification | Image |
| 41. | Driving Lane Detection | Government | Identify and segment the lanes on the road | Image Segmentation | Image |
| 42. | Audio Emotion Detection | Banking | Classify human emotions from audio signals | Audio Classification | Audio |
| 43. | Document Classification | Banking | Classify different types of documents based on their content | Image Classification | Image |
| 44. | Germination Seedling Detection | Agriculture | Detect seedlings to assess quality of germination process | Object Detection | Image |
| 45. | Traffic Light Detection | Government | Detect the traffic lights in image data | Object Detection | Image |
| 46. | Trash Detection | Government | Identify and classify various types of trash objects | Object Detection | Image |
| 47. | Hardhat Detection | Manufacturing | Detect the presence of hardhats in images | Object Detection | Image |
| 48. | Heartbeat Sound Classification | Healthcare | Classify heartbeat sounds into different categories | Audio Classification | Audio |
| 49. | Disaster Tweets Classification | Government | Classify tweets as either disaster-related or not disaster-related | Text Classification | Text |
| 50. | Email Spam Classification | Security | Classify emails into spam or non-spam categories | Text Classification | Text |
| 51. | Online Reviews Classification | Media | Classify the online reviews as positive, negative, and neutral | Text Classification | Text |
| 52. | Spoken Digits Classification | Manufacturing | Classify the spoken digits into 0 to 9 | Audio Classification | Audio |
| 53. | Similar Ads Image Identification | Manufacturing | Identify the images of same bicycle within the same advertisement | Image Metric Learning | Image |
| 54. | Car Parts Segmentation | Manufacturing | Segment the car parts in a given image containing one or more cars | Image Semantic Segmentation | Image |
| 55. | Hotel Recommendations | Marketing | Rate the quality of 6 differnet facilities of the hotels using their reviews | Text Regression | Text |
| 56. | Environmental Sound Classification | Government | Classify the environmental sounds using audio classification techniques | Audio Classification | Audio |
| 57. | Physical Objects Entity Recognition | Media | Recognize the physical objects of the tokens in each text | Text Token Classification | Text |
| 58. | News Summarization | Media | Summarize the news articles | Text Sequence to Sequence | Text |
| 59. | Article Readability Scoring | Marketing | Calculate the readability scores of the texts | Text Regression | Text |
| 60. | Toxic Comment Classification | Media | Classify the online comments into different categories | Text Classification | Text |
| 61. | Person Re-Identification | Banking | Identify the images of the same person from set of person images | Image Metric Learning | Image |
| 62. | Protests Related Texts Identification | Security | Identify the protests related texts | Text Classification | Text |
| 63. | Similar Questions Grouping | Media | Identify the similar questions asked in an online forum | Text Metric Learning | Text |
| 64. | Search Queries Wellformedness | Media | Rate the wellformedness of the online search engine queries | Text Regression | Text |
| 65. | Question Answering in Contexts | Media | Answer the questions from given contexts | Text Span Prediction | Text |
| 66. | Speech Commands Classification | Banking | Transcribe the spoken commands and convert them into actionable instructions | Speech Recognition | Speech |
| 67. | Speech Intent Classification | Banking | Categorize the user's intended action or query based on the spoken inputs | Speech Recognition | Speech |
| 68. | Speech Transcript Generation | Government | Generate transcripts for US english speech samples | Speech Recognition | Speech |
| 69. | CT-Scan Chest and Lungs Segmentation | Healthcare | Segment the human chest and lungs in CT-Scan images | 3D Image Semantic Segmentation | 3D Image |
| 70. | Nerves Segmentation in Ultrasound Images | Healthcare | Segment a collection of nerves in the ultrasound images | Image Semantic Segmentation | Image |
| 71. | Covid-19 Cough Audio Regression | Healthcare | Predict the severity of cough using cough audio | Audio Regression | Text |
| 72. | News Article Classification | Media | Classify news articles in different categories | Text Classification | Text |
| 73. | Fake News Classification | Media | Detect fake news using text classification techniques | Text Classification | Text |
| 74. | Biomedical Text Classification | Healthcare | Classify biomedicine scientific articles in different categories | Text Classification | Text |
| 75. | Book Rating Regression | Media | Predict the book rating using their reviews | Text Classification | Text |
| 76. | Disease Classification | Healthcare | Predict potential diseases or medical conditions based on reported symptoms | Text Classification | Text |
| 77. | Tweet Emotion Detection | Media | Detect and classify emotions expressed in tweets | Text Classification | Text |
| 78. | Fake Job Posting Detection | Security | Detect and classify fraudulent job postings in online platforms | Text Classification | Text |
| 79. | Mental Disorder identification | Healthcare | Identify and classify mental disorders | Text Classification | Text |
| 80. | SQL Create Context | Research | Generate SQL queries or statements based on contextual information | Text Span Prediction | Text |
| 81. | Adversarial-QA | Security | Develop and test robust question-answering systems against adversarial attacks | Text Span Prediction | Text |
| 82. | Scientific Papers NER | Research | Extract and classify named entities from scientific papers | Text Sequence to Sequence | Text |
| 83. | Bank Emotion Detection | Banking | Detect and classify customer emotions during interactions with banking services | Text Classification | Text |
| 84. | Insurance Holder Detection | Banking | Determine and classify individuals holding insurance policies or claims | Text Span Prediction | Text |
| 85. | Music Instrument Classification | Media | Classify musical instruments based on audio recordings | Audio Classification | Audio |
| 86. | Human Stress Prediction | Healthcare | Predict individuals stress levels based on physiological or behavioral signals | Text Classification | Text |
| 87. | Gun Type Classification | Government | Recognize and classify firearms or gun-related objects from images or video data | Audio Classification | Audio |
| 88. | Siren Sounds Classification | Government | Detect and classify emergency vehicle siren sounds from audio data | Audio Classification | Audio |
| 89. | Speaker Recognition | Banking | Identify and verify individuals based on vocal characteristics or speech patterns | Audio Classification | Audio |
| 90. | Malicious Drones Prediction | Security | Predict and classify potentially harmful or unauthorized drones | Image Classification | Image |
| 91. | Face Mask Detection | Government | Detect and classify individuals wearing or not wearing face masks from images or video streams | Object Detection | Image |
| 92. | Road Footpath Recognition | Government | Detect and recognize footpaths or pedestrian walkways in images or video data | Image Semantic Segmentation | Image |
| 93. | Resistor Image Classification | Manufacturing | Classify an input resistor image into its corresponding type or category | Image Classification | Image |
| 94. | Tire Texture Classification | Manufacturing | Classify an input tire image into its corresponding texture category | Image Classification | Image |
| 95. | Flower Recognition | Agriculture | Classify an input flower image into its corresponding species or category | Image Classification | Image |
| 96. | Railway Defect Classification | Government | Classify an input image or video frame into the corresponding defect category | Image Classification | Image |
| 97. | Motorcycle Rider Segementation | Government | Segment the motorcycle and the rider from the surrounding nighttime environment | Image Semantic Segmentation | Image |
| 98. | Football Player segmentation | Sports | Segment each football player present in the video frames | Image Classification | Image |
| 99. | Malaria Cell Detection | Healthcare | Identify and classify red blood cells as infected or uninfected with malaria parasites | Image Classification | Image |
| 100. | Audio Anomaly Detection | Manufacturing | Classifies the audio segments into 4 machine parts: valve, fan, slider and pump | Audio Classification | Audio |
| 101. | Deforestation Segmentation | AI4Good | Segment deforested areas with multi-modal satellite imagery | Image Segmentation | Image |
To contribute, please follow the steps mentioned here
Please create an Issue for any improvements, suggestions or errors in the content.
H2O Use-Case Catalog is licensed under the Mozilla Public License Version 2.0. See LICENSE for more information.
Please note that all contributions to this catalog are subject to the copyright terms specified in the COPYRIGHT file.
- Laksika Tharmalingam (https://github.com/laksi1999)
- Sanjeepan Sivapiran (https://github.com/sanjeepan23)
- Shivam Bansal (https://github.com/shivam5992)