网络
我的PyTorch COVID-Net实现,有关原始工作,请参见: :
这个github的目的是重现结果,而不是要求最先进的性能!
另外,也可以在此处使用Google Colab Notebook进行即插即用培训和评估[
即将更新~~ 9月中旬
目录
入门
安装
要安装所需的python软件包,请使用以下命令
pip install -r requirements.txt
用法
训练
网络将形状为(N,224,224,3)的图像作为输入,并将softmax概率输出为(N,C),其中N是批次数,C是输出类别。
要从头开始训练网络,只需执行python main.py参数进行训练
-h, --help show this help message and exit
--batch_size BATCH_SIZE
batch size for training
--log_interval LOG_INTERVAL