pipert.contrib.canny_demo.save_model

Module Contents

Classes

Net

class pipert.contrib.canny_demo.save_model.Net(threshold=10.0, use_cuda=False)[source]

Bases: torch.nn.Module

forward(self, img)[source]
pipert.contrib.canny_demo.save_model.model[source]