Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Refactor `IMAGE_CLASSIFIER_HEADS` to be general classification heads

See original GitHub issue

🚀 Feature

We have the IMAGE_CLASSIFIER_HEADS registry in image/classification/heads.py but it’s actually useful for all classification tasks where you have an embedding followed by a classification layer. We should refactor it out and rename it to just CLASSIFIER_HEADS in core/classification.py

Issue Analytics

State:
Created a year ago
Reactions:1
Comments:5 (5 by maintainers)

Top GitHub Comments

1reaction

Abelarmcommented, Oct 2, 2022

Hi @Abelarm - Yeah, I can reproduce. I’ll have to check tomorrow on this (sorry, it’s a bit late here), but for now, to unblock you from working on this, I guess you can safely ignore this error, and do:
# build flash in editable mode
pip install -e .

# in case you are working on docs
git submodule update --init --recursive
# install doc requirements
pip install -r requirements/docs.txt

Sorry didn’t know your time-zone, I am not blocked (just created the PR) I just wanted to point the error to you guys, so you can track it

1reaction

krshrimalicommented, Oct 2, 2022

Hi, @Abelarm - I’m sure @ethanwharris would love for you to pick this up. I’m assigning you this issue, please let us know if you have any queries. 😃

Top Results From Across the Web

Build a Deep CNN Image Classifier with ANY Images - YouTube

Get the Code https://github.com/nicknochnack/ImageClassificationSo...you wanna build your own image classifier eh?

Image Classification - Lightning Flash

Typically, Image Classification is used to identify images containing a single object. The task predicts which 'class' the image most likely belongs to...

Multi-Label Image Classification with TensorFlow 2.0

Train a multi-label image classifier with macro soft-F1 loss in TensorFlow ... The new classification head will be trained from scratch so that...

Bridging Image Classification and Image-text Alignment for ...

Abstract. Image classification, which classifies images by pre-defined categories, has been the dominant approach to visual representation.

Image Classification in Machine Learning [Intro + Tutorial]

Once you've annotated enough instances, you are ready to train your first image classifier using V7! Simply head over to the "Neural Networks"...