Web: https://www.reddit.com/r/deeplearning/comments/xg5ebw/keras_unet_collection_and_binary_classification/

Sept. 16, 2022, 10:45 p.m. | /u/jantonio78

Deep Learning reddit.com

Hi there, I'm trying to learn to use the keras-unet-collection package. I created a simple unet model for segmentation of grayscale images with a shape of (256, 256, 1):

models.unet_2d(input_size=(256,256,1), filter_num=[64, 128, 256, 512, 1024], n_labels=2)

The output of the model has a shape of (256, 256, 2), but my masks have a shape of (256, 256, 1). I tried using n_labells=1, because one neuron for the channel should be enough (either there is a 1 and the pixel is …

binary classification collection deeplearning keras unet

Postdoctoral Fellow: ML for autonomous materials discovery

@ Lawrence Berkeley National Lab | Berkeley, CA

Research Scientists

@ ODU Research Foundation | Norfolk, Virginia

Embedded Systems Engineer (Robotics)

@ Neo Cybernetica | Bedford, New Hampshire

2023 Luis J. Alvarez and Admiral Grace M. Hopper Postdoc Fellowship in Computing Sciences

@ Lawrence Berkeley National Lab | San Francisco, CA

Senior Manager Data Scientist

@ NAV | Remote, US

Senior AI Research Scientist

@ Earth Species Project | Remote anywhere

Research Fellow- Center for Security and Emerging Technology (Multiple Opportunities)

@ University of California Davis | Washington, DC

Staff Fellow - Data Scientist

@ U.S. FDA/Center for Devices and Radiological Health | Silver Spring, Maryland

Staff Fellow - Senior Data Engineer

@ U.S. FDA/Center for Devices and Radiological Health | Silver Spring, Maryland

Computational Linguist

@ Constant Contact | Waterloo, ON/Hybrid, ON

Senior Data Scientist

@ Advian | Espoo, Finland

Data Scientist, EU In-stock Management

@ Gopuff | London, England