EDUENet This network is mainly enhamcement the COVID-19 CT images.
- Requirement
(1) python envirenment
opencv-python
scikit-image
pillow
torchvision
hdf5storage
ninja
pytorch
(2) halcon envirement
- Datasets
The data set we used is available for download on: https://www.kaggle.com/plameneduardo/sarscov2- ctscan-dataset
- Training
(1) Change the location of the dataset under train.json.
(2) python train.py
- Testing
(1) Change the location of the image you want to test under test.py
(2) The tested images are run under ED_model.hdev in the halcon.
The software is for educaitonal and academic research purpose only.