OCDNet
OCDNet for scene text detection. Detects arbitrary-oriented text regions in natural images using a differentiable binarization approach.
Set
model.pretrained_model_path
for pretrained weights.
For TAO Deploy TensorRT actions (
gen_trt_engine
, TensorRT
evaluate
, and TensorRT
inference
), read
references/tao-deploy-ocdnet.md
first. Deploy spec templates live in this skill's
references/
folder with the
spec_template_deploy_*.yaml
prefix.
The PyT OCDNet CLI supports
train
,
evaluate
,
export
,
inference
,
prune
,
quantize
, and
default_specs
. It does not expose PyT-side
retrain
or
gen_trt_engine
subcommands. The model skill exposes
retrain
by running
ocdnet train
with
model.load_pruned_graph: true
and
model.pruned_graph_path
. Resume from an epoch checkpoint uses
ocdnet train
plus
train.resume_training_checkpoint_path
. TensorRT engine generation is owned by the deploy workflow.
Dataclass Schemas
Generated TAO Core schemas are packaged in
schemas/.schema.json
, with
schemas/manifest.json
listing available actions. Each generated schema also emits
references/spec_template_.yaml
from the schema top-level
default
field. AutoML enablement is declared at the model layer in
references/skill_info.yaml
via
automl_enabled
. Runnable AutoML still requires
schemas/train.schema.json
and
references/spec_template_train.yaml
to exist and parse. Use the packaged train schema for
automl_default_parameters
,
automl_disabled_parameters
, defaults, min/max bounds, enums, option weights, math conditions, dependencies, and popular parameters. Do not expect
~/tao-core
at runtime; maintainers regenerate schemas/templates before packaging the skill bank.
Train Action Policy
This model is AutoML-enabled at the model layer. Before handling any train-stage request, read
references/skill_info.yaml
and resolve the run override from either an explicit
automl_policy
value or the user's workflow request. Use
automl_policy: on
by default and only expose
on
/
off
in new launch prompts. Treat phrases like "turn off AutoML", "disable AutoML", "no HPO", or "plain training" as
automl_policy: off
for this run only. When
automl_policy: on
,
automl_enabled: true
, and both
schemas/train.schema.json
and
references/spec_template_train.yaml
are packaged, route the train action through
tao-skill-bank:tao-run-automl
by default with this model's
skill_dir
. Preserve workflow/application overrides for datasets, specs, output directories, GPU/platform settings, parent checkpoints, and
automl_policy
. Use direct model training only when
automl_policy: off
or the packaged train schema/template is missing; in the missing-schema case, report that AutoML is enabled but not runnable for this model until schemas are generated.
Show more
Installs
585
Repository
nvidia/skills
GitHub Stars
1.9K
First Seen
Jun 8, 2026
Security Audits
Gen Agent Trust Hub
Pass
Socket
Pass
Snyk
Pass