Skip to content

Commit

Permalink
Refactor with generating sequence
Browse files Browse the repository at this point in the history
  • Loading branch information
@oarriaga
oarriaga committed Jun 3, 2020
1 parent de27ba3 commit 2a3d8b0
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions examples/discovery_of_latent_keypoints/discover_latent_keypoints.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
from paz.models import Projector
from paz.pipelines import KeypointNetInference
from paz.pipelines import KeypointSharedAugmentation
from paz.core.sequencer import GeneratingSequencer
from paz.abstract import GeneratingSequence
from paz.optimization import KeypointNetLoss
from paz.optimization.callbacks import DrawInferences

Expand Down Expand Up @@ -99,9 +99,8 @@

# setting sequencer
input_shape = (args.image_size, args.image_size, 3)
projector = Projector(focal_length, True)
processor = KeypointSharedAugmentation(scene, projector, args.image_size)
sequencer = GeneratingSequencer(processor, args.batch_size, as_list=True)
processor = KeypointSharedAugmentation(scene, args.image_size)
sequencer = GeneratingSequence(processor, args.batch_size, as_list=True)

# model instantiation
model = KeypointNetShared(input_shape, args.num_keypoints,
Expand Down

0 comments on commit 2a3d8b0

Please sign in to comment.