https://github.com/nateraw/huggingpics

🤗🖼️ HuggingPics: Fine-tune Vision Transformers for anything using images found on the web.
https://github.com/nateraw/huggingpics

huggingface machine-learning pytorch vision-transformers

Last synced: 6 days ago
JSON representation

🤗🖼️ HuggingPics: Fine-tune Vision Transformers for anything using images found on the web.

• Host: GitHub
• URL: https://github.com/nateraw/huggingpics
• Owner: nateraw
• Created: 2021-06-29T21:34:02.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2024-05-07T23:39:54.000Z (8 months ago)
• Last Synced: 2025-01-05T02:09:41.988Z (13 days ago)
• Topics: huggingface, machine-learning, pytorch, vision-transformers
• Language: Jupyter Notebook
• Homepage:
• Size: 972 KB
• Stars: 289
• Watchers: 4
• Forks: 27
• Open Issues: 3
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# 🤗🖼️HuggingPics

Fine-tune Vision Transformers for **anything** using images found on the web.

Check out the video below for a walkthrough of this project! ⤵️

[![IMAGE ALT TEXT](http://img.youtube.com/vi/f9ZjgWBAxEQ/0.jpg)](http://www.youtube.com/watch?v=f9ZjgWBAxEQ "Video Title")

## Usage

Click on the link below to try it out:



## How does it work?

### 1. You define your search terms

![pick search terms](images/pick_search_terms.png)

### 2. We download ~150 images for each and use them to fine-tune a ViT

![image search results](images/image_search_results.png)

### 3. You push your model to HuggingFace's Hub to share your results with the world

![push to hub](images/push_to_hub.png)

### Your auto-generated model repo will look something like [this](https://huggingface.co/nateraw/rare-puppers). Pretty cool, eh? 😎

![push to hub](images/rare_puppers_repo.png)

## Examples

💡 If you need some inspiration, take a look at the examples below:

|            | [nateraw/rare-puppers](https://huggingface.co/nateraw/rare-puppers) | [nateraw/pasta-pizza-ravioli](https://huggingface.co/nateraw/pasta-pizza-ravioli) | [nateraw/baseball-stadium-foods](https://huggingface.co/nateraw/baseball-stadium-foods) | [nateraw/denver-nyc-paris](https://huggingface.co/nateraw/denver-nyc-paris) |

| ---------- | ------------------------------------------------------------------- | --------------------------------------------------------------------------------- | --------------------------------------------------------------------------------------- | --------------------------------------------------------------------------- |

| **term_1** | samoyed                                                             | pizza                                                                             | cotton candy                                                                            | denver                                                                      |

| **term_2** | shiba inu                                                           | pasta                                                                             | hamburger                                                                               | new york city                                                               |

| **term_3** | corgi                                                               | ravioli                                                                           | hot dog                                                                                 | paris                                                                       |

| **term_4** |                                                                     |                                                                                   | nachos                                                                                  |                                                                             |

| **term_5** |                                                                     |                                                                                   | popcorn                                                                                 |                                                                             |

You can see a full list of model repos created using this tool by [clicking here](https://huggingface.co/models?filter=huggingpics)