|
|
|
|
|
|
|
|
return self |
|
|
return self |
|
|
|
|
|
|
|
|
@torch.no_grad() |
|
|
@torch.no_grad() |
|
|
@torch.cuda.amp.autocast() |
|
|
|
|
|
|
|
|
@torch.cuda.amp.autocast(torch.cuda.is_available()) |
|
|
def forward(self, imgs, size=640, augment=False, profile=False): |
|
|
def forward(self, imgs, size=640, augment=False, profile=False): |
|
|
# Inference from various sources. For height=640, width=1280, RGB images example inputs are: |
|
|
# Inference from various sources. For height=640, width=1280, RGB images example inputs are: |
|
|
# filename: imgs = 'data/samples/zidane.jpg' |
|
|
# filename: imgs = 'data/samples/zidane.jpg' |