active-vision 0.4.0__py3-none-any.whl → 0.4.1__py3-none-any.whl

active_vision/__init__.py

@@ -1,3 +1,3 @@
-__version__ = "0.4.0"
+__version__ = "0.4.1"
 
 from .core import *

{active_vision-0.4.0.dist-info → active_vision-0.4.1.dist-info}/METADATA

@@ -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.1.dist-info/RECORD

@@ -0,0 +1,6 @@
+active_vision/__init__.py,sha256=vauWDAlrr6fiIylIKSzErXOEopRtTsBk8G4hC9418M0,43
+active_vision/core.py,sha256=ZDRylM3KsoLxy9qA9bld4WxzcKcyCwH8IJ1cFxtz5mE,41607
+active_vision-0.4.1.dist-info/METADATA,sha256=LpgLc_E7jJVXxUHrIPv-1RZq_CEE3enyb0O2PDZMrJM,17262
+active_vision-0.4.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+active_vision-0.4.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+active_vision-0.4.1.dist-info/RECORD,,

active_vision-0.4.0.dist-info/RECORD

@@ -1,6 +0,0 @@
-active_vision/__init__.py,sha256=fzly76fAU-lTwO6Ne4bvQjwryPBaB640_waXGbVmEts,43
-active_vision/core.py,sha256=ZDRylM3KsoLxy9qA9bld4WxzcKcyCwH8IJ1cFxtz5mE,41607
-active_vision-0.4.0.dist-info/METADATA,sha256=8uGHfmSNjsPOxu6tTfUIs2yVlQ3EKvBVMJTB0owSR0E,17954
-active_vision-0.4.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-active_vision-0.4.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-active_vision-0.4.0.dist-info/RECORD,,