search

Ruixxxx / LSSANet

[Official Repo: MICCAI 2022] LSSANet: A Long Short Slice-Aware Network for Pulmonary Nodule Detection
GNU General Public License v3.0
12 stars 3 forks source link

batch_size #3

Closed kk17860 closed 1 year ago

kk17860 commented 1 year ago

Thanks for your wonderful work in pulmonary nodule detection. Why is batch_size equal to 1 in single_test.py? It takes a lot of time to test. Can I set batch_size equal to 2?

Ruixxxx commented 1 year ago

Thanks for your wonderful work in pulmonary nodule detection. Why is batch_size equal to 1 in single_test.py? It takes a lot of time to test. Can I set batch_size equal to 2?

Thanks for your interest! Since the test CT images are not cropped to 128x128x128 and the single_test.py does not support multi-gpu test, the batch_size is set to 1. You may set the batch_size to 2 if you have enough GPU memory.

kk17860 commented 1 year ago

Thanks for your reply.