search

ZexinChen / AlphaTracker

AlphaTracker is a computer vision pipeline with the practical and real-time advantages , which requires minimal hardware requirements and produces reliable tracking of multiple unmarked animals. An easy-to-use user interface further enables manual inspection and curation of results.
61 stars 16 forks source link

Errors in training AlphaTracker #10

Open Cocochen1998 opened 2 years ago

Cocochen1998 commented 2 years ago

First of all, thanks for developing AlphaTracker! It is really a useful tool in animal behavior study. Unfortunately, I encountered with some errors. Could you please help me with these errors? I used AlphaTracker via Google Colab. And there were errors in Step 8.

image

And I also got warnings in Step7.

image image
laurelrr commented 2 years ago

Hello! Thank you for your interest in using AlphaTracker.

I would like to collect some information from you while I investigate your issue. Can you provide the following information please:

  1. Could you please verify that you ran Steps 4-6 and upload the output from step 6 here?
  2. Did you alter any of the variables in Step 6, such as changing the names of the training data directory in your google drive?
  3. Did you get a message to 'RESTART RUNTIME' and press the restart runtime button in Step 7? If so, please rerun the colab notebook from the beginning and do NOT restart the runtime. (This will erase all your variables including mounting your google drive.)

Thanks!

Cocochen1998 commented 2 years ago

Hello! Thanks for replying. I did run Steps 4-6. And here is the output from step 6.

image

And a TRAINING_DATA file also appeared in my google drive with my renamed training images in it. I altered the variables(image_root_list, json_file_path and video_path) according to the training data directory in my google drive in Step 6. I pressed the restart runtime button in Step 7 and reran the notebook from the beginning. This time I did not restart the runtime. But the error occurred in Step 8 anyway.

image
Cocochen1998 commented 2 years ago

And I also tried to install alphatracker on linux environment. Could you please tell me how to modify the Makefile in install.sh? I am new to the linux environment.

image
Fang-Haoshu commented 2 years ago

Hi, what is your Nvidia GPU card and your driver version?

Cocochen1998 commented 2 years ago

Hello, thanks for you reply! My Nvidia GPU card is Quadro RTX 4000, and my driver version is 470.103.01.

Fang-Haoshu commented 2 years ago

Hi，if you follow the install.sh and use conda to install cuda-toolkit-11.1, then the current make file should work fine and no modification is needed. You can refer to https://github.com/ZexinChen/AlphaTracker/blob/main/Tracking/AlphaTracker/train_yolo/darknet/Makefile#L8 if you want to use other cuda version.

Cocochen1998 commented 2 years ago

Hello! Thanks a lot for your help! I have solved my problem.