FoundationVision / GLEE

[CVPR2024 Highlight]GLEE: General Object Foundation Model for Images and Videos at Scale
https://glee-vision.github.io/
MIT License
1.02k stars 82 forks source link

too many values to unpack (expected 2) #27

Open hezta opened 4 months ago

hezta commented 4 months ago

运行时出现如下问题,请问如何解决: File "E:\hezt\vis\lib\site-packages\gradio\queueing.py", line 527, in process_events response = await route_utils.call_process_api( File "E:\hezt\vis\lib\site-packages\gradio\route_utils.py", line 270, in call_process_api output = await app.get_blocks().process_api( File "E:\hezt\vis\lib\site-packages\gradio\blocks.py", line 1847, in process_api result = await self.call_function( File "E:\hezt\vis\lib\site-packages\gradio\blocks.py", line 1433, in call_function prediction = await anyio.to_thread.run_sync( File "E:\hezt\vis\lib\site-packages\anyio\to_thread.py", line 56, in run_sync return await get_async_backend().run_sync_in_worker_thread( File "E:\hezt\vis\lib\site-packages\anyio_backends_asyncio.py", line 2144, in run_sync_in_worker_thread return await future File "E:\hezt\vis\lib\site-packages\anyio_backends_asyncio.py", line 851, in run result = context.run(func, args) File "E:\hezt\vis\lib\site-packages\gradio\utils.py", line 805, in wrapper response = f(args, **kwargs) File "E:\tool\GLEE-new\app.py", line 169, in segmentimage (outputs,) = GLEEmodel(infer_image, prompt_list, task="coco", batch_name_list=batch_category_name, is_train=False) ValueError: too many values to unpack (expected 2)

stihuangyuan commented 3 months ago

same error

Caesar6666666 commented 2 months ago

有人解决了吗

stihuangyuan commented 2 months ago

有人解决了吗

change app.py line 169 (outputs,_) = GLEEmodel(infer_image, prompt_list, task="coco", batch_name_list=batch_category_name, istrain=False) to (outputs,) = GLEEmodel(infer_image, prompt_list, task="coco", batch_name_list=batch_category_name, is_train=False)[0] same for line 175

yueyunqingwu commented 1 month ago

我在使用pro模型推理时遇到该问题,定位在GLEE_Model类forward方法的返回值没有根据is_train这个布尔值进行区分,按理说is_train=False时只用return推理结果即可,稍微改一下代码即可,位置在projects/GLEE/glee/models/glee_model.py的359-370行。