search

yuhuan-wu / JCS

[IEEE TIP 2021] COVID-CS Dataset and Code of JCS: An explainable COVID-19 diagnosis system by joint classification and segmentation
48 stars 13 forks source link

How can i use the stage 1 feature in the code? thanks. #5

Closed neverstoplearn closed 2 years ago

neverstoplearn commented 2 years ago

where i need to change?

yuhuan-wu commented 2 years ago

set input_features of the backbone code to True so the network can be fed with the stage-1 features. Two places: https://github.com/yuhuan-wu/JCS/blob/fa4bb6ae452c6aa042db8c762822e72df65d9519/Models/res2net.py#L122 https://github.com/yuhuan-wu/JCS/blob/fa4bb6ae452c6aa042db8c762822e72df65d9519/Models/vgg.py#L7

neverstoplearn commented 2 years ago

set input_features of the backbone code to True so the network can be fed with the stage-1 features. Two places:

https://github.com/yuhuan-wu/JCS/blob/fa4bb6ae452c6aa042db8c762822e72df65d9519/Models/res2net.py#L122

https://github.com/yuhuan-wu/JCS/blob/fa4bb6ae452c6aa042db8c762822e72df65d9519/Models/vgg.py#L7

thanks.