Commit 35e9d291 authored by anivegesana's avatar anivegesana
Browse files

Remove debugging registrations

parent 0bfc786c
...@@ -64,7 +64,6 @@ class RevNet(hyperparams.Config): ...@@ -64,7 +64,6 @@ class RevNet(hyperparams.Config):
# Specifies the depth of RevNet. # Specifies the depth of RevNet.
model_id: int = 56 model_id: int = 56
from official.vision.beta.projects.yolo.configs.backbones import DarkNet
@dataclasses.dataclass @dataclasses.dataclass
class Backbone(hyperparams.OneOfConfig): class Backbone(hyperparams.OneOfConfig):
...@@ -86,4 +85,3 @@ class Backbone(hyperparams.OneOfConfig): ...@@ -86,4 +85,3 @@ class Backbone(hyperparams.OneOfConfig):
efficientnet: EfficientNet = EfficientNet() efficientnet: EfficientNet = EfficientNet()
spinenet: SpineNet = SpineNet() spinenet: SpineNet = SpineNet()
mobilenet: MobileNet = MobileNet() mobilenet: MobileNet = MobileNet()
darknet: DarkNet = DarkNet()
...@@ -48,6 +48,7 @@ class Parser(parser.Parser): ...@@ -48,6 +48,7 @@ class Parser(parser.Parser):
aug_rand_hflip=True, aug_rand_hflip=True,
dtype='float32'): dtype='float32'):
"""Initializes parameters for parsing annotations in the dataset. """Initializes parameters for parsing annotations in the dataset.
Args: Args:
output_size: `Tenssor` or `list` for [height, width] of output image. The output_size: `Tenssor` or `list` for [height, width] of output image. The
output_size should be divided by the largest feature stride 2^max_level. output_size should be divided by the largest feature stride 2^max_level.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment