Bohao-Lee / CME

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Code of base training.

Retiina opened this issue · comments

Sir, would you like to upload the code of base model training? i.e., the code of Class Max-margin

I have uploaded the base training in the code.

 python tool/train_decoupling_disturbance.py cfg/fewshot/metayolo_split${split}.data cfg/darknet_dynamic.cfg cfg/reweighting_net_decoupling.cfg darknet19_448.conv.23 backup/split${split}_base

in train_model.sh. Hope this can help u.

@Bohao-Lee Thank you! May I ask where is the implementation of Class Max-margin?

You can view it in core/region_loss.py, class RegionLossV3 . Hope this can help u.