tusen-ai / simpledet

A Simple and Versatile Framework for Object Detection and Instance Recognition
Apache License 2.0
3.08k stars 488 forks source link

测试脚本内存溢出问题 #347

Closed zerojuzi closed 3 years ago

zerojuzi commented 3 years ago

我观察到detection_test.py脚本中第89行 loader = Loader(roidb=roidb, transform=transform, data_name=data_name, label_name=label_name, batch_size=1, shuffle=False, num_worker=4, num_collector=2, worker_queue_depth=2, collector_queue_depth=2) 会不停的被创建,python无法回收,当测试大批量数据时容易导致内存泄漏,我手动清除变量 del loader gc.collect() 也无法解决 请问作者有什么好的解决办法吗,非常感谢!!!!!