snowzach / doods2

API for detecting objects in images and video streams using Tensorflow
MIT License
221 stars 28 forks source link

ERROR - Could not create detector pytorch/pytorch: No module named 'psutil' #65

Closed Travis-Prall closed 1 year ago

Travis-Prall commented 1 year ago

Appears I can't use the pytorch detector with the new image.

2022-11-06 13:55:39,914 - doods.doods - INFO - Registered detector type:tflite name:default 2022-11-06 13:55:41,215 - doods.doods - INFO - Registered detector type:tensorflow name:tensorflow /usr/local/lib/python3.8/dist-packages/torch/hub.py:266: UserWarning: You are about to download and run code from an untrusted repository. In a future release, this won't be allowed. To add the repository to your trusted list, change the command to {calling_fn}(..., trust_repo=False) and a command prompt will appear asking for an explicit confirmation of trust, or load(..., trust_repo=True), which will assume that the prompt is to be answered with 'yes'. You can also use load(..., trust_repo='check') which will only prompt for confirmation if the repo is not already trusted. This will eventually be the default behaviour warnings.warn( Downloading: "https://github.com/ultralytics/yolov5/zipball/master" to /root/.cache/torch/hub/master.zip 2022-11-06 13:55:42,912 - doods.doods - ERROR - Could not create detector pytorch/pytorch: No module named 'psutil' 2022-11-06 13:55:42,938 - uvicorn.error - INFO - Started server process [1] 2022-11-06 13:55:42,938 - uvicorn.error - INFO - Waiting for application startup. 2022-11-06 13:55:42,939 - uvicorn.error - INFO - Application startup complete. 2022-11-06 13:55:42,939 - uvicorn.error - INFO - Uvicorn running on http://0.0.0.0:8080 (Press CTRL+C to quit)

snowzach commented 1 year ago

It's because the pytorch/yolo is basically downloaded at will.. they changed some of their requirements, I've updated the image, it should work again.

snowzach commented 1 year ago

Assuming it's fixed...