yuhuan-wu / JCS

[IEEE TIP 2021] COVID-CS Dataset and Code of JCS: An explainable COVID-19 diagnosis system by joint classification and segmentation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

neverstoplearn opened this issue · comments

where i need to change?

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

def __init__(self, block, layers, baseWidth = 26, scale = 4, num_classes=1000, input_features=False):

def __init__(self, input_features=False):

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

def __init__(self, block, layers, baseWidth = 26, scale = 4, num_classes=1000, input_features=False):

def __init__(self, input_features=False):

thanks.