active-vision 0.4.0__tar.gz → 0.4.1__tar.gz

{active_vision-0.4.0 → active_vision-0.4.1}/.github/workflows/docs.yml

@@ -28,17 +28,5 @@ jobs:
               run: |
                   cp README.md docs/index.md
 
-                  jupyter nbconvert nbs/imagenette/quickstart.ipynb --to markdown --output-dir docs/
-                  sed -i 's/\x1b\[[0-9;]*m//g' docs/quickstart.md
-
-                  jupyter nbconvert nbs/imagenette/preparing_data.ipynb --to markdown --output-dir docs/
-                  sed -i 's/\x1b\[[0-9;]*m//g' docs/preparing_data.md
-
-                  jupyter nbconvert nbs/imagenette/active_learning.ipynb --to markdown --output-dir docs/
-                  sed -i 's/\x1b\[[0-9;]*m//g' docs/active_learning.md
-
-                  jupyter nbconvert nbs/imagenette/train.ipynb --to markdown --output-dir docs/
-                  sed -i 's/\x1b\[[0-9;]*m//g' docs/train.md
-
             - name: Deploy
               run: mkdocs gh-deploy --force

{active_vision-0.4.0 → active_vision-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: active-vision
-Version: 0.4.0
+Version: 0.4.1
 Summary: Active learning for computer vision.
 Project-URL: Homepage, https://github.com/dnth/active-vision
 Project-URL: Bug Tracker, https://github.com/dnth/active-vision/issues
@@ -16,6 +16,7 @@ Requires-Dist: loguru>=0.7.3
 Requires-Dist: seaborn>=0.13.2
 Requires-Dist: timm>=1.0.13
 Requires-Dist: transformers>=4.48.0
+Requires-Dist: xinfer>=0.3.2
 Provides-Extra: dev
 Requires-Dist: black>=22.0; extra == 'dev'
 Requires-Dist: flake8>=4.0; extra == 'dev'
@@ -42,8 +43,6 @@ Requires-Dist: pygments; extra == 'docs'
 Requires-Dist: pymdown-extensions; extra == 'docs'
 Requires-Dist: sphinx; extra == 'docs'
 Requires-Dist: watchdog; extra == 'docs'
-Provides-Extra: zero-shot
-Requires-Dist: x-infer>=0.3.2; extra == 'zero-shot'
 Description-Content-Type: text/markdown
 
 [![Python Version](https://img.shields.io/badge/python-3.10%2B-blue?style=for-the-badge&logo=python&logoColor=white)](https://pypi.org/project/active-vision/)
@@ -139,6 +138,8 @@ pip install -e .
 [![Open In Colab][colab_badge]](https://colab.research.google.com/github/dnth/active-vision/blob/main/nbs/imagenette/quickstart.ipynb)
 [![Open In Kaggle][kaggle_badge]](https://kaggle.com/kernels/welcome?src=https://github.com/dnth/active-vision/blob/main/nbs/imagenette/quickstart.ipynb)
 
+The following are code snippets for the active learning loop in active-vision. I recommend running the quickstart notebook in Colab or Kaggle to see the full workflow.
+
 ```python
 from active_vision import ActiveLearner
 
@@ -149,11 +150,10 @@ al = ActiveLearner(name="cycle-1")
 al.load_model(model="resnet18", pretrained=True)
 
 # Load dataset 
-train_df = pd.read_parquet("training_samples.parquet")
 al.load_dataset(train_df, filepath_col="filepath", label_col="label", batch_size=8)
 
 # Train model
-al.train(epochs=10, lr=5e-3, head_tuning_epochs=3)
+al.train(epochs=10, lr=5e-3)
 
 # Evaluate the model on a *labeled* evaluation set
 accuracy = al.evaluate(eval_df, filepath_col="filepath", label_col="label")
@@ -177,8 +177,8 @@ samples = al.sample_combination(
     },
 )
 
-# Launch a Gradio UI to label the low confidence samples, save the labeled samples to a file
-al.label(samples, output_filename="combination.parquet")
+# Launch a Gradio UI to label the samples, save the labeled samples to a file
+al.label(samples, output_filename="samples.parquet")
 ```
 
 ![Gradio UI](https://raw.githubusercontent.com/dnth/active-vision/main/assets/labeling_ui.png)
@@ -191,18 +191,11 @@ Once complete, the labeled samples will be save into a new df.
 We can now add the newly labeled data to the training set.
 
 ```python
-# Add newly labeled data to the dataset
 al.add_to_dataset(labeled_df, output_filename="active_labeled.parquet")
 ```
 
 Repeat the process until the model is good enough. Use the dataset to train a larger model and deploy.
 
-> [!TIP]
-> For the toy dataset, I got to about 93% accuracy on the evaluation set with 200+ labeled images. The best performing model on the [leaderboard](https://github.com/fastai/imagenette) got 95.11% accuracy training on all 9469 labeled images.
-> 
-> This took me about 6 iterations of relabeling. Each iteration took about 5 minutes to complete including labeling and model training (resnet18). See the [notebook](./nbs/04_relabel_loop.ipynb) for more details.
->
-> But using the dataset of 200+ images, I trained a more capable model (convnext_small_in22k) and got 99.3% accuracy on the evaluation set. See the [notebook](./nbs/05_retrain_larger.ipynb) for more details.
 
 
 ## 📊 Benchmarks

{active_vision-0.4.0 → active_vision-0.4.1}/README.md

@@ -91,6 +91,8 @@ pip install -e .
 [![Open In Colab][colab_badge]](https://colab.research.google.com/github/dnth/active-vision/blob/main/nbs/imagenette/quickstart.ipynb)
 [![Open In Kaggle][kaggle_badge]](https://kaggle.com/kernels/welcome?src=https://github.com/dnth/active-vision/blob/main/nbs/imagenette/quickstart.ipynb)
 
+The following are code snippets for the active learning loop in active-vision. I recommend running the quickstart notebook in Colab or Kaggle to see the full workflow.
+
 ```python
 from active_vision import ActiveLearner
 
@@ -101,11 +103,10 @@ al = ActiveLearner(name="cycle-1")
 al.load_model(model="resnet18", pretrained=True)
 
 # Load dataset 
-train_df = pd.read_parquet("training_samples.parquet")
 al.load_dataset(train_df, filepath_col="filepath", label_col="label", batch_size=8)
 
 # Train model
-al.train(epochs=10, lr=5e-3, head_tuning_epochs=3)
+al.train(epochs=10, lr=5e-3)
 
 # Evaluate the model on a *labeled* evaluation set
 accuracy = al.evaluate(eval_df, filepath_col="filepath", label_col="label")
@@ -129,8 +130,8 @@ samples = al.sample_combination(
     },
 )
 
-# Launch a Gradio UI to label the low confidence samples, save the labeled samples to a file
-al.label(samples, output_filename="combination.parquet")
+# Launch a Gradio UI to label the samples, save the labeled samples to a file
+al.label(samples, output_filename="samples.parquet")
 ```
 
 ![Gradio UI](https://raw.githubusercontent.com/dnth/active-vision/main/assets/labeling_ui.png)
@@ -143,18 +144,11 @@ Once complete, the labeled samples will be save into a new df.
 We can now add the newly labeled data to the training set.
 
 ```python
-# Add newly labeled data to the dataset
 al.add_to_dataset(labeled_df, output_filename="active_labeled.parquet")
 ```
 
 Repeat the process until the model is good enough. Use the dataset to train a larger model and deploy.
 
-> [!TIP]
-> For the toy dataset, I got to about 93% accuracy on the evaluation set with 200+ labeled images. The best performing model on the [leaderboard](https://github.com/fastai/imagenette) got 95.11% accuracy training on all 9469 labeled images.
-> 
-> This took me about 6 iterations of relabeling. Each iteration took about 5 minutes to complete including labeling and model training (resnet18). See the [notebook](./nbs/04_relabel_loop.ipynb) for more details.
->
-> But using the dataset of 200+ images, I trained a more capable model (convnext_small_in22k) and got 99.3% accuracy on the evaluation set. See the [notebook](./nbs/05_retrain_larger.ipynb) for more details.
 
 
 ## 📊 Benchmarks