search

hhk7734 / tensorflow-yolov4

YOLOv4 Implemented in Tensorflow 2.
MIT License
136 stars 75 forks source link

Installation error in Windows #5

Closed Twinkle126 closed 4 years ago

Twinkle126 commented 4 years ago

Hi,

When I run command python -m pip install yolov4 in my virtual env, I am getting below error- installation_error_windows

My system setup is as below- Python 3.6.10 Windows 10 Cuda 10.1 cuDNN 7.6.0

hhk7734 commented 4 years ago

@Twinkle126

I tested it on Linux only. :sweat_smile: I'll test it after a while

hhk7734 commented 4 years ago

Did you install TensorFlow 2?

Twinkle126 commented 4 years ago

Yes, I installed TF2.

On Mon, 29 Jun, 2020, 1:21 pm Hyeonki Hong, notifications@github.com wrote:

Did you install TensorFlow 2?

— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub https://github.com/hhk7734/tensorflow-yolov4/issues/5#issuecomment-650994294, or unsubscribe https://github.com/notifications/unsubscribe-auth/AIS5HX6QLZYLSC5THZH7G3TRZBBYVANCNFSM4OK3PMDQ .

hhk7734 commented 4 years ago

Hmm... I tested yolov4 and it was successfull.

python -m pip install -U setuptools pip
python -m pip install tensorflow opencv-python ipython
python -m pip install yolov4
ipython

In [1]: from yolov4.tf import YOLOv4
   ...:
   ...: yolo = YOLOv4()
   ...:
   ...: yolo.classes = "C:/Users/hhk7734/Desktop/yolov4/data/classes/coco.names"
   ...:
   ...: yolo.make_model()
   ...: yolo.load_weights("C:/Users/hhk7734/Desktop/yolov4.weights", weights_type="yolo")
   ...:
   ...: yolo.inference(
   ...:     media_path="C:/Users/hhk7734/Desktop/yolov4/data/kite.jpg",
   ...: )
Twinkle126 commented 4 years ago

It seems issue with Python 3.6.10. I will upgrade it to 3.7 and then use it. Thanks for looking into this and your prompt reply.