Doo Yon Kim comments

Repositories
Issues
Comments

Results 1 comments of


                                            Doo Yon Kim

Pre processing of input data

``` image_transforms = transforms.Compose([ transforms.Resize((256, 256)), transforms.ToTensor(), ]) def classify1(model, image_transforms, image_path ): image = Image.open( image_path) image = image.convert('RGB') image = image_transforms(image) image = image.unsqueeze(0) image = image.cuda() output...