Kapo Model structure

[abhigoku10]

@Kanav123 @wmcnally i am having a framework that has both yolov5 and kapao in the same folder structure , when i run inference of the kapao it tries to load from the model folder yolo.py file , i tried renaming the file and loading the kapao but it gives an error, when looked deep into it during training the model node type properties are model.yolo_ . Can we change the model node properties while retraining ? can you please share ur thoughts

Thanks in adavance

[wmcnally]

Sorry, I don’t understand what you’re trying to do. Why do you have the yolov5 project in the same folder structure?

[abhigoku10]

@wmcnally apologies for my above explanation, please find the below explanation i have framework which does arms and weapons and human pose so its a common project structure so i have both of them in the same package when i load kapao then i get error where the model is loading from the base yolov5 instead of kapo model/yolo.py , even if i change the file name to models/yolo_kapao.py i get error missing file since model properties is loading yolo.py file how to modify this

[wmcnally]

Ah, I see. You could try retraining KAPAO from scratch, e.g., python train.py --weights '' --cfg yolov5s6.yaml. I do not have another solution for this at the moment. This might be a good place to start looking: https://stackoverflow.com/questions/13398462/unpickling-python-objects-with-a-changed-module-path/13405732

[abhigoku10]

@wmcnally thanks for the response
1.to retraining, the model from scratch based on the command you had mentioned what is the accuracy drop we get and how much time will it take to train on the T4/V100 gpu system

[wmcnally]

I expect the accuracy to drop 1-2 AP points without COCO pretraining, but I'm not sure exactly how much it will drop. Training KAPAO-S on CrowdPose takes approximately 6 hours on 4 V100 GPUs using a global batch size of 128.