search

healthonrails / annolid

An annotation and instance segmentation-based multiple animal tracking and behavior analysis package.
Other
41 stars 9 forks source link

Error in training the model #7

Closed Hatem-Jr closed 3 years ago

Hatem-Jr commented 3 years ago

Hi it's me again XD, I'm having an issue where I used the dataset in the github repo and changed the file locations in the data.yaml file to match the dataset file and picture locations but here's what's in the data.yaml file that I use as a config file for training (these paths are copied from the file explorer and the backslash is changed to match the format):

DATASET: name: 'novelctrlk6_8' train_info: 'D:/custom code/Weights/Github Dataset/novelctrlk6_8_coco_dataset/train/annotations.json' train_images: 'D:/custom code/Weights/Github Dataset/novelctrlk6_8_coco_dataset/train/JPEGImages' valid_info: 'D:/custom code/Weights/Github Dataset/novelctrlk6_8_coco_dataset/valid/annotations.json' valid_images: 'D:/custom code/Weights/Github Dataset/novelctrlk6_8_coco_dataset/valid/JPEGImages' class_names: ['nose', 'left_ear', 'right_ear', 'tail_base', 'mouse', 'centroid'] YOLACT: name: 'novelctrlk6_8' dataset: 'dataset_novelctrlk6_8_coco' max_size: 512

and here's the output error in the console: Screenshot (11)

Plus I re-installed annolid again according to the annolid installation document this time: https://healthonrails.github.io/annolid/guide.html#video-guide so as not to put that error on the installation because this time there was no installation error and this was the final message: Screenshot (12)

I know I'm asking too much but if there could be a way on how to train or if I'm even using the correct files whether I'm training or using visualization or even tracking animals because I feel a part of it is that I'm using the wrong files to begin with

Thanks :D ps:- can you take down those 2 videos because I might get in trouble with my uni: https://www.youtube.com/watch?v=1Ua7Mpsi1yM&t=13s https://www.youtube.com/watch?v=3ScyTbTa7fc

I sent them to you on a google drive and was expecting if you were gonna send anything to send them on a drive as well Sorry :)

healthonrails commented 3 years ago

If you want to train YOLACT on your own computer, you need to install Pytorch 1.4.0 and torchvision==0.5.0. Please check the details from here https://github.com/healthonrails/annolid/blob/master/docs/tutorials/Train_networks_tutorial_v1.0.1.ipynb and https://github.com/healthonrails/yolact.

I have deleted the videos based on your video.

Hatem-Jr commented 3 years ago

I did what you said now is that normal ? Screenshot (13) I tried installing them without specifying the version and it worked but same error appeared in the first comment nonetheless Thanks for deleting the videos btw :)

healthonrails commented 3 years ago

You might need to check your CUDA versions.

CUDA 10.1

pip install torch==1.4.0 torchvision==0.5.0

CUDA 9.2

pip install torch==1.4.0+cu92 torchvision==0.5.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html For more info, please check here https://pytorch.org/get-started/previous-versions/.