Search in sources :

Example 6 with ProductSet

use of com.google.cloud.vision.v1.ProductSet in project java-vision by googleapis.

the class ITSystemTest method setUp.

@BeforeClass
public static void setUp() throws IOException {
    imageAnnotatorClient = ImageAnnotatorClient.create();
    /* create product */
    productSearchClient = ProductSearchClient.create();
    Product createProduct = Product.newBuilder().setName(PRODUCT_ID).setDisplayName(PRODUCT_DISPLAY_NAME).setProductCategory(PRODUCT_CATEGORY).build();
    CreateProductRequest productRequest = CreateProductRequest.newBuilder().setParent(LOCATION_NAME).setProduct(createProduct).build();
    product = productSearchClient.createProduct(productRequest);
    productName = getName(product.getName());
    formatProductName = ProductName.of(PROJECT_ID, COMPUTE_REGION, productName).toString();
    /* create product set */
    ProductSet createProductSet = ProductSet.newBuilder().setDisplayName(PRODUCT_SET_DISPLAY_NAME).build();
    CreateProductSetRequest productSetRequest = CreateProductSetRequest.newBuilder().setParent(LOCATION_NAME).setProductSet(createProductSet).setProductSetId(PRODUCT_SET_ID).build();
    productSet = productSearchClient.createProductSet(productSetRequest);
    productSetName = getName(productSet.getName());
    formatProductSetName = ProductSetName.of(PROJECT_ID, COMPUTE_REGION, productSetName).toString();
    /* add product to product set */
    AddProductToProductSetRequest productToProductSetRequest = AddProductToProductSetRequest.newBuilder().setName(formatProductSetName).setProduct(formatProductName).build();
    productSearchClient.addProductToProductSet(productToProductSetRequest);
    /* create reference image */
    ReferenceImage createReferenceImage = ReferenceImage.newBuilder().setUri(SAMPLE_BUCKET + "product_search/shoes_1.jpg").build();
    CreateReferenceImageRequest imageRequest = CreateReferenceImageRequest.newBuilder().setParent(formatProductName).setReferenceImage(createReferenceImage).build();
    referenceImage = productSearchClient.createReferenceImage(imageRequest);
    referenceImageName = getName(referenceImage.getName());
    formatReferenceImageName = ReferenceImageName.of(PROJECT_ID, COMPUTE_REGION, productName, referenceImageName).toString();
}
Also used : CreateProductRequest(com.google.cloud.vision.v1.CreateProductRequest) CreateReferenceImageRequest(com.google.cloud.vision.v1.CreateReferenceImageRequest) Product(com.google.cloud.vision.v1.Product) CreateProductSetRequest(com.google.cloud.vision.v1.CreateProductSetRequest) ProductSet(com.google.cloud.vision.v1.ProductSet) ReferenceImage(com.google.cloud.vision.v1.ReferenceImage) AddProductToProductSetRequest(com.google.cloud.vision.v1.AddProductToProductSetRequest) BeforeClass(org.junit.BeforeClass)

Example 7 with ProductSet

use of com.google.cloud.vision.v1.ProductSet in project java-vision by googleapis.

the class ProductSetManagement method createProductSet.

// [START vision_product_search_create_product_set]
/**
 * Create a product set
 *
 * @param projectId - Id of the project.
 * @param computeRegion - Region name.
 * @param productSetId - Id of the product set.
 * @param productSetDisplayName - Display name of the product set.
 * @throws IOException - on I/O errors.
 */
public static void createProductSet(String projectId, String computeRegion, String productSetId, String productSetDisplayName) throws IOException {
    try (ProductSearchClient client = ProductSearchClient.create()) {
        // A resource that represents Google Cloud Platform location.
        String formattedParent = ProductSearchClient.formatLocationName(projectId, computeRegion);
        // Create a product set with the product set specification in the region.
        ProductSet myProductSet = ProductSet.newBuilder().setDisplayName(productSetDisplayName).build();
        CreateProductSetRequest request = CreateProductSetRequest.newBuilder().setParent(formattedParent).setProductSet(myProductSet).setProductSetId(productSetId).build();
        ProductSet productSet = client.createProductSet(request);
        // Display the product set information
        System.out.println(String.format("Product set name: %s", productSet.getName()));
    }
}
Also used : ProductSearchClient(com.google.cloud.vision.v1.ProductSearchClient) CreateProductSetRequest(com.google.cloud.vision.v1.CreateProductSetRequest) ProductSet(com.google.cloud.vision.v1.ProductSet)

Aggregations

ProductSet (com.google.cloud.vision.v1.ProductSet)5 ProductSearchClient (com.google.cloud.vision.v1.ProductSearchClient)3 Test (org.junit.Test)3 CreateProductSetRequest (com.google.cloud.vision.v1.CreateProductSetRequest)2 ListProductSetsPagedResponse (com.google.cloud.vision.v1.ProductSearchClient.ListProductSetsPagedResponse)2 AbstractMessage (com.google.protobuf.AbstractMessage)2 AddProductToProductSetRequest (com.google.cloud.vision.v1.AddProductToProductSetRequest)1 CreateProductRequest (com.google.cloud.vision.v1.CreateProductRequest)1 CreateReferenceImageRequest (com.google.cloud.vision.v1.CreateReferenceImageRequest)1 GetProductSetRequest (com.google.cloud.vision.v1.GetProductSetRequest)1 Product (com.google.cloud.vision.v1.Product)1 ReferenceImage (com.google.cloud.vision.v1.ReferenceImage)1 BeforeClass (org.junit.BeforeClass)1