search

KidsWithTokens / MedSegDiff

Medical Image Segmentation with Diffusion Model
MIT License
976 stars 145 forks source link

infer #152

Open 1390806607 opened 5 months ago

1390806607 commented 5 months ago

Hello, may I ask that the shape of your inference code output is [b,2,w,h]. Why only use the -1 dimension

1390806607 commented 5 months ago 
        if args.version == 'new':
            enslist.append(sample[:,-1,:,:])
        else:
            enslist.append(co)
Issues-translate-bot commented 5 months ago

Bot detected the issue body's language is not English, translate it automatically. 👯👭🏻🧑‍🤝‍🧑👫🧑🏿‍🤝‍🧑🏻👩🏾‍🤝‍👨🏿👬🏿

if args.version == 'new': enslist.append(sample[:,-1,:,:]) else: enslist.append(co)