voxel51/fiftyone
Vedi su GitHub[FR] Add remaining models from TF2 object detection model zoo
Open
#746 aperta il 7 gen 2021
coreenhancementgood first issuemodel zoo
Metriche repository
- Star
- (4021 star)
- Metriche merge PR
- (Merge medio 3g 11h) (161 PR mergiate in 30 g)
Descrizione
Currently, FiftyOne only contains CenterNet 512x512 with the Hourglass104 backbone from the TF2 model zoo: https://github.com/tensorflow/models/blob/archive/research/object_detection/g3doc/tf2_detection_zoo.md
Other models can be added by updating fiftyone/zoo/models/manifest-tf.json to add relevant information for the new models following centernet-hg104-512-coco-tf2.
It can then be tested with:
conda create -n fo
conda activate fo
conda install tensorflow-gpu=2.2.0
conda install ipython
git clone https://github.com/voxel51/fiftyone.git
cd fiftyone
bash install.bash
eta install models
# Add new models
import fiftyone as fo
import fiftyone.zoo as foz
dataset, session = fo.quickstart()
model = foz.load_zoo_model("NEW_MODEL_NAME", error_level=1)
dataset.apply_model(model, label_field="predictions")
session = fo.launch_app(dataset)