xmuyzz / 3d-cnn-pytorch Goto Github PK
View Code? Open in Web Editor NEWPyTorch implementation for 3D CNN models for medical image data (1 channel gray scale images).
PyTorch implementation for 3D CNN models for medical image data (1 channel gray scale images).
Channel x Time(or NumFrames) x Height x Width
I am attempting to load my model in the following format
How do input 16 frames of size 456x456 into the EfficientNet model?
I am trying to classify 16 frame snippets from video clips.
#load model
from efficientnet_pytorch_3d import EfficientNet3D
model_EfficientNet3D = EfficientNet3D.from_name("efficientnet-b7", in_channels=3)
summary(model_EfficientNet3D, input_size=(3, 16, 456, 456))
I have 16 images I want to send into the EfficientNet3D, is this possible?
A similar comment was made by @shijianjian here : shijianjian/EfficientNet-PyTorch-3D#11 (comment)
"Say, change from Conv3D(kernel_size=(3, 3, 3)) to Conv3D(kernel_size=(1, 3, 3))will probably work for your case."
I am very lost here because I dont understand where to actually change this code.
I cant even find this specific code in the model file: Conv3D(kernel_size=(3, 3, 3))
Also since I am using the pip install efficientnet-pytorch for this EfficientNet3D, I am having trouble understanding how to manipulate the actual model code since its a pip install.
If I was to manually load the efficientnet-pytorch model with PyTorch, where and how would I be able to load the model weights?
Please help me in any way, this is a wonderful project and I am grateful for the contribution. Just need a bit of support on loading the model.
How can I use this 3D-EfficientNet model as a backbone feature extractor? I would need to export features at a certain layer instead of getting a final classification.
Thanks so much !!
Could you please explain the arguement sample_size
. Thanks!
good job do you have plans to continue updating?
A declarative, efficient, and flexible JavaScript library for building user interfaces.
๐ Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
An Open Source Machine Learning Framework for Everyone
The Web framework for perfectionists with deadlines.
A PHP framework for web artisans
Bring data to life with SVG, Canvas and HTML. ๐๐๐
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
Some thing interesting about web. New door for the world.
A server is a program made to process requests and deliver data to clients.
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
Some thing interesting about visualization, use data art
Some thing interesting about game, make everyone happy.
We are working to build community through open source technology. NB: members must have two-factor auth.
Open source projects and samples from Microsoft.
Google โค๏ธ Open Source for everyone.
Alibaba Open Source for everyone
Data-Driven Documents codes.
China tencent open source team.