search

layumi / 2015_Face_Detection

CVPR2015 Cascade CNNs for Face Detection
Other
136 stars 70 forks source link

net24 and net48 are gpuArray #4

Open ikvision opened 8 years ago

ikvision commented 8 years ago

Hi, Thank you for your excellent work, clear code and easy to use and adapt.

On Test branch, pretrained models f24net-cpu.mat and f48net-cpu.mat are saved in gpu mode. To avoid warning in test time on cpu, need to run: NET = VL_SIMPLENN_MOVE(NET, 'cpu')

layumi commented 8 years ago

@ikvision thank you! You're right. I don't change some filter's weight from gpu array to single. I think Matlab will do it automatically although it gives warning.

ikvision commented 8 years ago

@layumi , from my understanding, in case the computer has no GPU matlab will create a warning with no side effects, as GpuArray will be converted to CPU on load. But if the computer has a GPU, net24 and net48 will be loaded to GPU memory while the rest of the networks will be loaded to CPU memory. For each input image, part of the flow will run on GPU and rest will on CPU. During test time image's pixels will be transferred from CPU to GPU and back. I think, it may cause slowdown vs a full flow on CPU as intented