voxel51/fiftyone

[FR] Add remaining models from TF2 object detection model zoo

Open

#746 创建于 2021年1月7日

在 GitHub 查看
 (5 评论) (0 反应) (0 负责人)Python (400 fork)batch import
coreenhancementgood first issuemodel zoo

仓库指标

Star
 (4,021 star)
PR 合并指标
 (平均合并 3天 11小时) (30 天内合并 161 个 PR)

描述

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)

贡献者指南